跳到正文
arXiv cs.AI· Taehyeon Kim, Hyunsoo Lee, Youngsoo Jang, Moontae Lee·· 3 天前AI 评分27

RPO:通过根 token 策略优化让大推理模型自适应思考

Efficiently Learning To Reason or Not to Reason: Root-token Policy Optimization for Adaptive Thinking

AI 导读

研究提出 RPO(Root-token Policy Optimization),通过组相对奖励和组内优势仅训练初始根 token(如是否调用思考标签),让大推理模型自行决定何时推理。该方法在多个模型族和规模上实现难度感知的自适应思考,训练算力仅为先前自适应推理方法的约 2%,并大幅降低显存占用。该工作已被 ACL 2026 接收为 Oral。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:大型推理模型(LRM)通过在给出答案前外显明确的推理轨迹来取得强劲表现,但面临过度思考的挑战,即对难度不同的查询分配统一且繁重的计算量。虽然专有模型通过不透明的路由来缓解这一问题,但开源 LRM 仍缺乏一种高效机制来内化自适应推理,原因在于训练成本高昂且训练方法披露有限。为此,我们提出了 RPO(根令牌策略优化),这是一个使 LRM 能够自行决定何时推理的框架,它仅通过组相对奖励和组内优势来训练初始根令牌(例如是否调用 <think> 标签)。通过聚焦这一关键分支点,RPO 大幅降低了训练开销和显存占用。在多个模型系列和规模上,RPO 仅以先前自适应推理方法约 2% 的训练算力就学会了难度感知的自适应思考。
评论: ACL 2026 口头报告
主题: 计算与语言(cs.CL);人工智能(cs.AI)
引用为: arXiv:2610.17767 [cs.CL]
  (或 arXiv:2610.17767v1 [cs.CL] 用于此版本)
  https://doi.org/10.48550/arXiv.2610.17767

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

提交历史

来自:Taehyeon Kim [查看邮箱]
[v1] 2026年10月13日星期二 05:53:34 UTC(474 KB)

来源:arXiv cs.AI · arxiv.org