跳到主要內容
AI News HubLIVE
站內改寫2 分鐘閱讀

超越線上與離線之爭:LLM智慧體蒸餾應在哪一層面跟隨學生模型?

文章摘要

該論文提出一種分層智慧體蒸餾框架,將“互動歷史由誰產生”和“動作令牌字首由誰生成”視為兩個獨立層面的線上程度,並系統研究了它們的配對、混合比例與訓練時序。實驗表明,讓學生模型產生歷史、教師模型生成動作字首通常是最強的單獨訓練訊號;兩層都完全跟隨學生效果最差;但完全線上資料在混合配置中仍有價值,且兩層的配對方式比線上資料總量更重要,先使用教師上下文、後使用完全線上上下文的效果優於靜態混合。

來源arXiv AI作者: Ya Gao, Zhengtao Zou, Pekka Marttinen
超越線上與離線之爭:LLM智慧體蒸餾應在哪一層面跟隨學生模型?
回報錯誤

更正管道尚未開通,可先複製下方文章資訊留存。

查看更正說明
直接讀正文

LLM 智慧體(agent)的蒸餾長期被簡化為一個二選一的問題:用教師模型生成的示範,還是用學生模型自己產生的互動經驗。這兩條路線大致對應離線策略(off-policy)與在策略(on-policy)蒸餾,二者都已被證明有效。但 arXiv 論文《Beyond On-Policy vs. Off-Policy: At What Level Should LLM Agent Distillation Follow the Student?》指出,這種二分法本身忽略了一個層次結構。作者主張,“在策略程度”至少應拆成兩個層面:一層決定互動歷史由哪一方產生,另一層決定當前動作的 token 字首由哪一方生成。這兩層並不必然使用同一個策略,因此可以獨立選擇、自由搭配。

圍繞這一觀察,論文提出了一個分層的智慧體蒸餾框架,把上述兩個層面的在策略性顯式形式化,並允許研究者控制它們的配對方式、混合比例以及訓練日程。為了在可控條件下研究該問題,作者採用自蒸餾(self-distillation)設定:教師模型是凍結的,並且擁有學生模型無法獲得的任務指導資訊,從而模擬“教師更強、學生需學習”的典型場景。實驗設計為三階段對照研究,覆蓋三個模型和四個互動式環境,規模上足以檢驗不同配置的穩定性。

研究給出的第一條結論較為反直覺:在單獨訓練時,最強的訊號通常來自“學生誘導的互動歷史”搭配“教師生成的動作字首”。也就是說,讓歷史由學生自己探索產生,但在生成具體動作時以教師的高質量字首為引導。相對地,如果兩個層面都完全跟隨學生,即完全在策略,效果普遍最差。這提示在策略資料並非越多越好,歷史與動作層面的來源需要區別對待。

不過,完全在策略的資料並非沒有價值。作者發現,當把它作為混合配置的一部分而非單獨使用引入時,它能帶來收益。更值得注意的是,即便兩個層面上的教師/學生比例完全相同,不同的配對方式也會導致顯著不同的效能。這說明“有多少在策略資料”並不是唯一關鍵,“在策略資料出現在哪個層面、如何與教師資料配對”同樣重要。

最後,訓練日程也影響結果。論文發現,把完全由教師生成的上下文放在訓練前期、將完全在策略的上下文推遲到後期,比靜態混合更能提升表現。綜合這些發現,作者為智慧體蒸餾提供了可操作的實踐原則:應回答在哪裡跟隨學生、跟隨多少、以及在什麼時候跟隨,而不是僅在整體上選擇在策略或離線策略。

展開要點與分析

文章情報

工程師進階

要點

  • 論文將智慧體蒸餾中的線上程度拆分為兩個層級:互動歷史由誰誘導,以及當前動作的令牌字首由誰生成。
  • 在三個模型和四個互動環境中,學生誘導歷史搭配教師生成動作字首通常提供最強的單獨訓練訊號,而完全跟隨學生在兩層都表現最差。
  • 兩層教師/學生比例相同的混合資料可能產生顯著不同的效能,說明配對方式比線上資料總量更重要。
  • 先使用完全由教師生成的上下文、再延遲引入完全線上上下文,優於靜態混合訓練。

要點與分析由自動化流程生成,可能有誤,請結合原始來源核實。