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

生成、過濾、控制、重放:LLM強化學習推出策略綜合綜述

文章摘要

強化學習已成為提升大型語言模型推理能力的關鍵後訓練工具,但推出(rollout)設計常被忽視。本文提出生成-過濾-控制-重放(GFCR)生命週期分類法,將推出管道分解為四個模組化階段,並補充了可靠性、覆蓋率和成本敏感性等評估標準。透過數學、程式碼、多模態推理及智慧體技能等多領域案例驗證框架,並提供常見故障診斷索引,為構建高效、可復現的推出管道提供統一框架。

來源arXiv Machine Learning作者: Rohan Surana, Gagan Mundada, Xunyi Jiang, Chuhan Wang, Zhenwei Tang, Difan Jiao, Zihan Huang, Yuxin Xiong, Junda Wu, Sheldon Yu, Xintong Li, Raghav Jain, Nikki Kuang, Sizhe Zhou, Bowen Jin, Zhendong Chu, Tong Yu, Ryan Rossi, Kuan-Hao Huang, Jingbo Shang, Jiawei Han, Julian McAuley
生成、過濾、控制、重放:LLM強化學習推出策略綜合綜述
回報錯誤

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

查看更正說明
直接讀正文

大型語言模型(LLM)在推理能力上的突破離不開強化學習(RL)作為後訓練工具的應用。然而,在RL流程中,推出(rollout)——即從提示到終止的完整軌跡取樣,包括中間推理步驟以及可選的工具或環境互動——的設計往往被低估,卻直接決定了最佳化器學習的資料質量。近日,由Rohan Surana等22位作者聯合發表的一篇綜合綜述,對LLM強化學習中的推出策略進行了系統研究,提出了名為“生成-過濾-控制-重放”(GFCR)的生命週期分類法,為構建高效、可復現的推出管道提供了統一框架。

GFCR將推出流程分解為四個模組化階段。生成(Generate)階段負責提出候選軌跡和拓撲結構,可以採用多種取樣策略;過濾(Filter)階段透過驗證器、裁判或評論者構建中間訊號,以篩選高質量軌跡;控制(Control)階段在預算約束下決定計算分配、繼續、分支或停止,例如自適應計算分配和早期退出;重放(Replay)階段則在不更新權重的情況下跨推出保留和重用工件,包括能夠自主生成新訓練任務的自我進化課程。該分類法不依賴於特定最佳化器,因此適用於多種RL變體,如PPO、GRPO等。

為評估推出管道的效能,綜述還提出了一個標準分類法,涵蓋可靠性、覆蓋率和成本敏感性三個維度。可靠性指生成高質量解決方案的能力,覆蓋率強調探索多樣候選路徑,成本敏感性則關注計算和標註效率。這三者之間存在權衡,例如高覆蓋率可能增加過濾成本,而高可靠性可能需要更多的計算資源。論文詳細分析了這些權衡,並給出了針對不同場景的推薦配置。

在方法綜合方面,論文從可驗證獎勵RL到過程監督、基於裁判的門控、引導和樹/段推出、自適應計算分配、早期退出和部分推出、吞吐量最佳化,以及用於自我改進的重放/重組等技術進行了全面梳理。這些方法在多個領域得到了驗證:在數學推理中,GFCR框架幫助模型生成更嚴謹的證明步驟;在程式碼/SQL生成中,透過過濾和重放機制顯著提高了程式碼的準確率和可執行性;在多模態推理中,結合視覺和語言資訊的推出策略提升了複雜場景的理解能力;在工具使用代理中,自適應控制階段使得代理能夠高效呼叫外部API;在智慧體技能基準測試中,展示了技能歸納、重用和跨任務遷移的能力。

最後,綜述提供了一個診斷索引,將常見推出病理(如取樣效率低、忽視探索、計算浪費、訊號噪聲等)對映到GFCR模組和相應的緩解槓桿。例如,取樣效率低可以透過改進生成階段的先驗分佈或增加重放階段的複用率來解決;忽視探索則需要在控制階段引入探索獎勵或調整分支策略。論文還指出了構建可重現、計算高效且值得信賴的推出管道所面臨的開放挑戰,包括自動化超引數調優、多工平衡、以及防止獎勵駭客等。這項工作為LLM後訓練的社群研究和工程實踐提供了重要參考,有望推動RL在LLM中的更廣泛應用。

展開要點與分析

文章情報

工程師進階

要點

  • 提出GFCR分類法,統一描述推出管道,分為生成、過濾、控制、重放四個階段。
  • 涵蓋可驗證獎勵RL、過程監督、自適應計算分配等多種方法,適用於多種RL變體。
  • 透過數學推理、程式碼生成、工具使用等案例驗證框架,提供故障診斷索引。

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