谷歌130億參數(shù)多語言模型mT5重磅來襲,101種語言輕松遷移
Facebook剛剛開源多語種機(jī)器翻譯模型「M2M-100」,這邊谷歌也來了。谷歌宣布,基于T5的mT5多語言模型正式開源,最大模型130億參數(shù),與Facebook的M2M相比,參數(shù)少了,而且支持更多語種。
前幾天,F(xiàn)acebook發(fā)了一個(gè)百種語言互譯的模型M2M-100,這邊谷歌著急了,翻譯可是我的老本行啊。
剛剛,谷歌也放出了一個(gè)名為 mT5的模型,在一系列英語自然處理任務(wù)上制服了各種SOTA。
你發(fā),我也發(fā),你支持100種,我支持101種?。m然多這一種沒有多大意義,但氣勢(shì)上不能輸)
mT5是谷歌 T5模型的多語種變體,訓(xùn)練的數(shù)據(jù)集涵蓋了101種語言,包含3億至130億個(gè)參數(shù),從參數(shù)量來看,的確是一個(gè)超大模型。
多語言模型是AI的橋梁,但難以避免「有毒」輸出
世界上成體系的語言現(xiàn)在大概有7000種,縱然人工智能在計(jì)算機(jī)視覺、語音識(shí)別等領(lǐng)域已經(jīng)超越了人類,但只局限在少數(shù)幾種語言。
想把通用的AI能力,遷移到一個(gè)小語種上,幾乎相當(dāng)于從頭再來,有點(diǎn)得不償失。
所以跨語種成為了AI能力遷移的重要橋梁。
多語言人工智能模型設(shè)計(jì)的目標(biāo)就是建立一個(gè)能夠理解世界上大部分語言的模型。
多語言人工智能模型可以在相似的語言之間共享信息,降低對(duì)數(shù)據(jù)和資源的依賴,并且允許少樣本或零樣本學(xué)習(xí)。隨著模型規(guī)模的擴(kuò)大,往往需要更大的數(shù)據(jù)集。
C4是從公共網(wǎng)站獲得的大約750gb 的英文文本的集合,mC4是 C4的一個(gè)變體,C4數(shù)據(jù)集主要為英語任務(wù)設(shè)計(jì),mC4搜集了過去71個(gè)月的網(wǎng)頁數(shù)據(jù),涵蓋了107種語言,這比 C4使用的源數(shù)據(jù)要多得多。
mC4中各種語言的網(wǎng)頁數(shù)量
有證據(jù)表明,語言模型會(huì)放大數(shù)據(jù)集中存在的偏差。
雖然一些研究人員聲稱,目前的機(jī)器學(xué)習(xí)技術(shù)難以避免「有毒」的輸出,但是谷歌的研究人員一直在試圖減輕 mT5的偏見,比如過濾數(shù)據(jù)中含有偏激語言的頁面,使用 cld3檢測(cè)頁面的語言,將置信度低于70% 的頁面直接刪除。
mT5:使用250000詞匯,多語言數(shù)據(jù)采樣策略是關(guān)鍵
mT5的模型架構(gòu)和訓(xùn)練過程與T5十分相似,mT5基于T5中的一些技巧,比如使用GeGLU的非線性(Shazeer,2020年),在較大模型中縮放dmodel而不是dff來對(duì)T5進(jìn)行改進(jìn),并且僅對(duì)未標(biāo)記的數(shù)據(jù)進(jìn)行預(yù)訓(xùn)練而不會(huì)出現(xiàn)信息丟失。
訓(xùn)練多語言模型的最重要的一點(diǎn)是如何從每種語言中采樣數(shù)據(jù)。
但是,這種選擇是零和博弈:如果對(duì)低資源語言的采樣過于頻繁,則該模型可能會(huì)過擬合;如果對(duì)高資源語言的訓(xùn)練不夠充分,則模型的通用性會(huì)受限。
因此,研究團(tuán)隊(duì)采用Devlin和Arivazhagan等人使用的方法,并根據(jù)概率p(L)∝ | L |^α,對(duì)資源較少的語言進(jìn)行采樣。其中p(L)是在預(yù)訓(xùn)練期間從給定語言中采樣的概率,| L |是該語言中樣本的數(shù)量,α是個(gè)超參數(shù),谷歌經(jīng)過實(shí)驗(yàn)發(fā)現(xiàn)α取0.3的效果最好。
為了適應(yīng)更多的語言,mT5將詞匯量增加到250,000個(gè)單詞。與T5一樣,使用SentencePiece和wordPiece來訓(xùn)練模型。
Sentencepiece示意
那采樣之后有的字符沒覆蓋到怎么辦?
研究團(tuán)隊(duì)為了適應(yīng)具有大字符集的語言(比如中文),使用了0.99999的字符覆蓋率,但還啟用了SentencePiece的「字節(jié)后退」功能,以確??梢晕ㄒ痪幋a任何字符串。
為了讓結(jié)果更直觀,研究人員與現(xiàn)有的大規(guī)模多語言預(yù)訓(xùn)練語言模型進(jìn)行了簡要比較,主要是支持?jǐn)?shù)十種語言的模型。
mT5專治各種SOTA,但基準(zhǔn)測(cè)試未必能代表實(shí)力
截至2020年10月,實(shí)驗(yàn)中最大 mT5模型擁有130億個(gè)參數(shù),超過了所有測(cè)試基準(zhǔn),包括來自 XTREME 多語言基準(zhǔn)測(cè)試的5個(gè)任務(wù),涵蓋14種語言的 XNLI 衍生任務(wù),分別有10種、7種和11種語言的 XQuAD、 MLQA 和 TyDi QA/閱讀理解基準(zhǔn)測(cè)試,以及有7種語言的 PAWS-X 釋義識(shí)別。
實(shí)驗(yàn)結(jié)果可以看到,在閱讀理解、機(jī)器問答等各項(xiàng)基準(zhǔn)測(cè)試中mT5模型都優(yōu)于之前的預(yù)訓(xùn)練語言模型。
至于基準(zhǔn)測(cè)試能否充分反映模型在生產(chǎn)環(huán)境中的表現(xiàn),就另當(dāng)別論了。
對(duì)預(yù)訓(xùn)練語言模型最直白的測(cè)試方法就是開放域問答,看訓(xùn)練后的模型能否回答沒見過的新問題,目前來看,即使強(qiáng)如GPT-3,也經(jīng)常答非所問。
但是谷歌的研究人員斷言,mT5是向功能強(qiáng)大的模型邁出的一步,而這些模型不需要復(fù)雜的建模技術(shù)。
總的來說,mT5展示出了跨語言表征學(xué)習(xí)中的重要性,并表明了通過過濾、并行數(shù)據(jù)或其他一些調(diào)優(yōu)技巧,實(shí)現(xiàn)跨語言能力遷移是可行的。
這個(gè)源自T5的模型,完全適用于多語言環(huán)境。