跳到主要內容
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)設定:教師模型是凍結的,並且擁有學生模型無法獲得的任務指導信息,從而模擬“教師更強、學生需學習”的典型場景。實驗設計為三階段對照研究,覆蓋三個模型和四個交互式環境,規模上足以檢驗不同配置的穩定性。

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

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

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

展開要點與分析

文章情報

工程師進階

要點

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

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