arXiv:2608.03545cs.CL2026-08被引 1

通过提示调节推理一致性,提升大模型测试时强化学习效果

Hi-TTRL: Regulating Consensus with Hints for Test-Time Reinforcement Learning

论文配图:Hi-TTRL: Regulating Consensus with Hints for Test-Time Reinforcement Learning
图 1 · 摘自论文原文
  • 用提示引导采样,动态调节推理一致性强度
  • 在多个数据集上相比标准TTRL提升性能,尤其在低一致性场景
  • 适合需要可靠推理的生成式AI任务,如复杂决策与问答

测试时强化学习(TTRL)通过多数投票构建伪标签,无需标注数据即可提升大语言模型的推理能力。然而,奖励信号对一致性强度(即同一回答在采样组中出现的频率)极为敏感。一致性过低会放大不可靠伪标签的更新,过高则导致奖励对比度下降、梯度消失。本文提出Hi-TTRL,利用提示在采样过程中调节一致性强度。该方法先估计部分采样组的一致性强度,若超出目标区间,则调用马尔可夫链蒙特卡洛(MCMC)提示采样器,基于幂变换前缀分布生成提示,通过调整幂指数实现尖锐或平坦的目标分布,引导一致性进入目标区间。实验表明,Hi-TTRL在多个数据集和模型架构上均优于标准TTRL,消融实验与一致性调节分析验证了自适应提示引导机制的有效性。

原文摘要 · Abstract (English)

Test-time reinforcement learning (TTRL) improves the reasoning capabilities of large language models without labeled data by updating the policy with pseudo-labels constructed through majority voting. While effective, the reward signal assigned from majority voting is highly sensitive to consensus strength, defined as the frequency of the most common answer within a rollout group. In TTRL, consensus strength plays a dual role: it reflects both the reliability of the pseudo-label and the distribution of advantages. Low consensus can amplify updates from unreliable pseudo-labels through disproportionately large advantages, whereas high consensus reduces reward contrast and ultimately yields vanishing gradients. In this paper, we introduce Hi-TTRL, a test-time reinforcement learning framework that utilizes hints during sampling to regulate rollout consensus strength. Hi-TTRL first estimates consensus strength from a partial rollout group. When the consensus strength falls outside a target interval, it invokes a Markov chain Monte Carlo (MCMC) hint sampler. The sampler targets the power-transformed prefix distribution and uses finite-step approximate sampling to generate rollout prefixes as hints. By tuning the power exponent, Hi-TTRL generates hints with a sharpened or flattened power target, steering rollout consensus strength toward the target interval. Experiments on multiple datasets and backbones show that Hi-TTRL consistently improves over standard TTRL, with ablations and consensus-steering analyses validating the effectiveness of adaptive hint-guided consensus regulation.

强化学习提示工程一致性调节大模型推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。