0
  • 聊天消息
  • 系統(tǒng)消息
  • 評論與回復(fù)
登錄后你可以
  • 下載海量資料
  • 學(xué)習(xí)在線課程
  • 觀看技術(shù)視頻
  • 寫文章/發(fā)帖/加入社區(qū)
會員中心
創(chuàng)作中心

完善資料讓更多小伙伴認(rèn)識你,還能領(lǐng)取20積分哦,立即完善>

3天內(nèi)不再提示

基于RoBERTa的語義理解模型的構(gòu)建

深度學(xué)習(xí)自然語言處理 ? 來源:美團(tuán)技術(shù)團(tuán)隊 ? 作者:駱穎,徐俊,謝睿 ? 2022-07-08 14:38 ? 次閱讀

美團(tuán)技術(shù)團(tuán)隊 近日,美團(tuán)搜索與NLP部NLP中心語義理解團(tuán)隊的小樣本學(xué)習(xí)模型FSL++在中文小樣本語言理解權(quán)威評測基準(zhǔn)FewCLUE榜單登頂,在自然語言推理(OCNLI)單任務(wù)中取得第一,并在極少數(shù)樣本(一個類別僅100余個)的條件下,在新聞分類(TNEWS)、科學(xué)文獻(xiàn)學(xué)科分類(CSLDCP)任務(wù)上超過了人類識別精確度。

1 概述

2 方法介紹

2.1 增強(qiáng)預(yù)訓(xùn)練

2.2 模型結(jié)構(gòu)

2.3 數(shù)據(jù)增強(qiáng)

2.4 集成學(xué)習(xí)&自訓(xùn)練

3 實驗結(jié)果

3.1 數(shù)據(jù)集介紹

3.2 實驗對比

4 小樣本學(xué)習(xí)策略在美團(tuán)場景的應(yīng)用

5 總結(jié)

1 概述

CLUE(Chinese Language Understanding Evaluation)[1]是中文語言理解權(quán)威測評榜單,包含了文本分類、句間關(guān)系、閱讀理解等眾多語義分析和語義理解類子任務(wù),對學(xué)術(shù)界和工業(yè)界都產(chǎn)生了較大的影響。

8ba72012-e8a5-11ec-ba43-dac502259ad0.png

圖1 FewCLUE榜單(截止到2022-04-18)

FewCLUE[2,3]是CLUE中專門用于中文小樣本學(xué)習(xí)評測的一個子榜,旨在結(jié)合預(yù)訓(xùn)練語言模型通用和強(qiáng)大的泛化能力,探索小樣本學(xué)習(xí)最佳模型和在中文上的實踐。FewCLUE的部分?jǐn)?shù)據(jù)集只有一百多條有標(biāo)簽樣本,可以衡量模型在極少有標(biāo)簽樣本下的泛化性能,發(fā)布后吸引了包括網(wǎng)易、微信AI、阿里巴巴、IDEA研究院、浪潮人工智能研究院等多家企業(yè)與研究院的參與。不久前,美團(tuán)平臺搜索與NLP部NLP中心語義理解團(tuán)隊的小樣本學(xué)習(xí)模型FSL++以優(yōu)越的性能在FewCLUE榜單上取得第一名,達(dá)到SOTA水平。

2 方法介紹

大規(guī)模預(yù)訓(xùn)練模型雖然在各大任務(wù)里面取得非常好的效果,但是在特定的任務(wù)上,還是需要許多標(biāo)注數(shù)據(jù)。美團(tuán)的各個業(yè)務(wù)中,有著豐富的NLP場景,往往需要較高的人工標(biāo)注成本。在業(yè)務(wù)發(fā)展早期或者新的業(yè)務(wù)需求需要快速上線時,往往會出現(xiàn)標(biāo)注樣本不足的現(xiàn)象,使用傳統(tǒng)Pretrain(預(yù)訓(xùn)練)+ Fine-Tune(微調(diào))的深度學(xué)習(xí)訓(xùn)練方法往往達(dá)不到理想的指標(biāo)要求,因此研究小樣本場景的模型訓(xùn)練問題就變得非常必要。

本文提出了一套大模型 + 小樣本的聯(lián)合訓(xùn)練方案FSL++,綜合了模型結(jié)構(gòu)優(yōu)選、大規(guī)模預(yù)訓(xùn)練、樣本增強(qiáng)、集成學(xué)習(xí)以及自訓(xùn)練等模型優(yōu)化策略,最終在中文語言理解權(quán)威評測基準(zhǔn)下的FewCLUE榜單取得了優(yōu)異的成績,并且在部分任務(wù)上性能超過了人類水平,而在部分任務(wù)上(如CLUEWSC)還有一定的提升空間。

FewCLUE發(fā)布后,網(wǎng)易伏羲使用自研的EET模型[4],并通過二次訓(xùn)練增強(qiáng)模型的語義理解能力,再加入模版進(jìn)行多任務(wù)學(xué)習(xí);IDEA研究院的二郎神模型[5]在BERT模型的基礎(chǔ)上使用更先進(jìn)的預(yù)訓(xùn)練技術(shù)訓(xùn)練大模型,在下游任務(wù)微調(diào)的過程中用加入動態(tài)Mask策略的Masked Language Model(MLM)作為輔助任務(wù)。這些方法都使用Prompt Learning作為基本的任務(wù)架構(gòu),跟這些自研的大模型相比,我們的方法主要在Prompt Learning框架的基礎(chǔ)上加入了樣本增強(qiáng)、集成學(xué)習(xí)以及自學(xué)習(xí)等模型優(yōu)化策略,極大地提高模型的任務(wù)表現(xiàn)和魯棒性,同時這套方法可以適用于各種預(yù)訓(xùn)練模型,更加靈活便捷。

