Test-Time Training offers a promising way to improve the reasoning ability of large language models (LLMs) by adapting the model using only the test questions. However, existing methods struggle with difficult reasoning problems for two reasons: raw test questions are often too difficult to yield high-quality pseudo-labels, and the limited size of test sets makes continuous online updates prone to instability. To address these limitations, we propose TTCS, a co-evolving test-time training framework. Specifically, TTCS initializes two policies from the same pretrained model: a question synthesizer and a reasoning solver. These policies evolve through iterative optimization: the synthesizer generates progressively challenging question variants conditioned on the test questions, creating a structured curriculum tailored to the solver's current capability, while the solver updates itself using self-consistency rewards computed from multiple sampled responses on both original test and synthetic questions. Crucially, the solver's feedback guides the synthesizer to generate questions aligned with the model's current capability, and the generated question variants in turn stabilize the solver's test-time training. Experiments show that TTCS consistently strengthens the reasoning ability on challenging mathematical benchmarks and transfers to general-domain tasks across different LLM backbones, highlighting a scalable path towards dynamically constructing test-time curricula for self-evolving. Our code and implementation details are available at https://github.com/XMUDeepLIT/TTCS.
翻译:测试时训练通过仅利用测试问题来适应模型,为提高大语言模型的推理能力提供了一种有前景的途径。然而,现有方法在处理困难推理问题时面临两个主要挑战:原始测试问题往往过于困难,难以产生高质量的伪标签;同时,测试集规模有限,使得连续的在线更新容易不稳定。为克服这些局限性,我们提出了TTCS,一个协同进化的测试时训练框架。具体而言,TTCS从同一个预训练模型初始化两个策略:一个题目合成器和一个推理求解器。这些策略通过迭代优化共同进化:合成器以测试问题为条件,生成逐步具有挑战性的问题变体,从而为求解器当前的能力量身定制一个结构化的课程;同时,求解器利用在原始测试问题和合成问题上通过多次采样响应计算出的自洽性奖励来更新自身。关键的是,求解器的反馈指导合成器生成与模型当前能力相匹配的问题,而生成的问题变体反过来又稳定了求解器的测试时训练。实验表明,TTCS持续增强了在不同大语言模型骨干上,于具有挑战性的数学基准测试上的推理能力,并能迁移到通用领域任务,这突显了一条为自进化动态构建测试时课程的可扩展路径。我们的代码和实现细节可在 https://github.com/XMUDeepLIT/TTCS 获取。