文章来源 | 恒源云社区(专注人工智能/深度学习GPU免费加速平台,官方体验网址:https://gpushare.com)
原文作者 | Mathor
原文地址 | https://gpushare.com/forum/topic/681/%E6%9C%AA%E9%97%BBprompt%E5%90%8D?_=1635736812393&lang=zh-CN
个人觉得2021年NLP最火的两个idea,一个是对比学习(Contrastive Learning),另一个就是Prompt
Prompt说简单也简单,看了几篇论文以及博客后发现其实就是构建一个语言模版,从而实现无监督训练。但是细想起来又觉得复杂,因为总感觉里面还有很多细节,因此本文就来从头梳理一下Prompt(Prompt很多地方会翻译成「范式」,但是「范式」这个词本身也不好理解,因此读者把他看作是「模板」即可)
今天我还与室友讨论预训练模型(例如BERT)到底做了什么,我给出的回答是
预训练模型提供了一个非常好的初始化参数,这组参数在预训练任务上的表现非常好(预训练损失非常低),但是由于下游任务千奇百怪,我们需要在这组参数的基础上进行Fine-tune以适应我们的下游任务(使得下游任务的损失值非常低)
上面这段话其实隐含了目前做NLP任务的大致流程,即"Pre-train, Fine-tune",而对我们来说实际上大部分时候都是直接拿别人预训练好的模型做Fine-tune,并没有Pre-train这一步
融入了Prompt的模式大致可以归纳成"Pre-train, Prompt, and Predict",在该模式中,下游任务被重新调整成类似预训练任务的形式。例如,通常的预训练任务有MLM(Masked Language Model),在文本情感分类任务中,对于"I love this movie"这句输入,可以在后面加上Prompt:“the movie is ___”,组成如下这样一句话:
I love this movie, the movie is ___
然后让预训练模型用表示情感的答案(例如"great"、"terrible"等)做完形填空,最后再将该答案转换为情感分类的标签。这样一来,我们就可以通过构造合适的「模板」,控制模型的输出空间,从而训练一个完全无监督的预训练模型来解决各种各样的下游任务
注意,Prompt设计的这种完形填空和MLM任务是有区别的,二者虽然都是都是词分类,但是候选集不同,MLM的候选词是整个词库,不过如果是生成任务,那么Prompt和MLM的候选集就是一样的,都是整个词库
对于输入文本 x x x,存在一个函数 f Prompt ( x ) f_{\text{Prompt}}(x) fPrompt(x),将 x x x转化成 x ‘ x ‘ x‘的形式,即
x ’ = f Prompt ( x ) x^{’}=f_{\text{Prompt}}(x) x’=fPrompt(x)
该函数通常会进行两步操作:
以前文提到的例子为例,在文本情感分类任务中,假设输入是
x = "I love this movie"
使用的模板是
[X]. Overall, it was a [Z] movie
那么得到的 x ′ x ′ x′就应该是
I love this movie. Overall, it was a [Z] movie
在实际情况中,Prompt来填充答案的位置一般在句中或句末。如果在句中,一般称这种Prompt为Cloze Prompt;如果在句末,一般称这种Prompt为Prefix Prompt。 [ X ] [X] [X]和 [ Z ] [Z] [Z]的位置、数量以及使用模板句的不同,都有可能对结果造成影响,因此需要灵活调整
上面讲的都是简单的情感分类任务的Prompt设计,读者看到这里自然而然的会想到,其他NLP任务的Prompt如何设计呢?实际上刘鹏飞大神在他的论文中给我们提供了一些参考
Text Generation中摘要任务里有一个关键字TL;DR,这其实是Too Long; Don’t Read的缩写
有上述Prompt的基础后,我们可以得知Prompt的设计主要包含两部分:
[X]. Overall, It was [Z]
在基于Prompt的微调方法中,不同的模板和标签词对最终结果影响很大,下图是陈丹琦团队论文中的实验结果
从上图我们可以看出两点:
great/terribel
和cat/dog
这两组标签词的效果不一样,而且即便是相同标签词,互换顺序也会导致最终效果有所变化,例如cat/dog
和dot/cat
Prompt大概可以从下面三个角度进行设计:
Prompt的形状主要指的是 [ X ] [X] [X]和 [ Z ] [Z] [Z]的位置和数量。上文提到的Cloze Prompt与Maksed Language Model的训练方式非常类似,因此对于MLM任务来说,Cloze Prompt更合适;对于生成任务或者使用自回归LM解决的任务,Prefix Prompt更合适
Prompt的模板最开始是人工设计的,人工设计一般基于人类的自然语言知识,力求得到语义流畅且高效的「模板」。例如,Petroni等人在著名的LAMA数据集中为知识探针任务人工设计了Cloze Templates;Brown等人为问答、翻译和探针等任务设计了Prefix Templates。人工设计模板的优点是直观,但缺点是需要很多实验、经验以及语言专业知识。下图是GPT Understands, Too论文中的一个实验结果
可以看到不同的Prompt只有细微的区别,有的甚至只是增加减少一个词,但是最后的结果会差几十个点
为了解决人工设计模板的缺点,许多研究员开始探究如何自动学习到合适的模板。自动学习的模板又可以分为离散(Discrete Prompts)和连续(Continuous Prompts)两大类。离散方法主要包括:Prompt Mining,Prompt Paraphrasing,Gradient-based Search,Prompt Generation和Prompt Scoring;连续的则主要包括Prefix Tuning,Tuning Initialized with Discrete prompts,Hard-Soft Prompt Hybrid Tuning,P-Tuning v2
简单说一下上述几种方法,首先是离散的Prompt Mining,这篇文章发表在TACL 2020,讲的是如何拿预训练语言模型当作「知识库」使用,并且引入了依存树和Paraphrase(转述)等方法来挖掘更好的「模板」,下图是实验结果
可以看到,被挖掘出来的若干「连接谓词」相比于人工设计的「模板」结果提升还是很明显的
有很多种方法可以实现Prompt Paraphrsing,例如「回译」,我们通过DeepL翻译看个例子:
这样我们就得到了x shares a border with y
的一个Prompt Paraphrasing:x and y share a boundary
论文BARTScore干脆给我们提供了一张表,里面有各种词组的同义替换,这个我再熟悉不过了,因为以前英语考试我也背过类似的东西
Gradient-based Search(基于梯度的搜索)是由论文AUTOPROMPT提出的,这篇文章发表在EMNLP 2020,它的主要思想用下面这张图就可以表示
上图中,a real joy
是原始的输入句子 x i n p xinp xinp ,红色的Trigger tokens是由 x i n p xinp xinp「激发」的相关词汇集合 x t r i g x trig xtrig ,根据 T e m p l a t e λ Template λ Templateλ的配置,将 x t r i g xtrig xtrig和 x i n p xinp xinp 组合起来构造最终的输入 x p r o m p t xprompt xprompt,送入Masked LM预测情感标签。下面的表格增加了很多NLP其他任务的例子
关于如何生成 x t r i g xtrig xtrig集合,实际上主要使用的是HotFlip和对抗训练的思想,感兴趣的同学可以看原论文以及HotFlip: White-box adversarial examples for text classification、Universal Adversarial Triggers for Attacking and Analyzing NLP这两篇论文
Prompt Generation是陈丹琦团队的一项工作,主要是把Seq2Seq预训练模型T5应用到模板搜索的过程。T5基于多种无监督目标进行预训练,其中最有效的一个无监督目标就是:利用 < X >
Thank you <X> me to your party <Y> week
T5会在 < X > for inviting
,在 < Y > last
。很显然,T5这种方式很适合生成模板,而且不需要指定模板的token数。具体来说,有三种可能的生成方式
< S 1 S_1 S1> → < X {X} X> M ( y ) {M}(y) M(y) < Y {Y} Y> < S 1 S_1 S1>
< S 1 S_1 S1> → < S 1 S_1 S1>< X {X} X> M ( y ) {M}(y) M(y) < Y {Y} Y>
< S 1 S_1 S1> < S 2 S_2 S2>→ < S 1 S_1 S1> < X {X} X> M ( y ) {M}(y) M(y) < Y {Y} Y> < S 2 S_2 S2>
具体的模板生成过程如下图所示:
首先在标签词前后添加填充位 < X >
我还想说一下这篇论文中另外一个有意思的点,最后送入模型进行预测的句子还拼接上了每种类别的「示例」(Demonstration),如下图所示
这种Prompt的设计有点像是在做语义相似度任务, X X X为原始Input句子,已知 Y Y Y为正例, Z Z Z为负例,构造了如下形式的输入:
X是[MASK]例?Y为正例;Z为负例
这有点像是编程语言中的三目运算符,或者说相当于让模型比较 X X X与 Y Y Y、 Z Z Z的语义相似度。这里我们自然而然会想问: Y Y Y、 Z Z Z是如何挑选出来的?实际上是依据下面两条规则:
构造Prompt的初衷是能够找到一个合适的方法,让Pre-trained Language Model(PLM)更好地输出我们想要的结果,但其实并不一定要将Prompt的形式设计成人类可以理解的自然语言,只要机器理解就行了。因此,还有一些方法探索连续型Prompts——直接作用到模型的Embedding空间。连续型Prompts去掉了两个约束条件:
Prefix Tuning最开始由Li等人提出,这是一种在输入句子前添加一组连续型向量的方法,该方法保持PLM的参数不动,仅训练前缀(Prefix)向量。Prefix Tuning的提出主要是为了做生成任务,因此它根据不同的模型结构定义了不同的Prompt拼接方式,在GPT类的Auto-Regressive(自回归)模型上采用的是 [ P r e f i x ; x ; y ] [Prefix;x;y] [Prefix;x;y]的方式,在T5类的Encoder-Decoder模型上采用的是 [ P r e f i x ; x ; P r e f i x [Prefix;x;Prefix [Prefix;x;Prefix′ ; y ] ;y] ;y]的方式
输入部分 P r e f i x , x , y Prefix,x,y Prefix,x,y的Position id分别记作 P i d x Pidx Pidx, X i d x Xidx Xidx, Y i d x Yidx Yidx。Prefix Tuning初始化一个可训练的矩阵,记作 P P Pθ$∈R
∣P idx ∣×dim(h i ) ,其中
上述公式的含义是,索引iii如果属于前缀的部分,则从 P θ Pθ Pθ中抽取向量; i i i如果不是前缀部分,则由参数固定的预训练模型生成对应的向量。训练目标为:
max ϕ log p ϕ ( y ∣ x ) = ∑ i ∈ Yidx log p ϕ ( z i ∣ h < i ) \mathop{\text{max}}\limits_{\phi} \ \log p_{\phi}(y\mid x) = \sum\limits_{i\in \text{Y}{\text{idx}}} \log p{\phi} (z_i\mid h_{ϕmax logpϕ(y∣x)=i∈Yidx∑logpϕ(zi∣h<i)
P P Pθ本质上是一个矩阵,而生成一个矩阵的方法又很多,可以用
nn.Embedding()
,或者nn.Linear()
同样是在连续空间上搜索Prompt,OptiPrompt构建的「模板」并不局限于前缀,也可以在句子的中间
首先根据AutoPrompt定义一个Prompt模板:
[ x ] [ v ] 1 [ v ] 2 … [ v ] m [ MASK ] [x]\ [v]_1\ [v]_2\ …\ [v]_m\ [\text{MASK}] [x] [v]1 [v]2 … [v]m [MASK]
其中 [ v ] i [v]i [v]i为一个连续型向量(与BERT的输入维度一致)。OptiPrompt还考虑以人工构建的离散Prompt作为起点,在连续空间上进行搜索以构建较优的Prompt。例如 [ x ] [x] [x] is [ M A S K ] [MASK] [MASK]citizen可以转换为
[ x ] [ v ] 1 [ MASK ] [ v ] 2 [x]\ [v]_1\ [\text{MASK}]\ [v]_2 [x] [v]1 [MASK] [v]2
将is
和citizen
对应的input Embedding作为 [ v ] 1 [v]1 [v]1和 [ v ] 2 [v]2 [v]2的初始化
Hard-Soft Prompt Hybrid Tuning方法可以说是人工设计和自动学习的结合,它通常不单纯使用可学习的Prompt模板,而是在人工设计的模板中插入一些可学习的Embedding。实际上有了上面的基础我们都知道,连续的Prompt要比离散的Prompt好一点,但是在此基础上还有什么改进的余地吗?Liu等人提出的P-Tuning解决了Prompt token之间的关联性问题
之前连续的Prompt生成方式无非都是训练一个矩阵,然后通过索引出矩阵的某几行向量拼起来。坦白地说,我们希望这些prompt token Embedding之间有一个比较好的关联性,而不是独立地学习,为了解决这个问题,P-Tuning引入了一个Prompt Encoder(如下图b所示)
上图a是传统的离散型Prompt,我们把生成离散Prompt token的东西叫做Prompt Generator;上图b首先传入一些Virtual(Pseudo)token,例如BERT词表中的[unused1],[unused2],…当然,这里的token数目是一个超参数,插入的位置也可以调整。将这些Pseudo token通过一个Prompt Encoder得到连续的向量 h 0 h0 h0,…, h m h_m hm ,其中
即,Prompt Encoder是由BiLSTM+MLP组成的一个简单网络。作者还发现加入一些anchor token(领域或者任务相关的token)可以有助于Template的优化。例如文本蕴含任务,输入是前提和假设,判断是否蕴含。一个连续的模版是
[PRE] [ continuous tokens ] [ HYP ] [ continuous tokens ] [ MASK ] \text{[PRE]} [\text{continuous tokens}][\text{HYP}][\text{continuous tokens}] [\text{MASK}] [PRE][continuous tokens][HYP][continuous tokens][MASK]
在其中加入一个anchor token:[?]
效果会更好,此时模板变成
[PRE] [ continuous tokens ] [ HYP ] ? [ continuous tokens ] [ MASK ] \text{[PRE]} [\text{continuous tokens}][\text{HYP}]?[\text{continuous tokens}] [\text{MASK}] [PRE][continuous tokens][HYP]?[continuous tokens][MASK]
大家可能想问,如何优化P-tuning?实际上根据标注数据量的多少,分两种情况讨论
就在P-Tuning方法提出不久后,Liu等人又提出了P-Tuning v2,主要解决P-Tuning的两个问题:
Liu等人认为先前的P-Tuning只用了一层BiLSTM来编码Pseudo token,这是其推理能力不足的原因之一,因此v2版本提出Deep Prompt Tuning,用Prefix Tuning中的深层模型替换BiLSTM,如下图所示
P-Tuning v2相比于P-Tuning,区别在于:
关于P-Tuning还有一些碎碎念,主要是从各个博客上看到的,汇总在这里。首先是v1版本的LSTM,实际上引入LSTM目的是为了帮助「模板」生成的token(某种程度上)更贴近自然语言,或者说token之间的语义更流畅,但更自然的方法应该是在训练下游任务的时候,不仅预测下游任务的目标token(例如"great"、“terrible”),还应该同时做其他token的预测
比如,如果是MLM模型,那么也随机MASK掉其它的一些token来预测,如果是LM模型,则预测完整的序列,而不单单是目标词。这样做的理由是:因为我们的MLM/LM都是经过自然语言预训练的,所以我们认为它能够很好的完成序列的重构,即便一开始不能,随着迭代轮数的增加,模型也能很好完成这项任务。所以这本质上是让模型进行「负重训练」
在标准的Fine-tune过程中(如上图b所示),新引入的参数量可能会很大(独立于原始预训练模型外的参数),例如基于RoBERTa-large的二分类任务会新引入2048个参数(nn.Linear(1024, 2
)),如果你仅有例如64个标注数据这样的小样本数据集,微调会非常困难
为解决这一问题,Prompt应运而生(如上图a所示),直接将下游任务转换为输出空间有限的MLM任务。值得注意的是:上述方法在预训练参数的基础上进行微调,并且没有引入任何新参数,同时还减少了微调和预训练任务之间的差距。总的来说,这可以更有效地用于小样本场景
尽管Prompt研究搞得如火如荼,但目前仍存在许多问题值得研究者们去探究
最后我还想提一个实际Code过程中存在的问题。我们知道MLM任务会输出句子中[MASK]位置最有可能的词,而Prompt也类似的,例如下面的例子
这是一条__新闻。中国足球出线的可能性只有0.001%,留给中国队的时间不多了
这是一个新闻分类问题,真实标签有"体育"、“财经”、“娱乐"等,上面的样本很明显是一条体育新闻,因此我们希望模型对[MASK]部分输出"体育”,但事实真的如此吗?实际情况模型的输出可能是"足球",但你认为模型预测的"足球"有问题吗?好像也没啥毛病,因此这就引申出了Prompt的一个问题,是否应该限制模型的输出空间?
还是上面新闻分类的例子,我们是否应该限制模型输出的空间,让他固定只能预测"体育"、“财经”、"娱乐"这几个标签?或者我们干脆把这几个标签换成索引,那就是让模型从0,1,2这三个数字选一个。Wait Wait Wait,如果这么做的话,和Fine-Tune有什么区别,Fine-Tune也是把标签转换成索引,让模型看了句子之后,从这几个索引中选一个作为预测值
这么说的话,那我们就不应该限制模型的输出空间,可是这样的话[MASK]位置的输出就限制的太死了,必须一定是"good"、“财经"才算对,如果输出"nice”、“财政"就算错。实际上输出近义词或者相似词,在零样本的情况下会经常出现,但是如果你用一些有标签的样本去训练,模型自己就会慢慢固定输出空间。例如"财经”,它不会预测成"财政",只会预测成其它类型的新闻,例如"体育"
REFERENCES