PPT: Pre-trained Prompt Tuning for Few-shot Learning
Yuxian Gu1,3∗, Xu Han2,3∗, Zhiyuan Liu2,3,4, Minlie Huang1,3,4† 1The CoAI group, Tsinghua University, Beijing, China 2The THUNLP group, Tsinghua University, Beijing, China 3Institute for Artificial Intelligence, State Key Lab of Intelligent Technology and Systems, Beijing National Research Center for Information Science and Technology, Department of Computer Science and Technology, Tsinghua University, Beijing, China 4 Beijing Academy of Artificial Intelligence, BAAI, Beijing, China
Abstract
Prompts for pre-trained language models (PLMs) have shown remarkable performance by bridging the gap between pre-training tasks and various downstream tasks. Among these methods, prompt tuning, which freezes PLMs and only tunes soft prompts, provides an efficient and effective solution for adapting large-scale PLMs to downstream tasks. However, prompt tuning is yet to be fully explored. In our pilot experiments, we find that prompt tuning performs comparably with conventional full-model tuning when downstream data are sufficient, whereas it is much worse under few-shot learning settings, which may hinder the application of prompt tuning. We attribute this low performance to the manner of initializing soft prompts. Therefore, in this work, we propose to pre-train prompts by adding soft prompts into the pre-training stage to obtain a better initialization. We name this Pre-trained Prompt Tuning framework “PPT”. To ensure the generalization of PPT, we formulate similar classification tasks into a unified task form and pre-train soft prompts for this unified task. Extensive experiments show that tuning pre-trained prompts for downstream tasks can reach or even outperform full-model fin
原文 arXiv:2109.04332;中英对照 + 大白话阅读 https://aha.fim.ai/paper/2109.04332v3