這一年來(lái)，以 ChatGPT 和 GPT-4 為代表的大語(yǔ)言模型（LLM）發(fā)展迅速，緊隨其后，Meta 開源的 LLaMa、Llama 2 系列模型在 AI 界也引起的了不小的轟動(dòng)。但隨之而來(lái)的是爭(zhēng)議不斷，有人認(rèn)為 LLM 存在一些不可控的風(fēng)險(xiǎn)，給人類生存構(gòu)成一些潛在威脅。

為了應(yīng)對(duì)這些挑戰(zhàn)，對(duì) LLM 對(duì)齊的研究變得越來(lái)越重要，有研究者提出指令跟隨（instruction following），但這種方法需要大量的人工注釋。然而，注釋如此高質(zhì)量的指令跟隨數(shù)據(jù)集耗費(fèi)巨大。

本文來(lái)自Meta AI 的研究者提出了一種可擴(kuò)展的方法即指令回譯（instruction backtranslation），該方法通過(guò)自動(dòng)注釋相應(yīng)的指令來(lái)構(gòu)建高質(zhì)量的指令跟隨語(yǔ)言模型。

(資料圖片)

論文地址：https://arxiv.org/pdf/2308.06259.pdf

具體而言，該研究從一個(gè)語(yǔ)言模型開始，并作為種子模型，該模型在少量的種子數(shù)據(jù)以及 web 語(yǔ)料庫(kù)上進(jìn)行了微調(diào)。種子模型的作用是用來(lái)構(gòu)建訓(xùn)練樣本，然后這些樣本中的一些高質(zhì)量樣本將會(huì)被篩選出來(lái)，接著，這些數(shù)據(jù)被用來(lái)微調(diào)一個(gè)更強(qiáng)大的模型。

經(jīng)過(guò)兩輪迭代的數(shù)據(jù)集對(duì) LLaMa 進(jìn)行微調(diào)，所產(chǎn)生的模型 Humpback 在 Alpaca 排行榜上優(yōu)于其他現(xiàn)有的非蒸餾模型，如 LIMA、Claude、Guanaco 等。

Humpback 原意為座頭鯨，又名駝背鯨，Meta 將模型命名為 Humpback，也別有深意吧。

之所以稱為指令回譯，研究者表示這借鑒了機(jī)器翻譯中經(jīng)典的反向翻譯方法，其中人類編寫的目標(biāo)句子會(huì)自動(dòng)用模型生成的另一種語(yǔ)言的源句子進(jìn)行注釋。

圖靈獎(jiǎng)得主 Yann LeCun 高度概括了這項(xiàng)研究的方法，并稱贊 Meta 這項(xiàng)工作為對(duì)齊研究做出重要貢獻(xiàn)：

還有網(wǎng)友對(duì)這項(xiàng)研究進(jìn)行了很好的概括：數(shù)據(jù)質(zhì)量對(duì)大模型來(lái)說(shuō)確實(shí)很重要，研究過(guò)程中，他們使用不同級(jí)別的過(guò)濾數(shù)據(jù)，微調(diào)了一個(gè)模型，結(jié)果表明，只有最好的樣本才能得出比其他樣本表現(xiàn)更好的模型。

該論文提出了一種需要兩個(gè)步驟完成的新的數(shù)據(jù)增強(qiáng)范式。首先，必須擁有一組種子（指令、輸出）對(duì)和語(yǔ)料庫(kù)才能生成更多好的指令數(shù)據(jù)。

下圖比較了 Humpback 與一些開源模型和專有模型。

下表 4 表明，本文方法在 65B 和 33B 模型尺度上都是非蒸餾模型中表現(xiàn)最好的模型。

下面我們看看具體方法。

方法簡(jiǎn)介

該研究提出了一種自訓(xùn)練方法（self-training），該方法通常假定可以訪問(wèn)基本語(yǔ)言模型、少量種子數(shù)據(jù)和未標(biāo)記的樣本集（例如網(wǎng)絡(luò)語(yǔ)料庫(kù)）。未標(biāo)記數(shù)據(jù)往往是一大堆形態(tài)各異的文檔，由人類編寫，其中包括人類感興趣的各種話題內(nèi)容，但最重要的是沒(méi)有與指令進(jìn)行配對(duì)。

這里還有兩個(gè)關(guān)鍵的假設(shè)，第一個(gè)假設(shè)是這個(gè)非常大的文本集（未標(biāo)記樣本集）存在一些子集，適合作為某些用戶指令的生成樣本。第二個(gè)假設(shè)是可以預(yù)測(cè)這些候選答案的指令，這些指令可以用于形成高質(zhì)量樣本對(duì)，以訓(xùn)練指令遵循模型。

如下圖 1 所示，該研究提出指令回譯過(guò)程包含兩個(gè)核心步驟：

自增強(qiáng)：為未標(biāo)記的數(shù)據(jù)（即網(wǎng)絡(luò)語(yǔ)料庫(kù)）生成指令，以為指令調(diào)優(yōu)產(chǎn)生訓(xùn)練數(shù)據(jù)對(duì)（指令 - 輸出）。自管理：自主選擇高質(zhì)量樣本數(shù)據(jù)作為訓(xùn)練數(shù)據(jù)，以微調(diào)基礎(chǔ)模型來(lái)遵循指令，這種方法是迭代完成的。

其中，自管理步驟采用的 prompt 如下表 1 所示：

實(shí)驗(yàn)及結(jié)果

本文的數(shù)據(jù)集主要包括種子數(shù)據(jù)和增強(qiáng)數(shù)據(jù)，具體信息如表 2 和圖 2 所示：

圖 3 表示盡管擴(kuò)大了數(shù)據(jù)規(guī)模，但沒(méi)有自我管理（self-curation）的增強(qiáng)數(shù)據(jù)用來(lái)訓(xùn)練模型并不能提高指令跟隨性能。

下圖比較了不同指令調(diào)優(yōu)數(shù)據(jù)集的數(shù)據(jù)效率。

數(shù)據(jù)和模型的聯(lián)合擴(kuò)展：該研究發(fā)現(xiàn)在 7B 模型中觀察到的數(shù)據(jù)擴(kuò)展趨勢(shì)同樣也適用于更大的模型。例如對(duì) 65B 的種子模型增加高質(zhì)量的增強(qiáng)數(shù)據(jù)會(huì)帶來(lái)進(jìn)一步的改進(jìn)。

常識(shí)推理：該研究在五個(gè)常識(shí)推理基準(zhǔn)上進(jìn)行了測(cè)試，SIQA 、PIQA、Arc-Easy、Arc-Challenge 和 Openbook QA （OBQA） ， 結(jié)果總結(jié)于表 5 中。結(jié)果表明，與基礎(chǔ)模型相比，本文模型在社會(huì)推理等多個(gè)方面的表現(xiàn)有所提高。

MMLU：表 6 總結(jié)了不同模型在 MMLU（massive multitask language understanding）的結(jié)果。與基礎(chǔ)模型相比，本文微調(diào)模型提高了零樣本準(zhǔn)確率，但在 5 個(gè)樣本上下文示例中表現(xiàn)不佳。

關(guān)鍵詞：