arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2210.07565cs.CL
收藏

面向中文少样本学习的模块化提示多任务预训练

Multitask Pre-training of Modular Prompt for Chinese Few-Shot Learning

Fudan University(复旦大学)

Tianxiang Sun, Zhengfu He, Qin Zhu, Xipeng Qiu, Xuanjing Huang

AI总结:

针对少样本场景下提示调优效果不佳的问题,提出在38个中文任务上预训练的可组合模块化提示MP2,通过统一为阅读理解任务、选择性激活组合提示,在两种学习范式下均显著优于现有方法,且仅需8个参数即可快速适配下游任务。

摘要:

提示调优是一种使预训练语言模型适配下游任务的参数高效方法。尽管已有研究表明,当训练数据充足时,提示调优的性能可与全模型调优持平,但在少样本学习场景下它往往表现不佳。本文提出多任务预训练模块化提示(Multi-task Pre-trained Modular Prompt,MP2),以提升少样本学习场景下的提示调优效果。MP2是在38个中文任务上预训练得到的一组可组合提示。在下游任务中,这些预训练提示会被选择性激活并组合,从而对未见过的任务具备强大的组合泛化能力。为缩小预训练与微调之间的差距,我们将上游和下游任务统一建模为机器阅读理解任务。在梯度下降和黑盒调优两种学习范式下开展的大量实验表明,在少样本设置中,MP2的性能显著优于提示调优、全模型调优以及已有的提示预训练方法。此外,我们还证明,仅需学习8个参数来组合这些预训练模块化提示,MP2就能实现对下游任务惊人快速且强劲的适配效果。

英文摘要:

Prompt tuning is a parameter-efficient approach to adapting pre-trained language models to downstream tasks. Although prompt tuning has been shown to match the performance of full model tuning when training data is sufficient, it tends to struggle in few-shot learning settings. In this paper, we present Multi-task Pre-trained Modular Prompt (MP2) to boost prompt tuning for few-shot learning. MP2 is a set of combinable prompts pre-trained on 38 Chinese tasks. On downstream tasks, the pre-trained prompts are selectively activated and combined, leading to strong compositional generalization to unseen tasks. To bridge the gap between pre-training and fine-tuning, we formulate upstream and downstream tasks into a unified machine reading comprehension task. Extensive experiments under two learning paradigms, i.e., gradient descent and black-box tuning, show that MP2 significantly outperforms prompt tuning, full model tuning, and prior prompt pre-training methods in few-shot settings. In addition, we demonstrate that MP2 can achieve surprisingly fast and strong adaptation to downstream tasks by merely learning 8 parameters to combine the pre-trained modular prompts.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