用无标签测试数据让大模型自我进化,性能提升超2倍
TTRL: Test-Time Reinforcement Learning

- 利用测试时投票机制生成奖励信号,实现无监督强化学习
- 在AIME 2024数学题上使模型性能提升211%(pass@1)
- 适合无标注数据环境下模型持续优化,尤其适用于推理阶段改进
本文研究大语言模型在无显式标签的推理任务中进行强化学习的方法。核心挑战是在推理阶段无法获取真实答案的情况下进行奖励估计。尽管该设定看似难以突破,我们发现测试时扩展(TTS)中的常见做法,如多数投票(maj@n),可产生有效奖励信号,适用于强化学习训练。为此,本文提出测试时强化学习(TTRL),一种基于无标签数据训练大模型的新方法。TTRL通过利用预训练模型的先验知识,实现大模型的自演化。实验表明,TTRL在多种任务和模型上均显著提升性能。特别地,在AIME 2024数学任务中,Qwen-2.5-Math-7B的pass@1性能提升约211%,仅使用无标签测试数据。此外,尽管仅以maj@n为监督目标,TTRL性能仍持续超越初始模型的maj@n上限,并逼近直接使用带标签测试数据训练模型的水平。实验验证了TTRL在多任务上的通用有效性,凸显其在更广泛任务与领域中的潜力。
原文摘要 · Abstract (English)
This paper investigates Reinforcement Learning (RL) on data without explicit labels for reasoning tasks in Large Language Models (LLMs). The core challenge of the problem is reward estimation during inference while not having access to ground-truth information. While this setting appears elusive, we find that common practices in Test-Time Scaling (TTS), such as majority voting, yield surprisingly effective rewards suitable for driving RL training. In this work, we introduce Test-Time Reinforcement Learning (TTRL), a novel method for training LLMs using RL on unlabeled data. TTRL enables self-evolution of LLMs by utilizing the priors in the pre-trained models. Our experiments demonstrate that TTRL consistently improves performance across a variety of tasks and models. Notably, TTRL boosts the pass@1 performance of Qwen-2.5-Math-7B by approximately 211% on the AIME 2024 with only unlabeled test data. Furthermore, although TTRL is only supervised by the maj@n metric, TTRL has demonstrated performance to consistently surpass the upper limit of the initial model maj@n, and approach the performance of models trained directly on test data with ground-truth labels. Our experimental findings validate the general effectiveness of TTRL across various tasks and highlight TTRL's potential for broader tasks and domains. GitHub: https://github.com/PRIME-RL/TTRL
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。