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

可預測的GRPO:訓練動力學的封閉形式模型

文章摘要

本文提出了GRPO訓練動力學的第一性原理降階模型,將經驗性的單指數飽和律重新解釋為過阻尼極限,並引入了慣性項以描述慢啟動階段。模型提供了與可獨立測量量相關的預測,包括群大小不變性、重新整理間隔的穩定性閾值以及過阻尼到振盪的轉變。實驗表明,封閉形式軌跡在三個模型和兩個群大小上對訓練獎勵的擬合R²≥0.91,群大小不變性在獎勵曲線和八個數學基準的分佈外轉移上均成立。此外,模型提供了診斷獎勵曲線混淆的失敗模式(獎勵駭客、優勢退化、策略集中和動力學不穩定性)的方法。

來源arXiv Machine Learning作者: Rajat Ghosh, Datta Nimmaturi, Aryan Singhal, Vaishnavi Bhargava, Henry Wong, Johnu George, Debojyoti Dutta
可預測的GRPO:訓練動力學的封閉形式模型
回報錯誤

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

查看更正說明
直接讀正文

Group Relative Policy Optimization(GRPO)已成為提升大語言模型推理能力的標準方法,但其訓練動力學仍主要依賴經驗描述:獎勵軌跡常以低引數函式形式擬合,所得常數缺乏機械意義,超引數選擇依賴於反覆試錯。最近,Rajat Ghosh等人在arXiv提交的論文《Predictable GRPO: A Closed-Form Model of Training Dynamics》中,提出了一個基於第一性原理的降階模型,以解析方式刻畫GRPO的訓練動力學。

該模型具有三個關鍵結果。首先,它將經驗性的單指數飽和律納入其過阻尼極限,將擬合的平臺、時間尺度和尺寸指數重新解釋為底層勢能的固定點、逆剛度和曲率縮放指數,並透過保留的慣性項新增了單指數無法表示的慢啟動階段。其次,模型提供了與可獨立測量量相關的預測,而非依賴於擬合引數:確定性軌跡的群大小不變性(伴隨1/G的穩態波動)、重新整理間隔的銳利穩定性閾值,以及過阻尼到振盪的轉變。第三,模型提供了診斷工具,能夠分離獎勵曲線本身所混淆的失敗模式——獎勵駭客、優勢退化、策略集中和動力學不穩定性。

研究者在三個不同模型和兩個群大小上驗證了模型。封閉形式軌跡對訓練獎勵的擬合達到了R²≥0.91,且預測的群大小不變性不僅在獎勵曲線上成立,還在八個數學基準的分佈外轉移中保持。在平均場假設嚴格成立的受控精確降階設定中,softmax-bandit降階復現了預測的過阻尼到振盪轉變,並在獨立測量的剛度處定位了重新整理間隔穩定性閾值。

這項工作為理解和最佳化GRPO訓練提供了堅實的理論基礎,有望減少試錯成本並提升訓練穩定性。研究者指出,深度網路的完整演示留待未來工作。

展開要點與分析

文章情報

工程師進階

要點

  • 開發了GRPO訓練動力學第一性原理降階模型,包含慣性項,可描述慢啟動階段。
  • 模型預測群大小不變性、重新整理間隔穩定性閾值和過阻尼-振盪轉變。
  • 在三個模型和兩個群大小上,封閉形式軌跡擬合獎勵的R²≥0.91,群大小不變性在分佈外轉移中成立。
  • 提供區分獎勵駭客、優勢退化、策略集中和動力學不穩定性等失敗模式的診斷方法。

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