TTPO:测试时策略优化

Hugging Face Daily Papers 论文

摘要

TTPO引入了一种无标签的测试时训练方法,用于数学推理,通过非对称处理一致和不一致的运行轨迹,达到了与监督学习相当的性能。

最近一些突出的后训练方法,如强化学习(RL)和基于策略的自蒸馏(OPSD),推动了大型语言模型在数学推理方面的快速进展,但它们对真实标签的依赖阻碍了测试时训练(TTT)。用多数投票的伪标签替代真实标签是一个自然的替代方案,但它很脆弱:错误的投票会破坏教师模型并误导每个令牌。我们观察到,这种失败模式是非对称的:与伪标签不一致的运行轨迹通常是错误的,无论投票本身是否正确。基于这一观察,我们提出了测试时策略优化(TTPO),这是一个非对称目标,通过OPSD蒸馏一致的运行轨迹,并使用分组强化学习惩罚不一致的运行轨迹。令牌级别的选择进一步细化了两个分支:蒸馏降低了已收敛位置的权重,而强化学习仅惩罚置信错误。即使在频繁的伪标签错误下,这两个更新仍然保持良好基础,并且随着模型的改进,多数投票路由产生更紧密的自我监督。没有任何标签,TTPO在五个竞赛级基准测试中与标签监督的OPSD相匹配,在测试时训练中将Qwen3-1.7B从38.0%提升到45.2%,在不思考的情况下产生+25.2%到+36.4%的增益,并显示出强大的跨任务泛化能力。
查看原文
查看缓存全文

缓存时间: 2026/08/28 07:24

论文页 - TTPO: 测试时策略优化

来源: https://huggingface.co/papers/2608.27448

摘要

测试时策略优化通过非对称地蒸馏一致的推理路径并惩罚不一致的路径,实现了无需标签的测试时训练以增强数学推理能力,其性能可与监督学习方法相媲美。

近期的强化学习 (https://huggingface.co/papers?q=Reinforcement%20Learning) 和在策略自蒸馏 (https://huggingface.co/papers?q=On-Policy%20Self-Distillation) 等前沿后训练方法推动了大语言模型在数学推理方面的快速发展,但它们对真实标签的依赖使得测试时训练 (https://huggingface.co/papers?q=test-time%20training) 成为可能。用多数投票伪标签 (https://huggingface.co/papers?q=majority-vote%20pseudo-labels) 替代真实标签是一种自然的替代方案,但该方法很脆弱:一次错误的投票会污染教师模型并误导每一个生成的词元。我们观察到这种失败模式具有不对称性:与伪标签不一致的推理路径,无论投票本身是否正确,通常都是错误的。基于这一观察,我们提出了测试时策略优化 (https://huggingface.co/papers?q=Test-Time%20Policy%20Optimization) ,这是一种非对称目标函数:它通过在策略自蒸馏中蒸馏一致的推理路径,并通过分组强化学习 (https://huggingface.co/papers?q=Grouped%20RL) 惩罚不一致的推理路径。词元级选择 (https://huggingface.co/papers?q=Token-level%20selection) 进一步优化了两个分支:蒸馏过程降低已收敛位置的权重,而强化学习仅惩罚有把握的错误。即使在伪标签频繁出错的情况下,这两种更新也能保持稳定性,并且随着模型性能提升,多数投票路由能提供更紧密的自监督。在没有任何标签的情况下,TTPO在五个竞赛级基准测试上匹配了使用标签监督的在策略自蒸馏的性能,将Qwen3-1.7B在测试时训练中的表现从38.0%提升至45.2%,在无思考模式下带来了+25.2%至+36.4%的提升,并展现出强大的跨任务泛化能力。

查看arXiv页面 (https://arxiv.org/abs/2608.27448) 查看PDF (https://arxiv.org/pdf/2608.27448) 项目页面 (https://zju-real.github.io/TTPO/) GitHub8 (https://github.com/ZJU-REAL/TTPO) 添加到收藏 (https://huggingface.co/login?next=%2Fpapers%2F2608.27448)

在你的代理中获取此论文:

hf papers read 2608\.27448

没有最新的CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash

引用此论文的模型 0

无模型链接此论文

在模型README.md中引用 arxiv\.org/abs/2608\.27448 以从此页面链接。

引用此论文的数据集 0

无数据集链接此论文

在数据集README.md中引用 arxiv\.org/abs/2608\.27448 以从此页面链接。

引用此论文的Space 0

无Space链接此论文

在Space README.md中引用 arxiv\.org/abs/2608\.27448 以从此页面链接。

包含此论文的收藏 0

无收藏包含此论文

将此论文添加到收藏 (https://huggingface.co/new-collection) 以从此页面链接。

相似文章

Planned Test-Time Scaling 与协调推理路径

arXiv cs.CL

本文提出 Planned Test-Time Scaling (PTTS),一种通过协调推理分支来提升挑战性任务性能的方法,在数学推理基准测试中相比重复采样取得了显著提升。

TPPO:面向语言模型推理的追踪值 PPO

Hugging Face Daily Papers

该论文提出 Tracking-Value PPO(TPPO),通过在对数几率空间对前缀值进行贝叶斯分解,构建累积值参数化以更好地跟踪语言模型推理中的值演化。TPPO 在 Qwen3-4B-Base 和 Qwen3-8B-Base 上分别比 PPO 提升 7.0 和 8.4 个数学推理分数点,并改善了分布外泛化能力。

测试时个性化:针对缩放失败的一种诊断框架与概率修正方法

arXiv cs.LG

本文提出了测试时个性化(TTP),这是一种通过候选采样和基于奖励的选择来扩展推理时计算,从而提升大语言模型(LLM)个性化能力的框架。该研究诊断了标准奖励模型中的失效模式,并提出了一种概率个性化奖励模型以缓解这些问题。

模块化TTT:将测试时训练重新构想为可组合模块

Hugging Face Daily Papers

本文介绍了模块化TTT,这是一个将测试时训练的内部学习器表示为有向无环图的框架,能够对组件进行系统的消融和组合。作者在1000亿个token上训练了4.1亿和14.5亿参数规模的模型,取得了与GatedDeltaNet相当的性能。