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

ECHO: 测试时强化学习的熵-置信度混合优化

ECHO: Entropy-Confidence Hybrid Optimization for Test-Time Reinforcement Learning

Northeastern University, Shenyang, China(东北大学(沈阳));Shenzhen Campus of Sun Yat-sen University, China(中山大学深圳校区)

Chu Zhao, Enneng Yang, Yuting Liu, Jianzhe Zhao, Guibing Guo

AI总结:

针对测试时强化学习中高熵分支导致rollout崩溃和早期伪标签噪声引发过拟合的问题,提出熵-置信度混合组相对策略优化(ECHO),通过自适应分支控制和置信度剪枝缓解崩溃,并采用置信度自适应裁剪和优势塑造增强训练鲁棒性。

摘要:

测试时强化学习通过重复rollout生成多个候选答案,并利用多数投票构建的伪标签进行在线更新。为了减少开销并改进探索,先前的工作引入了树结构rollout,共享推理前缀并在关键节点分支以提高采样效率。然而,这种范式仍然面临两个挑战:(1) 高熵分支可能触发rollout崩溃,即分支预算集中在少数具有连续高熵片段的轨迹上,迅速减少有效分支数量;(2) 早期伪标签存在噪声和偏差,可能引发自我强化的过拟合,导致策略过早锐化并抑制探索。为了解决这些问题,我们提出了熵-置信度混合组相对策略优化(ECHO)。在rollout过程中,ECHO联合利用局部熵和组级置信度自适应控制分支宽度,并进一步引入在线置信度剪枝以终止持续低置信度的分支,避免高熵陷阱并缓解崩溃。在策略更新过程中,ECHO采用置信度自适应裁剪和熵-置信度混合优势塑造方法,以增强训练鲁棒性并减轻早期偏差。实验表明,ECHO在多个数学和视觉推理基准上取得了一致的性能提升,并在有限的rollout预算下更有效地泛化。

英文摘要:

Test-time reinforcement learning generates multiple candidate answers via repeated rollouts and performs online updates using pseudo-labels constructed by majority voting. To reduce overhead and improve exploration, prior work introduces tree structured rollouts, which share reasoning prefixes and branch at key nodes to improve sampling efficiency. However, this paradigm still faces two challenges: (1) high entropy branching can trigger rollout collapse, where the branching budget concentrates on a few trajectories with consecutive high-entropy segments, rapidly reducing the number of effective branches; (2) early pseudo-labels are noisy and biased, which can induce self-reinforcing overfitting, causing the policy to sharpen prematurely and suppress exploration. To address these issues, we propose Entropy Confidence Hybrid Group Relative Policy Optimization (ECHO). During rollout, ECHO jointly leverages local entropy and group level confidence to adaptively control branch width, and further introduces online confidence-based pruning to terminate persistently low confidence branches, avoiding high entropy traps and mitigating collapse. During policy updates, ECHO employs confidence adaptive clipping and an entropy confidence hybrid advantage shaping approach to enhance training robustness and mitigate early stage bias. Experiments demonstrate that ECHO achieves consistent gains on multiple mathematical and visual reasoning benchmarks, and generalizes more effectively under a limited rollout budget.

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

↑