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

更好的開始,更好的結束:引導式迭代自我推理蒸餾用於壓縮推理

文章摘要

本文提出BIRD,一種兩階段自我推理蒸餾方法,透過先取樣簡潔解並進行提示切換SFT,然後應用線上逆KL蒸餾,顯著提升了大語言模型在長鏈推理中的效率。在Qwen3-8B上,MATH-500準確率從86.2%提升至92.0%,同時響應長度從3099降至1115 tokens。

來源arXiv Computational Linguistics作者: Leichao Dong, Dongxu Zhang, Yiding Sun, Qirui Wang, Yuhan Wang, Lin Chen, Jihua Zhu
更好的開始,更好的結束:引導式迭代自我推理蒸餾用於壓縮推理
回報錯誤

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

查看更正說明
直接讀正文

大型推理模型通常依賴長鏈思維(CoT)來解決複雜問題,但大量的計算資源被浪費在冗餘推導、重複自我驗證和無關繞路上。現有的線上自我蒸餾方法透過讓學生模型在自身生成字首上匹配簡潔版本來降低計算成本。然而,這些方法存在一個初始化瓶頸:由於監督訊號僅應用於已訪問的字首,從冗長的基模型開始訓練會將KL散度損失置於充滿噪聲、冗餘或偏離正確方向的上下文之中。在這種情況下,簡潔的教師模型只能提供區域性修正,學生模型仍會探索高效推理器應避免的軌跡。

為了解決這一問題,來自多個機構的研究人員提出了BIRD(引導式迭代自我推理蒸餾),這是一種兩階段自我推理蒸餾方法。在第一階段,BIRD從基模型中取樣簡潔解,僅保留答案正確的軌跡,並執行輕量級的提示切換SFT步驟。這些軌跡在簡潔指令下生成,但在原始任務提示下學習,從而將指令誘導的簡潔性轉化為預設推理行為。第一階段結束後,模型已經具備了生成更簡潔推理路徑的能力。

第二階段,從這一預熱模型開始,BIRD應用線上逆KL蒸餾,使用簡潔的自教師模型在更乾淨、資訊量更大的字首上進行訓練。這使得模型能夠進一步壓縮推理鏈,同時保持甚至提升準確性。

在Qwen3系列模型上的實驗驗證了BIRD的有效性。以Qwen3-8B為例,在MATH-500基準測試上,準確率從86.2%提升至92.0%,同時平均響應長度從3099 tokens減少至1115 tokens,大幅降低了計算開銷。在AIME基準測試上也取得了類似的改善。這些結果強調了字首支援作為高效推理蒸餾核心因素的重要性。BIRD不僅提升了計算效率,還保持了模型的推理質量,為未來在資源受限場景下部署大型推理模型提供了新的思路。該論文由Leichao Dong等人於2026年7月17日提交至arXiv,屬於計算與語言(cs.CL)領域。研究團隊透過實驗證明,相比於簡單的提示工程或冷啟動線上蒸餾,BIRD在多個基準上均表現出更優的準確率-效率折中。這一工作為推理模型的壓縮和高效部署提供了實用方案,尤其適用於需要快速響應的應用場景。

展開要點與分析

文章情報

工程師進階

要點

  • 現有線上自我蒸餾方法存在初始化瓶頸,模型在噪聲和冗餘字首上訓練。
  • BIRD第一階段利用簡潔指令取樣和提示切換SFT將簡潔性轉化為預設行為。
  • 第二階段在更乾淨的字首上進行逆KL蒸餾,進一步提升效率。
  • 實驗表明BIRD在Qwen3系列上實現了更好的準確率-效率權衡。

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