Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

P-Tuning v2: Prompt Tuning Can Be Comparable to Fine-tuning Universally Across Scales and Tasks

About

Prompt tuning, which only tunes continuous prompts with a frozen language model, substantially reduces per-task storage and memory usage at training. However, in the context of NLU, prior work reveals that prompt tuning does not perform well for normal-sized pretrained models. We also find that existing methods of prompt tuning cannot handle hard sequence labeling tasks, indicating a lack of universality. We present a novel empirical finding that properly optimized prompt tuning can be universally effective across a wide range of model scales and NLU tasks. It matches the performance of finetuning while having only 0.1%-3% tuned parameters. Our method P-Tuning v2 is an implementation of Deep Prompt Tuning \cite{li2021prefix,qin2021learning} optimized and adapted for NLU. Given the universality and simplicity of P-Tuning v2, we believe it can serve as an alternative to finetuning and a strong baseline for future research.Our code and data are released at https://github.com/THUDM/P-tuning-v2.

Xiao Liu, Kaixuan Ji, Yicheng Fu, Weng Lam Tam, Zhengxiao Du, Zhilin Yang, Jie Tang• 2021

Related benchmarks

TaskDatasetResultRank
Commonsense ReasoningPIQA
Accuracy66.2
757
Natural Language UnderstandingGLUE (test)
SST-2 Accuracy92.2
416
Mathematical ReasoningGSM8K
Accuracy51.1
388
Question AnsweringSQuAD v1.1 (dev)
F1 Score94.4
380
Science Question AnsweringARC Challenge
Accuracy51.3
354
Question AnsweringOBQA
Accuracy76.1
347
Reading ComprehensionBoolQ
Accuracy71.2
279
Mathematical ReasoningGSM8K
GSM8K Accuracy (%)2.65
204
Mathematical ReasoningAQUA
Accuracy39.63
167
Question AnsweringSQuAD v2.0 (dev)
F185.5
163
Showing 10 of 32 rows

Other info

Follow for update