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

過程獎勵引導的自適應樹展開用於高效多輪強化學習

文章摘要

提出PATR方法,透過過程反饋評分部分軌跡、選擇性分支、共享字首和停止退化路徑,提升多輪強化學習效率。在FrozenLake和SWE-Bench上分別提升9.3和5.0分。

來源arXiv Computational Linguistics作者: Xintong Li, Sha Li, Yuwei Zhang, Changlong Yu, Rongmei Lin, Hongye Jin, Shuyi Guan, Xin Liu, Linwei Li, Qingyu Yin, Jingbo Shang
過程獎勵引導的自適應樹展開用於高效多輪強化學習
回報錯誤

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

查看更正說明
直接讀正文

強化學習(RL)已成為訓練大語言模型(LLM)智慧體的關鍵方法,但GRPO/RLOO等流行方法依賴於獨立取樣多條完整軌跡進行優勢估計。在長週期智慧體任務中,這種均勻展開策略可能將預算浪費在無資訊的死衚衕嘗試上,而有前途的中間狀態卻得不到充分探索。由於智慧體軌跡具有多輪結構(動作與觀察交替出現),自然可將一組軌跡組織成樹形,每一輪充當探索的決策點。這種視角將有效探索問題重新定義為決定在何處分支的問題。

為此,研究者提出了過程評分器引導的自適應樹展開(PATR)——一種用於多輪智慧體RL的質量感知展開框架。PATR利用任務合適的過程反饋來評分部分軌跡,從有前途的狀態選擇性分支,重用共享字首,並保守地停止退化路徑以減少抽樣浪費。具體而言,PATR使用一個過程評分器(例如基於規則或學習到的模型)為每個部分軌跡分配一個分數,然後根據分數決定是否從該狀態繼續展開或停止。共享字首意味著多個分支可以共享已經執行過的相同動作序列,從而避免重複計算。退化路徑的停止機制則防止了在無效方向上繼續浪費資源。得到的展開組仍可與標準策略最佳化(如PPO或GRPO)相容,同時在相同訓練預算下提供更高效的探索。

研究者在FrozenLake和極具挑戰性的SWE-Bench上評估了PATR。FrozenLake是一個經典的網格世界導航任務,而SWE-Bench是一個基於真實GitHub問題的軟體工程基準,之前很少有樹展開智慧體RL方法涉足。實驗表明,PATR在SWE-Bench上效能提升高達5.0個百分點,在FrozenLake上提升9.3個百分點,凸顯了過程引導的樹展開作為可擴充套件多輪RL的有效策略。這項工作為智慧體訓練的樣本效率提升開闢了新方向,有望在更復雜的現實任務中得到應用。

展開要點與分析

文章情報

工程師中級

要點

  • 現有GRPO/RLOO等方法均勻取樣完整軌跡,導致預算浪費在無資訊死衚衕,忽視有前途的中間狀態。
  • PATR利用任務相關過程反饋評分部分軌跡,從有前途狀態分支,共享字首,停止退化路徑。
  • 在FrozenLake和SWE-Bench上驗證,PATR效能分別提升9.3和5.0分。

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