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

基於同伴成功與失敗的多輪次線上策略蒸餾

文章摘要

提出MOPD框架,利用學生的多個生成軌跡(包括成功和失敗的)構建更有效的教師訊號,優於標準線上策略蒸餾方法。實驗表明,混合成功-失敗上下文能更好地對齊教師分數與驗證器獎勵,帶來更忠實、例項自適應的監督。

來源arXiv Machine Learning作者: Weichen Yu, Xiaomin Li, Yizhou Zhao, Xiaoze Liu, Ruowang Zhang, Haixin Wang, Yinyi Luo, Chen Henry Wu, Gaurav Mittal, Matt Fredrikson, Yu Hu
基於同伴成功與失敗的多輪次線上策略蒸餾
回報錯誤

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

查看更正說明
直接讀正文

大型語言模型在後期訓練中常使用稀疏的驗證器獎勵,這種獎勵僅能指示取樣軌跡是否成功,卻無法對推理過程中的成功或失敗位置提供具體指導。線上策略蒸餾(On-policy Distillation, OPD)透過訓練學生模型自身生成的軌跡來提供更密集的令牌級監督,但現有方法通常獨立處理每個軌跡,忽略了同一提示下其他嘗試中蘊含的豐富資訊。

為了克服這一侷限,研究團隊提出了多輪次線上策略蒸餾(Multi-Rollout On-Policy Distillation, MOPD),這是一種基於同伴條件的蒸餾框架。MOPD的核心創新在於,它利用學生模型在同一提示下生成的多個軌跡(包括成功和失敗的嘗試)來構建資訊更為豐富的教師訊號。具體來說,成功的同伴軌跡為正面的推理模式提供證據,而失敗的同伴軌跡則提供結構化的負面證據,幫助模型學會避免常見的錯誤模式。

研究探索了兩種同伴上下文構建方式:正面同伴模仿(positive peer imitation)和對比成功-失敗條件(contrastive success-failure conditioning)。在競爭程式設計、數學推理、科學問答和工具使用等多個基準測試上,MOPD均表現出優於標準線上策略基線的效能。進一步的教師訊號分析顯示,混合成功-失敗上下文能更好地使教師評分與驗證器獎勵對齊,這表明效能提升源於更忠實、例項自適應的監督訊號。這些結果證明,有效的線上策略蒸餾應充分利用學生的多輪次試錯行為,而非將各軌跡視為孤立樣本。

該工作由Weichen Yu等人完成,論文共23頁,涵蓋機器學習與人工智慧領域的研究。實驗覆蓋了多個具有挑戰性的推理任務,充分驗證了MOPD框架的通用性和有效性。

展開要點與分析

文章情報

工程師進階

要點

  • MOPD透過考慮同一提示下的多個嘗試(成功和失敗)來增強蒸餾。
  • 研究了兩種同伴上下文構建:正面模仿和對比成功-失敗條件。
  • 在程式設計、數學推理、科學問答和工具使用基準上均優於標準基線。
  • 混合成功-失敗上下文能產生更忠實、例項自適應的教師訊號。

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