用思维轨迹增强文本数据,让大模型更高效地学习。
Thinking Augmented Pre-training
- 用自动生成的思维过程扩展原始文本数据
- 在1000亿token上训练,数据效率提升3倍
- 适合想用少量高质量数据提升模型性能的研究者
本文提出一种简单且可扩展的方法——思维增强预训练(TPT),通过为现有文本数据添加自动生成的思维轨迹来提升大语言模型训练的数据效率。随着大模型预训练算力快速增长,高质量数据却仍稀缺,而某些高价值词元因背后逻辑复杂难以学习。TPT通过逐步推理和分解,使高价值词元更易被模型掌握。该方法在高达1000亿词元的不同训练配置中验证,涵盖受限与充足数据场景,以及从强开源检查点继续训练。实验表明,TPT显著提升多种模型规模和架构下的表现。对于30亿参数模型,在多个挑战性推理基准上,微调后性能提升超过10%;整体预训练数据效率提升达3倍。
原文摘要 · Abstract (English)
This paper introduces a simple and scalable approach to improve the data efficiency of large language model (LLM) training by augmenting existing text data with thinking trajectories. The compute for pre-training LLMs has been growing at an unprecedented rate, while the availability of high-quality data remains limited. Consequently, maximizing the utility of available data constitutes a significant research challenge. A primary impediment is that certain high-quality tokens are difficult to learn given a fixed model capacity, as the underlying rationale for a single token can be exceptionally complex and deep. To address this issue, we propose Thinking augmented Pre-Training (TPT), a universal methodology that augments text with automatically generated thinking trajectories. Such augmentation effectively increases the volume of the training data and makes high-quality tokens more learnable through step-by-step reasoning and decomposition. We apply TPT across diverse training configurations up to $100$B tokens, encompassing pre-training with both constrained and abundant data, as well as mid-training from strong open-source checkpoints. Experimental results indicate that our method substantially improves the performance of LLMs across various model sizes and families. Notably, TPT enhances the data efficiency of LLM pre-training by a factor of $3$. For a $3$B parameter model, it improves the post-training performance by over $10\%$ on several challenging reasoning benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。