FSL++整體模型結(jié)構(gòu)如下圖2所示。FewCLUE數(shù)據(jù)集為每個任務(wù)提供160條有標(biāo)簽數(shù)據(jù)以及接近兩萬條無標(biāo)簽數(shù)據(jù)。本次FewCLUE實踐中,我們先在Fine-Tune階段構(gòu)造多模板Prompt Learning,并對有標(biāo)簽數(shù)據(jù)采用對抗訓(xùn)練、對比學(xué)習(xí)、Mixup等增強(qiáng)策略。由于這些數(shù)據(jù)增強(qiáng)策略采用不同的增強(qiáng)原理,可以認(rèn)為這些模型之間差異性比較顯著,經(jīng)過集成學(xué)習(xí)之后會有比較好的效果。所以在采用數(shù)據(jù)增強(qiáng)策略進(jìn)行訓(xùn)練以后,我們擁有了多個弱監(jiān)督模型,并且用這些弱監(jiān)督模型在無標(biāo)簽數(shù)據(jù)上進(jìn)行預(yù)測,得到無標(biāo)簽數(shù)據(jù)的偽標(biāo)簽分布。之后,我們將多個經(jīng)過不同的數(shù)據(jù)增強(qiáng)模型預(yù)測得到的無標(biāo)簽數(shù)據(jù)的偽標(biāo)簽分布整合起來,得到一份總的無標(biāo)簽數(shù)據(jù)的偽標(biāo)簽分布,接著重新構(gòu)造多模板Prompt Learning,并再次使用數(shù)據(jù)增強(qiáng)策略,選擇最優(yōu)策略。目前,我們的實驗只進(jìn)行一輪迭代,也可以嘗試多輪迭代,不過隨著迭代次數(shù)增加,提升也不再明顯。

8be43c72-e8a5-11ec-ba43-dac502259ad0.png

圖2 FSL++模型框架

2.1 增強(qiáng)預(yù)訓(xùn)練

預(yù)訓(xùn)練語言模型是在龐大的無標(biāo)簽語料庫上進(jìn)行訓(xùn)練的。例如,RoBERTa[6]在160GB以上的文本進(jìn)行訓(xùn)練,包括百科全書、新聞文章、文學(xué)作品和Web內(nèi)容。通過這些模型學(xué)習(xí)到的表示,在包含多種來源的各種大小的數(shù)據(jù)集的任務(wù)中實現(xiàn)出色的性能。

FSL++模型使用RoBERTa-large模型作為基礎(chǔ)模型,并且采用融入領(lǐng)域知識的Domain-Adaptive Pretraining (DAPT)[7]預(yù)訓(xùn)練方法和融入任務(wù)知識的Task-Adaptive Pretraining (TAPT)[7]。DAPT旨在預(yù)訓(xùn)練模型的基礎(chǔ)上,增加大量領(lǐng)域內(nèi)無標(biāo)簽文本繼續(xù)訓(xùn)練語言模型,之后再在指定任務(wù)的數(shù)據(jù)集上進(jìn)行微調(diào)。

對目標(biāo)文本領(lǐng)域進(jìn)行繼續(xù)預(yù)訓(xùn)練,可以提高語言模型的性能,尤其是在與目標(biāo)文本領(lǐng)域相關(guān)的下游任務(wù)上的性能。并且,預(yù)訓(xùn)練文本與任務(wù)領(lǐng)域的相關(guān)度越高,帶來的提升越大。在本次實踐中,我們最終使用了在100G包含娛樂節(jié)目、體育、健康、國際事務(wù)、電影、名人等各個領(lǐng)域的語料的CLUE Vocab[8]上預(yù)訓(xùn)練得到的RoBERTa Large模型。TAPT指在預(yù)訓(xùn)練模型的基礎(chǔ)上,增加數(shù)量較少但與任務(wù)直接相關(guān)的無標(biāo)簽語料進(jìn)行預(yù)訓(xùn)練。針對TAPT任務(wù),我們選擇使用的預(yù)訓(xùn)練數(shù)據(jù)是FewCLUE榜單為每個任務(wù)提供的無標(biāo)簽數(shù)據(jù)。

除此之外,在針對句間關(guān)系任務(wù),如中文自然語言推理任務(wù)OCNLI、中文對話短文本匹配任務(wù)BUSTM的實踐中,我們使用在其他句間關(guān)系任務(wù)如中文自然語言推理數(shù)據(jù)集CMNLI、中文短文本相似度數(shù)據(jù)集LCQMC上進(jìn)行預(yù)訓(xùn)練的模型參數(shù)作為初始參數(shù),相比直接用原始模型完成任務(wù),也能提升一定的效果。

2.2 模型結(jié)構(gòu)

