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
751
Natural Language UnderstandingGLUE (test)
SST-2 Accuracy92.2
416
Question AnsweringSQuAD v1.1 (dev)
F1 Score94.4
380
Science Question AnsweringARC Challenge
Accuracy51.3
342
Mathematical ReasoningGSM8K
Accuracy51.1
312
Question AnsweringOBQA
Accuracy76.1
300
Reading ComprehensionBoolQ
Accuracy71.2
279
Question AnsweringSQuAD v2.0 (dev)
F185.5
163
Science Question AnsweringARC Easy
Accuracy68.5
155
Mathematical ReasoningAQUA
Accuracy39.63
146
Showing 10 of 29 rows

Other info

Follow for update