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

面向语言理解和生成的实例感知提示学习

Instance-aware Prompt Learning for Language Understanding and Generation

National Laboratory of Pattern Recognition, Institute of Automation, CAS(中国科学院自动化研究所模式识别国家重点实验室);School of Artificial Intelligence, University of Chinese Academy of Sciences(中国科学院大学人工智能学院);CAS Center for Excellence in Brain Science and Intelligence Technology(中国科学院脑科学与智能技术卓越创新中心)

Feihu Jin, Jinliang Lu, Jiajun Zhang, Chengqing Zong

摘要:

最近,提示学习已经成为一种新的范式,利用预训练的语言模型(PLM),并取得了可喜的成果,在下游任务的参数可以忽略不计的增加。离散和连续提示的当前用法假定特定任务的提示是固定的,并且任务中的所有样本共享相同的提示。然而,一个任务可能包含相当多样化的样本,其中一些是容易的,而另一些是困难的,并且需要多样化的提示。在本文中,我们提出了一种实例感知提示学习方法,该方法为每个实例学习不同的提示。具体来说,我们假设每个可学习的提示标记对不同的实例有不同的贡献,我们通过计算实例和每个提示标记之间的相关性得分来学习贡献。贡献加权提示将是实例感知的。我们将我们的方法应用到单向和双向的PLM上的语言理解和生成任务。大量的实验表明,我们的方法得到了相当大的改善相比,强基线。特别是,我们的方法达到了国家的最先进的SuperGLUE少样本学习基准。

英文摘要:

Recently, prompt learning has become a new paradigm to utilize pre-trained language models (PLMs) and achieves promising results in downstream tasks with a negligible increase of parameters. The current usage of discrete and continuous prompts assumes that the prompt is fixed for a specific task and all samples in the task share the same prompt. However, a task may contain quite diverse samples in which some are easy and others are difficult, and diverse prompts are desirable. In this paper, we propose an instance-aware prompt learning method that learns a different prompt for each instance. Specifically, we suppose that each learnable prompt token has a different contribution to different instances, and we learn the contribution by calculating the relevance score between an instance and each prompt token. The contribution weighted prompt would be instance aware. We apply our method to both unidirectional and bidirectional PLMs on both language understanding and generation tasks. Extensive experiments demonstrate that our method obtains considerable improvements compared to strong baselines. Especially, our method achieves the state-of-the-art on the SuperGLUE few-shot learning benchmark.

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

↑