天天看點

首次:微軟用GPT-4做大模型指令微調,新任務零樣本性能再提升

編輯:杜偉、陳萍

大模型指令微調水準在不斷地提高,這次微軟用上了 GPT-4。

我們知道,從谷歌 T5 模型到 OpenAI GPT 系列大模型,大語言模型(LLMs)已經展現出了令人印象深刻的泛化能力,比如上下文學習和思維鍊推理。同時為了使得 LLMs 遵循自然語言指令和完成真實世界任務,研究人員一直在探索 LLMs 的指令微調方法。實作方式有兩種:一是使用人類标注的 prompt 和回報在廣泛任務上微調模型,二是使用通過手動或自動生成指令增強的公共基準和資料集來監督微調。

在這些方法中,Self-Instruct 微調是一種簡單有效的方法,它從 SOTA 指令微調的教師 LLMs 生成的指令遵循資料中學習,使得 LLMs 與人類意圖對齊。事實證明,指令微調已經成為提升 LLMs 零樣本和小樣本泛化能力的有效手段。

最近,ChatGPT 和 GPT-4 的成功為使用指令微調來改進開源 LLMs 提供了巨大的機遇。Meta LLaMA 是一系列開源 LLMs,其性能與 GPT-3 等專有 LLMs 相媲美。為了教 LLaMA 遵循指令,Self-Instruct 因其卓越性能和低成本被快速采用。比如斯坦福的 Alpaca 模型使用由 GPT-3.5 生成的 52k 指令遵循樣本,Vicuna 模型使用約 700k 來自 ShareGPT 的指令遵循樣本。

為了推進 LLMs 指令微調的 SOTA 水準,微軟研究院在其論文《Instruction Turing with GPT-4》中首次使用 GPT-4 作為教師模型進行 self-intruct 微調。

首次:微軟用GPT-4做大模型指令微調,新任務零樣本性能再提升

研究者一方面釋出了 GPT-4 生成的資料,包括中英文的 52k 指令遵循資料集、GPT-4 生成的對三種指令微調模型的輸出進行評級的回報資料。

另一方面基于 GPT-4 生成的資料開發了指令微調的 LLaMA 模型和獎勵模型。為了評估指令微調 LLMs 的品質,研究者使用三個名額對測試樣本進行評估:對三個對齊标準的人工評估、基于 GPT-4 回報的自動評估以及非自然指令的 ROUGE-L(自動文摘評測方法之一)。

實驗結果驗證了使用 GPT-4 生成的資料進行 LLMs 指令微調的有效性。GPT-4 生成的 52k 中英指令遵循資料在新任務上實作了較以往 SOTA 模型更好的零樣本性能。目前,研究者已經公開了使用 GPT-4 生成的資料以及相關代碼。

資料集

該研究使用 GPT-4 生成以下四個資料集:

英語指令遵循資料集(English Instruction-Following Data):對于從 Alpaca 收集到的 52K 指令,每個指令都提供了一個英文 GPT-4 答案。該資料集主要用來探索和比較 GPT-4 答案和 GPT-3 答案的資料統計。

中文指令遵循資料集(Chinese Instruction-Following Data):該研究使用 ChatGPT 将 52K 指令翻譯成中文,并要求 GPT-4 用中文回答。

比較資料(Comparison Data):讓 GPT-4 給自己的反應打分,分數範圍從 1 到 10。此外,該研究還要求 GPT-4 對 GPT-4、GPT-3.5 和 OPT-IML 三種模型的響應進行比較和評分。這一資料集主要用來訓練獎勵模型。

非自然指令的回答(Answers on Unnatural Instructions):GPT-4 的回答在 68K 指令 - 輸入 - 輸出三組核心資料集上解碼。該子集用于量化 GPT-4 與指令微調模型之間的差距。

首次:微軟用GPT-4做大模型指令微調,新任務零樣本性能再提升

圖 1 比較了 GPT-4 和 GPT-3.5 的英文輸出響應集。圖 1 (a) 和 (b) 顯示了兩個輸出集合頻率高于 10 的動 - 名詞對(verb-noun pairs),圖 1 (c) 比較了兩個集合中出現頻率最高的 25 對單詞,圖 1 (d) 比較了序列長度的頻率分布,結果顯示,GPT-4 傾向于生成比 GPT-3.5 更長的序列。