FewCLUE包含多種任務(wù)形式,我們?yōu)槊糠N任務(wù)選擇了合適的模型結(jié)構(gòu)。文本分類任務(wù)和機(jī)器閱讀理解(MRC)任務(wù)本身的類別詞就攜帶了信息,因此更適合建模為Masked Language Model(MLM)形式;而句間關(guān)系任務(wù)判斷兩個句子的相關(guān)性,更類似于Next Sentence Prediction(NSP)[9]任務(wù)形式。因此,我們?yōu)榉诸惾蝿?wù)和閱讀理解任務(wù)選擇PET[10]模型,為句間關(guān)系任務(wù)選擇EFL[11]模型,EFL方法可以通過全局采樣構(gòu)造負(fù)樣本,學(xué)習(xí)到更魯棒的分類器。

2.2.1 Prompt Learning

Prompt Learning的主要目標(biāo)是盡可能減小預(yù)訓(xùn)練目標(biāo)與下游微調(diào)目標(biāo)的差距。通?,F(xiàn)有的預(yù)訓(xùn)練任務(wù)均包含MLM損失函數(shù),但是下游的任務(wù)則并未采用MLM,而是引入新的分類器,使得預(yù)訓(xùn)練任務(wù)和下游任務(wù)出現(xiàn)了不一致。Prompt Learning不引入額外的分類器或其他參數(shù),而是通過拼接模板(Template,即為輸入數(shù)據(jù)拼接語言片段,從而改造任務(wù)為MLM形式)和標(biāo)簽詞映射(Verbalizer,即為每個標(biāo)簽在詞表中找到對應(yīng)的詞,從而為MLM任務(wù)設(shè)定預(yù)測目標(biāo)),使得模型可以在少量樣本的條件下在下游任務(wù)中使用。

8c32bfaa-e8a5-11ec-ba43-dac502259ad0.png

圖3 Prompt Learning方法完成情感分析任務(wù)的流程圖

以圖3展示的電商評價情感分析任務(wù)EPRSTMT為例。給定文本“這個電影真不錯,值得第二次觀看!”,傳統(tǒng)的文本分類則是在CLS部分的Embedding接上分類器,并映射到0-1分類上(0:負(fù)向,1:正向)。這種方法在小樣本場景下需要訓(xùn)練新的分類器,比較難獲得好的效果。而基于Prompt Learning的方法則是創(chuàng)建模板“這是一條 [MASK] 評?!保賹⒛0迮c原文進(jìn)行拼接,訓(xùn)練時通過語言模型預(yù)測[MASK]位置的詞,再將其映射到對應(yīng)的類別上(好:正向, 差:負(fù)向)。

由于缺乏足夠數(shù)據(jù),有時很難確定表現(xiàn)最好的模板和標(biāo)簽詞映射。因此,也可以采用多模板與多標(biāo)簽詞映射的設(shè)計。通過設(shè)計多個模板,最終的結(jié)果采用多個模板的結(jié)果的整合,或設(shè)計一對多的標(biāo)簽詞映射,讓一個標(biāo)簽對應(yīng)多個詞。同上述例子,可以設(shè)計如下模板組合(左:同一個句子的多模板,右:多標(biāo)簽映射)。

8c889e3e-e8a5-11ec-ba43-dac502259ad0.png

圖4 PET多模板與多標(biāo)簽映射圖

任務(wù)樣例

8cbb95d2-e8a5-11ec-ba43-dac502259ad0.png

表1 FewCLUE數(shù)據(jù)集中PET模板構(gòu)建

2.2.2 EFL

EFL模型將兩個句子拼接在一起,用輸出層的[CLS]位置處的Embedding后接一個分類器完成預(yù)測。EFL的訓(xùn)練過程中,除了訓(xùn)練集的樣本,還會進(jìn)行負(fù)樣本構(gòu)造,訓(xùn)練過程中,在每個Batch里隨機(jī)選擇其他數(shù)據(jù)中的句子作為負(fù)樣本,通過構(gòu)造負(fù)樣本進(jìn)行數(shù)據(jù)增強(qiáng)。雖然EFL模型需要訓(xùn)練新的分類器,但目前有很多公開的文本蘊(yùn)含/句間關(guān)系數(shù)據(jù)集,如CMNLI、LCQMC等,可以通過在這些樣本上進(jìn)行持續(xù)學(xué)習(xí)(continue-train),再將學(xué)習(xí)到的參數(shù)遷移到小樣本場景中,用FewCLUE的任務(wù)數(shù)據(jù)集進(jìn)行進(jìn)一步微調(diào)。

任務(wù)樣例

8cda3816-e8a5-11ec-ba43-dac502259ad0.png

表2 FewCLUE數(shù)據(jù)集中EFL模板構(gòu)建

2.3 數(shù)據(jù)增強(qiáng)

數(shù)據(jù)增強(qiáng)方法主要有樣本增強(qiáng)和Embedding增強(qiáng)。NLP領(lǐng)域中,數(shù)據(jù)增強(qiáng)的目的是在不改變語義的前提下擴(kuò)充文本數(shù)據(jù)。主要的方法包括簡單文本替換、使用語言模型生成相似句子等,我們嘗試過EDA等擴(kuò)充文本數(shù)據(jù)的方法,但是一個詞的變化就可能導(dǎo)致整個句子的意思發(fā)生翻轉(zhuǎn),經(jīng)過替換的文本攜帶大量噪音,所以很難用簡單的規(guī)則樣本變化產(chǎn)生足夠的增強(qiáng)數(shù)據(jù)。而Embedding增強(qiáng),則不再對輸入進(jìn)行操作,轉(zhuǎn)而在Embedding層面進(jìn)行操作,可以通過對Embedding增加擾動或者插值等方式提升模型的魯棒性。

