改進(jìn)版的RoBERTa到底改進(jìn)了什么骇两?

在XLNet全面超越Bert后沒多久,F(xiàn)acebook提出了RoBERTa(a Robustly Optimized BERT Pretraining Approach)姜盈。再度在多個(gè)任務(wù)上達(dá)到SOTA低千。那么它到底改進(jìn)了什么?它在模型層面沒有改變Google的Bert馏颂,改變的只是預(yù)訓(xùn)練的方法示血。

1. 靜態(tài)Masking vs 動(dòng)態(tài)Masking

原來Bert對(duì)每一個(gè)序列隨機(jī)選擇15%的Tokens替換成[MASK],為了消除與下游任務(wù)的不匹配救拉,還對(duì)這15%的Tokens進(jìn)行(1)80%的時(shí)間替換成[MASK]难审;(2)10%的時(shí)間不變;(3)10%的時(shí)間替換成其他詞亿絮。但整個(gè)訓(xùn)練過程告喊,這15%的Tokens一旦被選擇就不再改變,也就是說從一開始隨機(jī)選擇了這15%的Tokens派昧,之后的N個(gè)epoch里都不再改變了黔姜。這就叫做靜態(tài)Masking。

而RoBERTa一開始把預(yù)訓(xùn)練的數(shù)據(jù)復(fù)制10份蒂萎,每一份都隨機(jī)選擇15%的Tokens進(jìn)行Masking秆吵,也就是說,同樣的一句話有10種不同的mask方式五慈。然后每份數(shù)據(jù)都訓(xùn)練N/10個(gè)epoch纳寂。這就相當(dāng)于在這N個(gè)epoch的訓(xùn)練中,每個(gè)序列的被mask的tokens是會(huì)變化的泻拦。這就叫做動(dòng)態(tài)Masking烈疚。

那么這樣改變是否真的有效果?作者在只將靜態(tài)Masking改成動(dòng)態(tài)Masking聪轿,其他參數(shù)不變的情況下做了實(shí)驗(yàn),動(dòng)態(tài)Masking確實(shí)能提高性能猾浦。

2. with NSP vs without NSP
原本的Bert為了捕捉句子之間的關(guān)系陆错,使用了NSP任務(wù)進(jìn)行預(yù)訓(xùn)練,就是輸入一對(duì)句子A和B金赦,判斷這兩個(gè)句子是否是連續(xù)的音瓷。在訓(xùn)練的數(shù)據(jù)中,50%的B是A的下一個(gè)句子夹抗,50%的B是隨機(jī)抽取的绳慎。

而RoBERTa去除了NSP,而是每次輸入連續(xù)的多個(gè)句子,直到最大長度512(可以跨文章)杏愤。這種訓(xùn)練方式叫做(FULL - SENTENCES)靡砌,而原來的Bert每次只輸入兩個(gè)句子。實(shí)驗(yàn)表明在MNLI這種推斷句子關(guān)系的任務(wù)上RoBERTa也能有更好性能珊楼。

3. 更大的mini-batch

原本的BERTbase 的batch size是256通殃,訓(xùn)練1M個(gè)steps。RoBERTa的batch size為8k厕宗。為什么要用更大的batch size呢画舌?(除了因?yàn)樗麄冇绣X玩得起外)作者借鑒了在機(jī)器翻譯中,用更大的batch size配合更大學(xué)習(xí)率能提升模型優(yōu)化速率和模型性能的現(xiàn)象已慢,并且也用實(shí)驗(yàn)證明了確實(shí)Bert還能用更大的batch size曲聂。直接看圖吧,不細(xì)說了佑惠。

4. 更多的數(shù)據(jù)朋腋,更長時(shí)間的訓(xùn)練

借鑒XLNet用了比Bert多10倍的數(shù)據(jù),RoBERTa也用了更多的數(shù)據(jù)兢仰。性能確實(shí)再次彪升乍丈。當(dāng)然,也需要配合更長時(shí)間的訓(xùn)練把将。

這一套下來轻专,最終在GLUE, RACE, SQuAD上都達(dá)到了SOTA的性能。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末察蹲,一起剝皮案震驚了整個(gè)濱河市请垛,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌洽议,老刑警劉巖宗收,帶你破解...
    沈念sama閱讀 211,948評(píng)論 6 492
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異亚兄,居然都是意外死亡混稽,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,371評(píng)論 3 385
  • 文/潘曉璐 我一進(jìn)店門审胚,熙熙樓的掌柜王于貴愁眉苦臉地迎上來匈勋,“玉大人,你說我怎么就攤上這事膳叨∏⒔啵” “怎么了?”我有些...
    開封第一講書人閱讀 157,490評(píng)論 0 348
  • 文/不壞的土叔 我叫張陵菲嘴,是天一觀的道長饿自。 經(jīng)常有香客問我汰翠,道長,這世上最難降的妖魔是什么昭雌? 我笑而不...
    開封第一講書人閱讀 56,521評(píng)論 1 284
  • 正文 為了忘掉前任复唤,我火速辦了婚禮,結(jié)果婚禮上城豁,老公的妹妹穿的比我還像新娘苟穆。我一直安慰自己,他們只是感情好唱星,可當(dāng)我...
    茶點(diǎn)故事閱讀 65,627評(píng)論 6 386
  • 文/花漫 我一把揭開白布雳旅。 她就那樣靜靜地躺著,像睡著了一般间聊。 火紅的嫁衣襯著肌膚如雪攒盈。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 49,842評(píng)論 1 290
  • 那天哎榴,我揣著相機(jī)與錄音型豁,去河邊找鬼。 笑死尚蝌,一個(gè)胖子當(dāng)著我的面吹牛迎变,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播飘言,決...
    沈念sama閱讀 38,997評(píng)論 3 408
  • 文/蒼蘭香墨 我猛地睜開眼衣形,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了姿鸿?” 一聲冷哼從身側(cè)響起谆吴,我...
    開封第一講書人閱讀 37,741評(píng)論 0 268
  • 序言:老撾萬榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎苛预,沒想到半個(gè)月后句狼,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 44,203評(píng)論 1 303
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡热某,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,534評(píng)論 2 327
  • 正文 我和宋清朗相戀三年腻菇,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片昔馋。...
    茶點(diǎn)故事閱讀 38,673評(píng)論 1 341
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡芜繁,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出绒极,到底是詐尸還是另有隱情,我是刑警寧澤蔬捷,帶...
    沈念sama閱讀 34,339評(píng)論 4 330
  • 正文 年R本政府宣布垄提,位于F島的核電站榔袋,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏铡俐。R本人自食惡果不足惜凰兑,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,955評(píng)論 3 313
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望审丘。 院中可真熱鬧吏够,春花似錦、人聲如沸滩报。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,770評(píng)論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽脓钾。三九已至售睹,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間可训,已是汗流浹背昌妹。 一陣腳步聲響...
    開封第一講書人閱讀 32,000評(píng)論 1 266
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留握截,地道東北人飞崖。 一個(gè)月前我還...
    沈念sama閱讀 46,394評(píng)論 2 360
  • 正文 我出身青樓,卻偏偏與公主長得像谨胞,于是被迫代替她去往敵國和親固歪。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 43,562評(píng)論 2 349