首次:微軟用GPT-4做大模型指令微調,新任務零樣本性能再提升

指令微調語言模型

該研究基于 LLaMA 7B checkpoint、并使用監督微調訓練了兩個模型:(i) LLaMA-GPT4 ,在 GPT-4 生成的 52K 英語指令遵循資料上訓練。(ii) LLaMA-GPT4-CN,在來自 GPT-4 生成的 52K 中文指令遵循資料上訓練完成。

獎勵模型

人類回報強化學習 (RLHF) 旨在使 LLM 行為與人類偏好保持一緻,獎勵模組化是其關鍵部分之一,這一問題被往往公式化為回歸任務,以預測給定提示和響應之間的獎勵。但這種方法通常需要大規模的比較資料,現有開源模型如 Alpaca、Vicuna 和 Dolly 由于标注比較資料成本很高,是以不涉及 RLHF。與此同時,最近的研究表明,GPT-4 能夠識别和修複自己的錯誤,并準确判斷響應的品質。是以,為了促進 RLHF 的研究,該研究使用 GPT-4 建立了比較資料,如上文所述。

為了評估資料品質,該研究還訓練了一個基于 OPT 1.3B 的獎勵模型進行該資料集的評估。比較資料的分布如圖 2 所示。

首次:微軟用GPT-4做大模型指令微調,新任務零樣本性能再提升

實驗

該研究利用以下三種類型進行評估:人類評估、GPT-4 以及非自然指令評估。結果證明,與其他機器生成的資料相比,使用 GPT-4 生成的資料是進行 LLM 指令微調的一種高效且有效的方法。接下來我們看看具體實驗過程。

人類評估

圖 3 (a) 為 LLaMA-GPT4 vs Alpaca 比較結果,實驗表明在 Helpfulness 這一名額下,GPT-4 以 54.12% 的得分勝出。圖 3 (b) 為 LLaMA-GPT4 vs GPT-4 比較結果,表明 GPT-4 指令微調的 LLaMA 的性能與原始的 GPT-4 類似。

首次:微軟用GPT-4做大模型指令微調,新任務零樣本性能再提升

與使用自動求值的 SOTA 進行比較

該研究使用 GPT-4 對不同模型在 80 個未見問題上的回答進行自動評估。首先從 LLaMA-GPT-4 (7B) 和 GPT-4 兩個聊天機器人中收集答案,并使用其他聊天機器人釋出答案,包括 LLaMA (13B),Alpaca (13B),Vicuna (13B),Bard (谷歌,2023) 和 ChatGPT。對于每次評估,該研究要求 GPT-4 對兩個模型之間的響應品質進行評分,評分範圍從 1 到 10。結果如圖 4 所示。

首次:微軟用GPT-4做大模型指令微調,新任務零樣本性能再提升

圖 4 (c,d) 比較了所有聊天機器人。LLaMA_GPT4 性能更高:7B LLaMA GPT4 的性能優于 13B Alpaca 和 LLaMA。然而,LLaMA_GPT4 與 GPT-4 等大型商業聊天機器人相比,仍有差距。

研究者在下圖 5 中進一步研究了所有聊天機器人的性能。首先使用 GPT-4 将聊天機器人的英文響應翻譯成中文,接着使用 GPT-4 将英文問題翻譯成中文以獲得答案。與 GPT-4 翻譯和生成的中文響應的比較如 5 (a) 和 5 (b) 所示,5 (c) 中顯示了所有被要求用中文回答的模型結果。

首次:微軟用GPT-4做大模型指令微調,新任務零樣本性能再提升

在下圖 6 中,研究者将 LLaMA-GPT4 與 GPT-4、Alpaca 非自然指令進行比較。結果顯示,LLaMA-GPT4 和 GPT-4 随 ground truth 響應長度的增加表現更好。這意味着當場景更具創意時,它們可以更好地遵循指令。當序列長度較短時,LLaMA-GPT4 和 GPT-4 都能生成包含簡單 ground truth 答案的響應,并且添加額外單詞可以使響應更像聊天。

首次:微軟用GPT-4做大模型指令微調,新任務零樣本性能再提升

論文位址:https://arxiv.org/pdf/2304.03277.pdf

項目位址:https://instruction-tuning-with-gpt-4.github.io/

GitHub 位址:https://github.com/Instruction-Tuning-with-GPT-4/GPT-4-LLM

繼續閱讀