因此,本次實踐中我們主要進(jìn)行Embedding增強(qiáng)。我們用的數(shù)據(jù)增強(qiáng)策略分別有Mixup[12]、Manifold-Mixup[13]、對抗訓(xùn)練(Adversarial training, AT) [14]和對比學(xué)習(xí)R-drop[15]。數(shù)據(jù)增強(qiáng)策略的詳細(xì)介紹見之前的技術(shù)博客小樣本學(xué)習(xí)及其在美團(tuán)場景中的應(yīng)用。

8cffd882-e8a5-11ec-ba43-dac502259ad0.png

表3 數(shù)據(jù)增強(qiáng)策略簡述

Mixup通過對輸入數(shù)據(jù)進(jìn)行簡單的線性變換,構(gòu)造新的組合樣本和組合標(biāo)簽,可以增強(qiáng)模型的泛化能力。在各種有監(jiān)督任務(wù)或者半監(jiān)督任務(wù)上,使用Mixup都能極大提高模型的泛化能力。Mixup方法可以視為正則化操作,它要求模型在特征層面生成的組合特征滿足線性約束,并且利用這種約束對模型施加正則化。直觀來看,當(dāng)模型的輸入為另外兩個輸入的線性組合時,其輸出也是這兩個數(shù)據(jù)單獨(dú)輸入模型后所得輸出的線性組合,其實就是要求模型近似為一個線性系統(tǒng)。

Manifold Mixup將上述的Mixup操作泛化到特征上。因為特征具有更高階的語義信息,所以在其維度上插值可能會產(chǎn)生更有意義的樣本。在類似于BERT[9]、RoBERTa[6]的模型中,隨機(jī)選擇層數(shù)k,對該層的特征表示進(jìn)行Mixup插值。普通的Mixup的插值發(fā)生在輸出層Embedding部分,而Manifold Mixup相當(dāng)于把這一系列插值操作加入到語言模型內(nèi)部的Transformers結(jié)構(gòu)的隨機(jī)某層中。

對抗訓(xùn)練通過在輸入樣本上增加微小的擾動來顯著提高模型Loss。對抗訓(xùn)練就是訓(xùn)練一個能有效識別原始樣本和對抗樣本的模型?;驹砭褪峭ㄟ^添加擾動構(gòu)造一些對抗樣本,交給模型去訓(xùn)練,提高模型在遇到對抗樣本時的魯棒性,同時也能提高模型的表現(xiàn)和泛化能力。對抗樣本需要具有兩個特點,分別是:

相對于原始輸入,所添加的擾動是微小的。

能使模型犯錯。對抗訓(xùn)練有兩個作用,分別是提高模型對惡意攻擊的魯棒性和提高模型的泛化能力。

R-Drop對同一個句子做兩次Dropout,并且強(qiáng)制由Dropout生成的不同子模型的輸出概率保持一致。Dropout的引入雖然效果很好,但是它會導(dǎo)致訓(xùn)練和推理過程的不一致性問題。為緩解這種訓(xùn)練推理過程的不一致性,R-Drop對Dropout進(jìn)行正則化處理,在兩個子模型產(chǎn)生的輸出中增加對輸出數(shù)據(jù)分布的限制,引入數(shù)據(jù)分布度量的KL散度損失,使得Batch內(nèi)同一個樣本生成的兩個數(shù)據(jù)分布盡量接近,具有分布一致性。具體來說,對于每個訓(xùn)練樣本,R-Drop最小化了由不同Dropout生成的子模型的輸出概率之間的KL 散度。R-Drop作為一種訓(xùn)練思想,可以用到大部分有監(jiān)督或半監(jiān)督的訓(xùn)練中,通用性強(qiáng)。

我們使用的三種數(shù)據(jù)增強(qiáng)策略,Mixup是在語言模型的輸出層Embedding和語言模型的內(nèi)部隨機(jī)某層Transformers的輸出層中做兩個樣本的線性變化,對抗訓(xùn)練是在樣本上增加微小的擾動,而對比學(xué)習(xí)是對同一個句子做兩次Dropout形成正樣本對,再用KL散度限制兩個子模型保持一致。三種策略都是通過在Embedding完成一些操作來增強(qiáng)模型的泛化性,經(jīng)過不同策略得到的模型分別都具有不同的偏好,這就為下一步的集成學(xué)習(xí)提供了條件。

2.4 集成學(xué)習(xí)&自訓(xùn)練

集成學(xué)習(xí)可以組合多個弱監(jiān)督模型,以期得到一個更好更全面的強(qiáng)監(jiān)督模型。集成學(xué)習(xí)潛在的思想是即便某一個弱分類器得到了錯誤的預(yù)測,其他的弱分類器也可以將錯誤糾正回來。如果待組合的各個模型之間差異性比較顯著,那么集成學(xué)習(xí)之后通常會有一個較好的結(jié)果。

