arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.21158cs.LGcs.CL
收藏

通过不确定性感知奖励实现的自进化大语言模型代理

SELAUR: Self Evolving LLM Agent via Uncertainty-aware Rewards

Johns Hopkins University(约翰霍普金斯大学);Arizona State University(亚利桑那州立大学);University of Illinois Chicago(伊利诺伊大学芝加哥分校);Purdue University(普渡大学)

Dengjia Zhang, Xiaoou Liu, Lu Cheng, Yaqing Wang, Kenton Murray, Hua Wei

AI总结:

SELAUR通过将不确定性直接融入奖励设计,提升大语言模型在多步决策中的探索效率和学习稳定性。

摘要:

大语言模型(LLMs)越来越多地被部署为多步决策代理,其中有效的奖励设计对于指导学习至关重要。尽管最近的研究探索了各种形式的奖励塑造和步骤级信用分配,但一个关键信号仍然被忽视:LLMs的内在不确定性。不确定性反映了模型的置信度,揭示了需要探索的地方,并在失败轨迹中提供有价值的学习提示。我们介绍了SELAUR:通过不确定性感知奖励实现的自进化LLM代理,这是一种将不确定性直接纳入奖励设计的强化学习框架。SELAUR将熵、最小置信度和边际度量整合到一个结合的token级不确定性估计中,提供密集的置信度对齐监督,并采用一种失败感知的奖励重塑机制,将这些不确定性信号注入步骤级和轨迹级奖励中,以提高探索效率和学习稳定性。在两个基准测试(ALFWorld和WebShop)上的实验表明,我们的方法在强基线之上一致提高了成功率。消融研究进一步展示了不确定性信号如何增强探索和鲁棒性。

英文摘要:

Large language models (LLMs) are increasingly deployed as multi-step decision-making agents, where effective reward design is essential for guiding learning. Although recent work explores various forms of reward shaping and step-level credit assignment, a key signal remains largely overlooked: the intrinsic uncertainty of LLMs. Uncertainty reflects model confidence, reveals where exploration is needed, and offers valuable learning cues even in failed trajectories. We introduce SELAUR: Self Evolving LLM Agent via Uncertainty-aware Rewards, a reinforcement learning framework that incorporates uncertainty directly into the reward design. SELAUR integrates entropy-, least-confidence-, and margin-based metrics into a combined token-level uncertainty estimate, providing dense confidence-aligned supervision, and employs a failure-aware reward reshaping mechanism that injects these uncertainty signals into step- and trajectory-level rewards to improve exploration efficiency and learning stability. Experiments on two benchmarks, ALFWorld and WebShop, show that our method consistently improves success rates over strong baselines. Ablation studies further demonstrate how uncertainty signals enhance exploration and robustness.

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

↑