arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.22293cs.CLcs.AIcs.LG
收藏

TIPS: 用于搜索增强的大语言模型的回合级信息潜力奖励塑造

TIPS: Turn-Level Information-Potential Reward Shaping for Search-Augmented LLMs

UC San Diego(加州大学圣地亚哥分校);AWS AI(亚马逊人工智能)

Yutao Xie, Nathaniel Thomas, Nicklas Hansen, Yang Fu, Li Erran Li, Xiaolong Wang

AI总结:

本文提出TIPS方法,通过回合级信息潜力奖励塑造提升搜索增强LLM的训练稳定性与性能,优于GRPO/PPO基线,在多个问答基准上取得显著提升。

摘要:

搜索增强的大语言模型(LLM)通过强化学习(RL)训练在开放领域问答(QA)中取得了强劲成果,但训练仍面临重大挑战。优化通常不稳定,因为奖励稀疏且在推理和工具调用之间难以分配信用。为此,我们引入回合级信息潜力奖励塑造(TIPS),一种简单的框架,根据教师模型下正确答案的增加可能性,为每个推理+工具调用段分配密集的回合级奖励。通过利用基于潜力的奖励塑造,TIPS提供细粒度且与策略无关的指导,克服了仅基于结果的优化的限制。在七个QA基准上评估,TIPS一致优于GRPO/PPO基线,并显著提高了训练稳定性。例如,使用Qwen-2.5 7B Instruct模型,TIPS相对于PPO提高了平均精确匹配分数11.8%和F1分数13.6%。我们的结果表明,回合级信息潜力奖励塑造为多轮LLM推理的稀疏奖励信用分配提供了一种有效且通用的解决方案。

英文摘要:

Search-augmented large language models (LLMs) trained with reinforcement learning (RL) have achieved strong results on open-domain question answering (QA), but training still remains a significant challenge. The optimization is often unstable due to sparse rewards and difficult credit assignments across reasoning and tool calls. To address this, we introduce Turn-Level Information Potential Reward Shaping (TIPS), a simple framework that assigns dense, turn-level rewards to each reasoning + tool-call segment based on the increased likelihood of the correct answer under a teacher model. By leveraging the potential-based reward shaping, TIPS offers fine-grained and policy-invariant guidance that overcomes the limitations of outcome-only optimization. Evaluated on seven QA benchmarks, TIPS consistently outperforms GRPO/PPO baselines and substantially improves training stability. For instance, with a Qwen-2.5 7B Instruct model, TIPS improves the average Exact Match score by 11.8% and F1 by 13.6% relative to PPO. Our results demonstrate that turn-level information-potential reward shaping provides an effective and general solution to sparse-reward credit assignment for multi-turn LLM reasoning.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