自訓(xùn)練使用少量的標(biāo)記數(shù)據(jù)和大量的未標(biāo)記數(shù)據(jù)對模型進(jìn)行聯(lián)合訓(xùn)練,首先使用經(jīng)過訓(xùn)練的分類器來預(yù)測所有未標(biāo)記數(shù)據(jù)的標(biāo)簽,然后選擇置信度較高的標(biāo)簽作為偽標(biāo)簽數(shù)據(jù),將偽標(biāo)記數(shù)據(jù)與人工標(biāo)記的訓(xùn)練數(shù)據(jù)聯(lián)合起來重新訓(xùn)練分類器。

集成學(xué)習(xí)+自訓(xùn)練是一套可以利用多個模型以及無標(biāo)簽數(shù)據(jù)的方案。這其中,集成學(xué)習(xí)的一般步驟為:訓(xùn)練多個不同的弱監(jiān)督模型,分別用每個模型預(yù)測無標(biāo)簽數(shù)據(jù)的標(biāo)簽概率分布,計算標(biāo)簽概率分布的加權(quán)和,得到無標(biāo)簽數(shù)據(jù)的偽標(biāo)簽概率分布。自訓(xùn)練指訓(xùn)練一個模型用于組合其他各個模型,其一般步驟為:訓(xùn)練多個Teacher模型,Student模型學(xué)習(xí)偽標(biāo)簽概率分布中高置信度樣本的Soft Prediction,Student模型作為最后的強(qiáng)學(xué)習(xí)器。

8d21daea-e8a5-11ec-ba43-dac502259ad0.png

圖5 集成學(xué)習(xí)+自訓(xùn)練結(jié)構(gòu)

在本次FewCLUE實踐中,我們先在Fine-Tune階段構(gòu)造多模板Prompt Learning,并對有標(biāo)注數(shù)據(jù)采用對抗訓(xùn)練、對比學(xué)習(xí)、Mixup等增強(qiáng)策略。由于這些數(shù)據(jù)增強(qiáng)策略采用不同的增強(qiáng)原理,可以認(rèn)為這些模型之間差異性比較顯著,經(jīng)過集成學(xué)習(xí)之后會有比較好的效果。

在采用數(shù)據(jù)增強(qiáng)策略進(jìn)行訓(xùn)練以后,我們擁有了多個弱監(jiān)督模型,并且用這些弱監(jiān)督模型在無標(biāo)簽數(shù)據(jù)上進(jìn)行預(yù)測,得到無標(biāo)簽數(shù)據(jù)的偽標(biāo)簽分布。之后,我們將多個經(jīng)過不同的數(shù)據(jù)增強(qiáng)模型預(yù)測得到的無標(biāo)簽數(shù)據(jù)的偽標(biāo)簽分布整合起來,得到一份總的無標(biāo)簽數(shù)據(jù)的偽標(biāo)簽分布。篩選偽標(biāo)簽數(shù)據(jù)的過程中,我們不一定會選擇置信度最高的樣本,因為如果每個數(shù)據(jù)增強(qiáng)模型給出的置信度都很高,說明這個樣本可能是容易學(xué)習(xí)的樣本,不一定有很大價值。

我們綜合多個數(shù)據(jù)增強(qiáng)模型給出的置信度,盡量選擇置信度較高,但是又不容易學(xué)習(xí)的樣本(比如多個模型預(yù)測不全部一致)。接著用標(biāo)注數(shù)據(jù)和偽標(biāo)注數(shù)據(jù)的集合重新構(gòu)造多模板Prompt Learning,再次使用數(shù)據(jù)增強(qiáng)策略,并選擇最好的策略。目前,我們的實驗?zāi)壳爸贿M(jìn)行一輪迭代,也可以嘗試多輪迭代,不過隨著迭代次數(shù)增加,提升也會減少,不再顯著。

3 實驗結(jié)果

3.1 數(shù)據(jù)集介紹

FewCLUE榜單提供了9個任務(wù),其中分別為4個文本分類任務(wù),2個句間關(guān)系任務(wù)和3個閱讀理解任務(wù)。文本分類任務(wù)有電商評價情感分析、科學(xué)文獻(xiàn)分類、新聞分類和App應(yīng)用描述主題分類任務(wù)。主要?dú)w類為短文本二分類、短文本多分類和長文本多分類。其中有的任務(wù)類別眾多,超過100類,并且出現(xiàn)了類別不均衡問題。句間關(guān)系任務(wù)有自然語言推理和短文本匹配任務(wù)。閱讀理解任務(wù)則有成語閱讀理解選擇填空,摘要判斷關(guān)鍵詞判別和代詞消歧任務(wù)。每個任務(wù)大體提供了160條有標(biāo)簽數(shù)據(jù)和兩萬條左右的無標(biāo)簽數(shù)據(jù)。因為長文本分類任務(wù)類別眾多,過于困難,也提供了更多的有標(biāo)簽數(shù)據(jù)。詳細(xì)的任務(wù)數(shù)據(jù)情況如表4所示:

8d426c7e-e8a5-11ec-ba43-dac502259ad0.png

表4 FewCLUE數(shù)據(jù)集任務(wù)介紹

3.2 實驗對比

表5展示了不同模型和參數(shù)量的實驗結(jié)果的對比。在RoBERTa Base實驗中,使用PET/EFL模型會超過傳統(tǒng)的直接Fine-Tune模型結(jié)果2-28PP。以PET/EFL模型為基礎(chǔ),為了探索大模型在小樣本場景中的效果,我們在RoBERTa Large上進(jìn)行了實驗,相對于RoBERTa Base,大模型可以提升模型0.5-13PP;為了更好地利用領(lǐng)域知識,我們進(jìn)一步在經(jīng)過CLUE數(shù)據(jù)集上增強(qiáng)預(yù)訓(xùn)練的RoBERTa Large Clue模型上進(jìn)行實驗,融入了領(lǐng)域知識的大模型進(jìn)一步提升結(jié)果0.1-9pp。基于此,在之后的實驗中,我們都在RoBERTa Large Clue上進(jìn)行實驗。

