跳到正文
arXiv cs.AI· Faeze Moradi Kalarde, Armin Toroghi, Nam Ho Koh, Doo Hwan Hwang, Seungjin Kim, Kee-Eung Kim, Scott Sanner·· 12 小时前AI 评分31

AdaRoll-GRPO:用贝叶斯自适应 MDP 分配 rollout 预算提升 GRPO 训练

Bayes-Adaptive MDPs for Rollout Allocation to Enhance GRPO Training

AI 导读

研究者提出 AdaRoll-GRPO,将 GRPO 训练中的 rollout 预算分配建模为贝叶斯自适应 MDP 的序贯决策问题,学习按题目难度自适应分配 rollout 的策略。该策略离线计算,训练时应用不引入额外在线优化开销。在多个推理数据集和 LLM 架构上,固定 rollout 预算下 AdaRoll-GRPO 持续提升 GRPO 训练表现。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:组相对策略优化(GRPO)在 LLM 后训练中展现出强劲性能。然而,它为每个训练问题分配统一的 rollout 预算,忽略了问题难度的差异。因此,高难度问题可能获得过少的 rollout,难以生成有信息量的正奖励信号,从而限制了它们对训练的贡献;而中等难度问题往往获得超出必要的 rollout。在本工作中,我们将难度感知的 rollout 预算分配问题形式化,以在总 rollout 预算与标准 GRPO 相同的前提下最大化训练效用。我们提出 AdaRoll-GRPO,它将 rollout 分配重新表述为序贯决策问题,并利用贝叶斯自适应马尔可夫决策过程(MDP)学习自适应分配策略。所学策略离线计算,并在 GRPO 训练期间应用,不引入额外的在线优化开销。在多个推理数据集和 LLM 架构上的大量实验表明,AdaRoll-GRPO 在固定 rollout 预算下持续提升 GRPO 训练性能,凸显了自适应 rollout 分配在 LLM 后训练中对 GRPO 的有效性。
评论: 22 页,8 幅图
主题: 机器学习(cs.LG);人工智能(cs.AI)
引用为: arXiv:2610.37429 [cs.LG]
  (或 arXiv:2610.37429v1 [cs.LG] 用于此版本)
  https://doi.org/10.48550/arXiv.2610.37429

由 DataCite 颁发的 arXiv DOI(待注册)

提交历史

来自:Faeze Moradi Kalarde [查看邮箱]
[v1] 2026 年 10 月 27 日星期二 15:41:57 UTC(348 KB)

来源:arXiv cs.AI · arxiv.org