研究测试时扩展中训练数据的作用,发现数据质量决定模型能否通过增加计算提升推理能力。
Understanding the Role of Training Data in Test-Time Scaling
- 用线性回归任务建模,分析训练数据与测试时计算的关系
- 训练数据不足时,额外计算反而降低性能,存在性能拐点
- 数据多样、相关且难度适中的任务集最有利于测试时扩展
测试时扩展通过分配更多计算生成更长的思维链(CoTs),提升大语言模型的推理能力,使模型能分解复杂问题、回溯纠错。尽管表现优异(如OpenAI o1和DeepSeek R1),但长思维链在何种训练数据条件下出现、何时能提升性能仍不明确。本文研究基于上下文权重预测任务训练的Transformer模型在测试时扩展的表现。理论分析揭示:在固定测试误差下,增加测试时计算可减少训练提示中的上下文长度;若下游任务所需技能未充分存在于训练数据中,增加测试时计算反而损害性能;我们通过特征协方差矩阵的最小特征值刻画任务难度,表明在多样、相关且具挑战性的任务集上训练,最有利于测试时扩展。实验验证了这些结论在大型非线性Transformer架构上的有效性。
原文摘要 · Abstract (English)
Test-time scaling improves the reasoning capabilities of large language models (LLMs) by allocating extra compute to generate longer Chains-of-Thoughts (CoTs). This enables models to tackle more complex problem by breaking them down into additional steps, backtracking, and correcting mistakes. Despite its strong performance--demonstrated by OpenAI's o1 and DeepSeek R1, the conditions in the training data under which long CoTs emerge, and when such long CoTs improve the performance, remain unclear. In this paper, we study the performance of test-time scaling for transformers trained on an in-context weight prediction task for linear regression. Our analysis provides a theoretical explanation for several intriguing observations: First, at any fixed test error, increasing test-time compute allows us to reduce the number of in-context examples (context length) in training prompts. Second, if the skills required to solve a downstream task are not sufficiently present in the training data, increasing test-time compute can harm performance. Finally, we characterize task hardness via the smallest eigenvalue of its feature covariance matrix and show that training on a diverse, relevant, and hard set of tasks results in best performance for test-time scaling. We confirm our findings with experiments on large, nonlinear transformer architectures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。