中科院計算所副研究員馮洋:神經(jīng)機(jī)器翻譯的訓(xùn)練改進(jìn)和解碼提速
2. 非自回歸解碼
最后介紹一下基于非自回歸的解碼方法,傳統(tǒng)的解碼方法是順序生成的。如果能夠使得解碼的時候并行的方式生成,這速度將會大大的提升。
傳統(tǒng)的非自回歸模型的做法是,在Transformer Encoder端頭部加一個Fertility預(yù)測,用來預(yù)測每個源端Token能翻譯成一個目標(biāo)端的Token,然后根據(jù)預(yù)測的結(jié)果,將源端的Token拷貝到Decoder的輸入,如果一個源端Token能夠翻譯兩個目標(biāo)Token,那就拷貝兩次,如果源端Token不會翻譯成目標(biāo)端Token,那就不拷貝。由于每一步輸出的譯文是沒有給到下一步的,所以是可以并行的。對于Fertility的訓(xùn)練是采用某種對齊模型,通過計算源端和目標(biāo)端的對齊關(guān)系,然后就可以得到源端和目標(biāo)端的對齊結(jié)果,就可以采用監(jiān)督的方式來訓(xùn)練Fertility分支。
該方法有一個問題,就是在翻譯當(dāng)前步的時候沒有考慮上一步的翻譯信息。這樣就可能導(dǎo)致翻譯結(jié)果的流暢度不夠好。我們的方法就是在該方法的基礎(chǔ)上添加了序列上的信息。這樣模型既能并行執(zhí)行,又能考慮的到前后的序列關(guān)系。
我們的工作分為兩個方面,一個是在訓(xùn)練上添加序列信息,一個是在模型上面同樣也添加序列信息。序列訓(xùn)練采用的是Reinforce的方法,Reinforce的方法非常難以訓(xùn)練,這是因為其方差非常大,方差大的原因是強化學(xué)習(xí)episode(一條軌跡從開始到結(jié)束)的搜索空間非常大,我們每次只是采樣出一個episode,然后根據(jù)這個episode進(jìn)行計算,通過大數(shù)定律,我們可以假設(shè)這最終得到的是一個梯度的無偏估計。但是在實際情況下,抖動是非常大的。
將Reinforce算法應(yīng)用到我們這個場景,首先看第一個公式,由于目標(biāo)端詞的概率是獨立的,所以就可以寫成連乘的形式,第二個公式就是傳統(tǒng)的Reinforce公式,就是翻譯的reward。是通過前向后向算法計算出來的當(dāng)前步的reward。
上面的slides介紹的是計算reward時候的不同,接下來看sampling機(jī)制的區(qū)別。根據(jù)生成前后詞的獨立性,每一步我們并不是采樣出一個詞,而是采樣出K+1個詞。這樣的話就可以看做我們一次更新的過程中考慮到更多的episode,而不是僅用一個episode就去訓(xùn)練了。具體的做法是,每一步,我們先取Top-K,計算一下?lián)p失函數(shù)的值,然后從剩下的Token中再采樣出來一個。我們將這兩部分的loss合起來,是為了保證無偏估計。為前k個翻譯的概率的和。
另外一個方法就是模型上的改進(jìn),在非自回歸層的上面加上自回歸層。具體的做法是,模型分為 Bottom Layer,F(xiàn)usion Layer,Top Layer。Bottom Layer就是之前介紹的非自回歸模型,F(xiàn)usion Layer的作用是將非自回歸模型的輸出和其Embedding整合起來,Top-Layer和Transformer 的解碼器基本一致。
實驗結(jié)果:AR(Transformer),NAT(非自回歸的方法),IRNAT(迭代的非自回歸方法),最后是我們提出的方法,第一種是在訓(xùn)練的過程中引入序列信息,第二是在模型上進(jìn)行改進(jìn)。作為對比的數(shù)據(jù)集有三個,前兩個數(shù)據(jù)集比較小。主要關(guān)注第三個數(shù)據(jù)集?梢钥闯觯褂肗AT來代替AR模型的話,效果會降6個點左右,迭代的方法會帶來1到2個點的提升。我們提出的reinforce方法和傳統(tǒng)的reinforce方法相比,有0.6個點的提升。加上回歸層的模型已經(jīng)接近Transformer的效果了。關(guān)于速度的提升,如果僅訓(xùn)練的時候采用序列信息,速度可以提升10倍。如果是NAT加上自回歸層的方法,速度也可以提高4倍左右。
這里有一些翻譯實例,可以看出 NAT-base的方法流暢性不夠好,重復(fù)很多“more more …”,因為沒有考慮序列信息,所以導(dǎo)致結(jié)果的流暢度不行。使用我們提出的reinforce方法,能夠一定程度上的緩解流暢度的問題,但是問題還是存在。通過使用NAT+AR的方法,能夠更好的緩解流暢度的問題。

請輸入評論內(nèi)容...
請輸入評論/評論長度6~500個字
最新活動更多
-
7月8日立即報名>> 【在線會議】英飛凌新一代智能照明方案賦能綠色建筑與工業(yè)互聯(lián)
-
7月22-29日立即報名>> 【線下論壇】第三屆安富利汽車生態(tài)圈峰會
-
7.30-8.1火熱報名中>> 全數(shù)會2025(第六屆)機(jī)器人及智能工廠展
-
7月31日免費預(yù)約>> OFweek 2025具身智能機(jī)器人產(chǎn)業(yè)技術(shù)創(chuàng)新應(yīng)用論壇
-
免費參會立即報名>> 7月30日- 8月1日 2025全數(shù)會工業(yè)芯片與傳感儀表展
-
即日-2025.8.1立即下載>> 《2024智能制造產(chǎn)業(yè)高端化、智能化、綠色化發(fā)展藍(lán)皮書》
推薦專題