8d6fff68-e8a5-11ec-ba43-dac502259ad0.png

表5 不同模型和參數(shù)量的實驗結(jié)果對比 (加粗紅色字體表示最好的結(jié)果)

表6展示了在PET/EFL模型上進(jìn)行數(shù)據(jù)增強(qiáng)和集成學(xué)習(xí)實驗結(jié)果, 可以發(fā)現(xiàn)即使是在大模型上使用數(shù)據(jù)增強(qiáng)策略,模型也能帶來0.8-9PP的提升,而進(jìn)一步進(jìn)行集成學(xué)習(xí)&自訓(xùn)練以后,模型表現(xiàn)會繼續(xù)提升0.4-4PP。

8dcb1e0c-e8a5-11ec-ba43-dac502259ad0.png

表6 基礎(chǔ)模型+數(shù)據(jù)增強(qiáng)+集成學(xué)習(xí)實驗效果 (加粗紅色字體表示最好的結(jié)果)

其中集成學(xué)習(xí)+自訓(xùn)練步驟中,我們嘗試了幾種篩選策略:

選擇置信度最高的樣本,這種策略帶來的提升在1PP以內(nèi),置信度最高的偽標(biāo)簽樣本中很多是多個模型預(yù)測一致且置信度都比較高的樣本,這部分樣本比較容易學(xué)習(xí),融入這部分樣本帶來的收益有限。

選擇置信度高且具有爭議性的樣本(存在至少一個模型和其他模型預(yù)測結(jié)果不一致,但多個模型總體置信度超過閾值1),這種策略規(guī)避了特別容易學(xué)習(xí)的樣本,又通過設(shè)置閾值避免帶來過多臟數(shù)據(jù),可以帶來0-3PP的提升;

融合上面的兩種策略,若多個模型對于一個樣本的預(yù)測結(jié)果是一致的,我們選擇置信度小于閾值2的樣本;對于存在至少一個模型和其他模型預(yù)測結(jié)果不一致的,我們選擇置信度大于閾值3的樣本。這種方式同時選擇了置信度較高的樣本保證輸出的可信度,又選擇了較有爭議的樣本保證篩選出來的偽標(biāo)簽樣本具有較大學(xué)習(xí)難度,可以帶來0.4-4PP的提升。

4 小樣本學(xué)習(xí)策略在美團(tuán)場景的應(yīng)用

在美團(tuán)的各個業(yè)務(wù)中,有著豐富的NLP場景,部分任務(wù)可以歸類為文本分類任務(wù)和句間關(guān)系任務(wù),以上提到的小樣本學(xué)習(xí)策略已經(jīng)應(yīng)用于美團(tuán)點評的各種場景,期望在數(shù)據(jù)資源稀少的情況下訓(xùn)練出比較好的模型。此外,小樣本學(xué)習(xí)策略已經(jīng)廣泛應(yīng)用于美團(tuán)內(nèi)部自然語言處理(NLP)平臺的各個NLP算法能力中,在眾多業(yè)務(wù)場景下落地并取得顯著收益,美團(tuán)內(nèi)部的工程師可通過該平臺來體驗NLP中心相關(guān)的能力。

文本分類任務(wù)

醫(yī)美題材分類:對美團(tuán)和點評的筆記內(nèi)容按題材分為8類:獵奇、探店、測評、真人案例、治療過程、避坑、效果對比、科普。用戶點擊某一種題材時,返回對應(yīng)的筆記內(nèi)容,上線至美團(tuán)和點評App醫(yī)療美容頻道的百科頁、方案頁經(jīng)驗分享,小樣本學(xué)習(xí)利用2,989條訓(xùn)練數(shù)據(jù)準(zhǔn)確率提升1.8PP,達(dá)到了89.24%。

攻略識別:從UGC和筆記中挖掘旅游攻略,提供旅游攻略的內(nèi)容供給,應(yīng)用于景點精搜下的攻略模塊,召回內(nèi)容為描述旅游攻略的筆記,小樣本學(xué)習(xí)利用384條訓(xùn)練數(shù)據(jù)準(zhǔn)確率提升2PP,達(dá)到了87%。

學(xué)城文本分類:學(xué)城(美團(tuán)內(nèi)部知識庫)有大量的用戶文本,經(jīng)歸納將文本分為17種類別,已有模型在700條數(shù)據(jù)上訓(xùn)練,通過小樣本學(xué)習(xí),在已有模型上提升模型精度2.5PP,達(dá)到84%。

項目篩選:LE生活服務(wù)/麗人等業(yè)務(wù)目前的評價列表頁混排評價的方式不便讓用戶快速找到?jīng)Q策信息,因此需要更有結(jié)構(gòu)化的分類標(biāo)簽來滿足用戶的需求,小樣本學(xué)習(xí)在這兩個業(yè)務(wù)上利用300-500條數(shù)據(jù)上準(zhǔn)確率均達(dá)到95%+(多個數(shù)據(jù)集分別提升1.5-4PP)。

