中科院計(jì)算所副研究員馮洋:神經(jīng)機(jī)器翻譯的訓(xùn)練改進(jìn)和解碼提速
2. 可導(dǎo)的序列級(jí)目標(biāo)
接下來(lái)介紹如果解決詞級(jí)匹配的對(duì)于好一點(diǎn)的匹配和差的匹配一視同仁的問(wèn)題。
這個(gè)是我們?cè)贓MNLP 2018上所做的工作。通過(guò)使用可導(dǎo)的序列級(jí)目標(biāo)來(lái)解決詞級(jí)匹配的問(wèn)題。
首先介紹一下傳統(tǒng)的序列級(jí)損失函數(shù)。傳統(tǒng)的序列級(jí)損失函數(shù)基本上都是基于N-gram正確率的損失函數(shù),比如,BLEU,GLEU等等。計(jì)算方法為,命中n-gram的個(gè)數(shù)/總共的n-gram的個(gè)數(shù)(candidate),其中n-gram的個(gè)數(shù)為其每個(gè)詞語(yǔ)出現(xiàn)頻次的乘積。
直接使用BLEU不可到的原因是因?yàn)椴僮髦杏衋rgmax,為了使其可導(dǎo),我們使用token的預(yù)測(cè)概率,而非使用argmax。這個(gè)方法和直接用BLEU作為Score,然后reinforce算法直接訓(xùn)練對(duì)比有啥優(yōu)勢(shì)?由于reinforce算法的方差比較大,所以在訓(xùn)練的時(shí)候是很難收斂的。而使用傳統(tǒng)的梯度下降的方法,訓(xùn)練過(guò)程就會(huì)平穩(wěn)的多。
這里是3-gram的例子,其中output是概率最高的詞,3-gram概率的是由獨(dú)立Token的輸出概率相乘得到結(jié)果,然后求和會(huì)得到The total probabilistic count of 3-grams。將匹配上的3-gram的概率拿出來(lái)求和作為分子,Total probabilistic count作為分母,計(jì)算得到 Precision of 3-grams。這就是我們的loss。
這個(gè)例子用來(lái)展示整個(gè)的訓(xùn)練過(guò)程,這里需要注意的一點(diǎn)就是,和傳統(tǒng)的teacher forcing方式不同,這里當(dāng)前步輸入的為上一步預(yù)測(cè)的結(jié)果(貪心搜索得到的結(jié)果),而不是ground truth的值。剩下的就是按照上頁(yè)slides介紹的來(lái)計(jì)算loss。對(duì)于loss采用傳統(tǒng)的梯度下降算法即可。下面貼的是在數(shù)據(jù)集上的結(jié)果。
從結(jié)果中可以看出,2-gram比4-gram的效果要好,這里我們給出的解釋是,過(guò)多的使用自己生成的去計(jì)算的話(huà),會(huì)存在一定程度上的錯(cuò)誤累積。
在訓(xùn)練的時(shí)候,也是可以通過(guò)teacher forcing的方式來(lái)訓(xùn)練的,但是從圖中可以看出,teacherforcing的方式收斂的比較快,但是效果不如greedy search的方式好。

發(fā)表評(píng)論
登錄
手機(jī)
驗(yàn)證碼
立即登錄即可訪(fǎng)問(wèn)所有OFweek服務(wù)
還不是會(huì)員?免費(fèi)注冊(cè)
忘記密碼請(qǐng)輸入評(píng)論內(nèi)容...
請(qǐng)輸入評(píng)論/評(píng)論長(zhǎng)度6~500個(gè)字
圖片新聞
最新活動(dòng)更多
-
7月8日立即報(bào)名>> 【在線(xiàn)會(huì)議】英飛凌新一代智能照明方案賦能綠色建筑與工業(yè)互聯(lián)
-
7月22-29日立即報(bào)名>> 【線(xiàn)下論壇】第三屆安富利汽車(chē)生態(tài)圈峰會(huì)
-
7月31日免費(fèi)預(yù)約>> OFweek 2025具身機(jī)器人動(dòng)力電池技術(shù)應(yīng)用大會(huì)
-
7.30-8.1火熱報(bào)名中>> 全數(shù)會(huì)2025(第六屆)機(jī)器人及智能工廠(chǎng)展
-
免費(fèi)參會(huì)立即報(bào)名>> 7月30日- 8月1日 2025全數(shù)會(huì)工業(yè)芯片與傳感儀表展
-
即日-2025.8.1立即下載>> 《2024智能制造產(chǎn)業(yè)高端化、智能化、綠色化發(fā)展藍(lán)皮書(shū)》
推薦專(zhuān)題
- 1 AI 眼鏡讓百萬(wàn) APP「集體失業(yè)」?
- 2 豆包前負(fù)責(zé)人喬木出軌BP后續(xù):均被辭退
- 3 一文看懂視覺(jué)語(yǔ)言動(dòng)作模型(VLA)及其應(yīng)用
- 4 “支付+”時(shí)代,支付即生態(tài) | 2025中國(guó)跨境支付十大趨勢(shì)
- 5 中國(guó)最具實(shí)力AI公司TOP10
- 6 特斯拉Robotaxi上路,馬斯克端上畫(huà)了十年的餅
- 7 國(guó)家數(shù)據(jù)局局長(zhǎng)劉烈宏調(diào)研格創(chuàng)東智
- 8 AI的夏天:第四范式VS云從科技VS地平線(xiàn)機(jī)器人
- 9 張勇等人退出阿里合伙人
- 10 深圳跑出40億超級(jí)隱形冠軍:賣(mài)機(jī)器人年入6.1億,港股上市