句間關(guān)系任務(wù)

醫(yī)美功效打標(biāo):對美團(tuán)和大眾點評的筆記內(nèi)容按功效進(jìn)行召回,功效的類型有:補(bǔ)水、美白、瘦臉、除皺等,上線至醫(yī)美頻道頁,有110種功效類型需要打標(biāo),小樣本學(xué)習(xí)僅用2909條訓(xùn)練數(shù)據(jù)準(zhǔn)確率達(dá)到了91.88%(提升2.8PP)。

醫(yī)美品牌打標(biāo):品牌上游企業(yè)有針對旗下產(chǎn)品進(jìn)行品牌宣傳和營銷的訴求,而內(nèi)容營銷是當(dāng)前主流、有效的營銷方式之一。品牌打標(biāo)就是為每種品牌如“伊膚泉”、“術(shù)唯可”召回詳細(xì)介紹該品牌的筆記內(nèi)容,共有103種品牌,已上線至醫(yī)美品牌館,小樣本學(xué)習(xí)僅用1676條訓(xùn)練數(shù)據(jù)準(zhǔn)確率達(dá)到了88.59%(提升2.9PP)。

5 總結(jié)

在本次榜單提交中,我們構(gòu)建了一種基于RoBERTa的語義理解模型,并通過增強(qiáng)預(yù)訓(xùn)練、PET/EFL模型、數(shù)據(jù)增強(qiáng)和集成學(xué)習(xí)&自訓(xùn)練來提升模型的效果。該模型能完成文本分類、句間關(guān)系推理任務(wù)和幾種閱讀理解任務(wù)。

通過參加本次測評任務(wù),我們對小樣本場景下的自然語言理解領(lǐng)域的算法和研究有了更深的認(rèn)識,也借此對前沿算法的中文落地能力進(jìn)行了摸底測試,為后續(xù)進(jìn)一步算法研究、算法落地打下了基礎(chǔ)。此外,本次數(shù)據(jù)集中的任務(wù)場景與美團(tuán)搜索與NLP部的業(yè)務(wù)場景存在很大相似性,該模型的很多策略也直接應(yīng)用在實際業(yè)務(wù)中,直接為業(yè)務(wù)賦能。

審核編輯:郭婷

聲明:本文內(nèi)容及配圖由入駐作者撰寫或者入駐合作網(wǎng)站授權(quán)轉(zhuǎn)載。文章觀點僅代表作者本人,不代表電子發(fā)燒友網(wǎng)立場。文章及其配圖僅供工程師學(xué)習(xí)之用,如有內(nèi)容侵權(quán)或者其他違規(guī)問題,請聯(lián)系本站處理。 舉報投訴
  • 數(shù)據(jù)集
    +關(guān)注

    關(guān)注

    4

    文章

    1201

    瀏覽量

    24622
  • nlp
    nlp
    +關(guān)注

    關(guān)注

    1

    文章

    486

    瀏覽量

    21991

原文標(biāo)題:美團(tuán)獲得小樣本學(xué)習(xí)榜單FewCLUE第一!Prompt Learning+自訓(xùn)練實戰(zhàn)

文章出處:【微信號:zenRRan,微信公眾號:深度學(xué)習(xí)自然語言處理】歡迎添加關(guān)注!文章轉(zhuǎn)載請注明出處。

收藏 人收藏

    評論

    相關(guān)推薦

    如何設(shè)定機(jī)器人語義地圖的細(xì)粒度級別

    和區(qū)域與封閉的語義標(biāo)簽集對應(yīng)的工作。然而,封閉集檢測在能夠表示的概念集方面存在固有的限制,并且不能很好地處理自然語言的內(nèi)在歧義性和可變性。為了克服這些限制,一組新的方法開始利用視覺語言基礎(chǔ)模型進(jìn)行開放集語義理解。這
    的頭像 發(fā)表于 11-12 10:54 ?142次閱讀
    如何設(shè)定機(jī)器人<b class='flag-5'>語義</b>地圖的細(xì)粒度級別

    AI大模型在自然語言處理中的應(yīng)用

    AI大模型在自然語言處理(NLP)中的應(yīng)用廣泛且深入,其強(qiáng)大的語義理解和生成能力為NLP任務(wù)帶來了顯著的性能提升。以下是對AI大模型在NLP中應(yīng)用的介紹: 一、核心應(yīng)用 文本生成 AI大模型
    的頭像 發(fā)表于 10-23 14:38 ?279次閱讀

    【《大語言模型應(yīng)用指南》閱讀體驗】+ 基礎(chǔ)知識學(xué)習(xí)

    的信息,提供更全面的上下文理解。這使得模型能夠更準(zhǔn)確地理解復(fù)雜問題中的多個層面和隱含意義。 2. 語義分析 模型通過訓(xùn)練學(xué)習(xí)到語言的
    發(fā)表于 08-02 11:03

    PyTorch神經(jīng)網(wǎng)絡(luò)模型構(gòu)建過程

    PyTorch,作為一個廣泛使用的開源深度學(xué)習(xí)庫,提供了豐富的工具和模塊,幫助開發(fā)者構(gòu)建、訓(xùn)練和部署神經(jīng)網(wǎng)絡(luò)模型。在神經(jīng)網(wǎng)絡(luò)模型中,輸出層是尤為關(guān)鍵的部分,它負(fù)責(zé)將模型的預(yù)測結(jié)果以合適
    的頭像 發(fā)表于 07-10 14:57 ?424次閱讀

    圖像分割與語義分割中的CNN模型綜述

    圖像分割與語義分割是計算機(jī)視覺領(lǐng)域的重要任務(wù),旨在將圖像劃分為多個具有特定語義含義的區(qū)域或?qū)ο蟆>矸e神經(jīng)網(wǎng)絡(luò)(CNN)作為深度學(xué)習(xí)的一種核心模型,在圖像分割與語義分割中發(fā)揮著至關(guān)重要的
    的頭像 發(fā)表于 07-09 11:51 ?617次閱讀

    大語言模型(LLM)快速理解

    自2022年,ChatGPT發(fā)布之后,大語言模型(LargeLanguageModel),簡稱LLM掀起了一波狂潮。作為學(xué)習(xí)理解LLM的開始,先來整體理解一下大語言模型。一、發(fā)展歷史大
    的頭像 發(fā)表于 06-04 08:27 ?875次閱讀
    大語言<b class='flag-5'>模型</b>(LLM)快速<b class='flag-5'>理解</b>

    大語言模型:原理與工程時間+小白初識大語言模型

    解鎖 我理解的是基于深度學(xué)習(xí),需要訓(xùn)練各種數(shù)據(jù)知識最后生成自己的的語言理解和能力的交互模型。 對于常說的RNN是處理短序列的數(shù)據(jù)時表現(xiàn)出色,耳真正厲害的是Transformer,此框架被推出后直接
    發(fā)表于 05-12 23:57

    【大語言模型:原理與工程實踐】大語言模型的評測

    。評測任務(wù)涵蓋了詞義理解、語義理解、閱讀理解、拼音理解以及古文及諺語理解等五大類別。在詞義理解
    發(fā)表于 05-07 17:12

    【大語言模型:原理與工程實踐】大語言模型的基礎(chǔ)技術(shù)

    全面剖析大語言模型的核心技術(shù)與基礎(chǔ)知識。首先,概述自然語言的基本表示,這是理解大語言模型技術(shù)的前提。接著,詳細(xì)介紹自然語言處理預(yù)訓(xùn)練的經(jīng)典結(jié)構(gòu)Transformer,以及其工作原理,為構(gòu)建
    發(fā)表于 05-05 12:17

    【大語言模型:原理與工程實踐】揭開大語言模型的面紗

    ,大語言模型還具備出色的關(guān)聯(lián)和關(guān)系理解能力,能夠捕捉數(shù)據(jù)間的隱含關(guān)系和語義聯(lián)系,進(jìn)行高級關(guān)聯(lián)推理。它還支持多步推理,能在推理過程中進(jìn)行多個步驟的演繹和分析。最后,通過學(xué)習(xí)大規(guī)模數(shù)據(jù),大語言模型
    發(fā)表于 05-04 23:55

    昆侖萬維天工3.0大模型性能顯著提升,天工SkyMusic音樂模型亮相

    天工3.0擁有4000億參數(shù),超越Grok-1的3140億參數(shù),成為全球最大的開源MoE大模型。該模型語義理解、邏輯推理、通用性、泛化性、不確定性知識以及學(xué)習(xí)能力等方面均有顯著提升,數(shù)學(xué)/推理/代碼/文創(chuàng)能力提高超過30%。
    的頭像 發(fā)表于 04-17 16:05 ?564次閱讀

    昆侖萬維天工大模型3.0將于4月17日公測,參數(shù)量高達(dá)4000億

    對比前代“天工 2.0”MoE 大模型,“天工 3.0”展現(xiàn)出在模型語義理解、邏輯推理方面的“驚人”實力提升,以及在通用性、泛化性、不確定性知識和學(xué)習(xí)能力等多個領(lǐng)域的進(jìn)步。
    的頭像 發(fā)表于 04-02 10:10 ?780次閱讀

    華為云攜手樂聚機(jī)器人,探索人形機(jī)器人大模型開發(fā)

    根據(jù)雙方約定,他們還將共同設(shè)計并打造一款通用的具身智能解決方案,以構(gòu)建互通互聯(lián)的“人形機(jī)器人+”開放生態(tài)系統(tǒng)。據(jù)悉,華為云的先行者——盤古大模型,擁有卓越的語義理解、動態(tài)規(guī)劃以及多模態(tài)信號理解
    的頭像 發(fā)表于 03-22 11:53 ?669次閱讀

    百川智能發(fā)布超千億大模型Baichuan 3

    百川智能近日發(fā)布了超千億參數(shù)的大語言模型Baichuan 3,引發(fā)了業(yè)界的廣泛關(guān)注。這款模型在多個權(quán)威通用能力評測中表現(xiàn)卓越,展現(xiàn)了其強(qiáng)大的語義理解和生成能力。
    的頭像 發(fā)表于 01-31 14:58 ?776次閱讀

    AI 模型構(gòu)建的五個過程詳解

    AI 模型構(gòu)建的過程 模型構(gòu)建主要包括 5 個階段,分別為模型設(shè)計、特征工程、模型訓(xùn)練、
    發(fā)表于 11-17 10:31 ?7936次閱讀
    AI <b class='flag-5'>模型</b><b class='flag-5'>構(gòu)建</b>的五個過程詳解