arXiv:2501.00522cs.CLcs.AI2025-01被引 3

用简化语言环境训练小模型,提升效率并降低资源消耗。

TinyHelen's First Curriculum: Training and Evaluating Tiny Language Models in a Simpler Language Environment

  • 构建去噪精简的数据流水线,保留文本特征同时减少复杂度
  • 小模型在简化数据上训练后,指令遵循能力优于原数据集
  • 适合研究高效训练、小模型优化及教学实验的团队使用

由于大规模数据集和模型带来的成本上升,语言模型(LM)及其应用智能体的训练与评估愈发昂贵,测试失败代价高昂。简化语言环境作为基础训练与测试场景,保留了基本常识与沟通能力,但形式更易处理,有望提升语言模型的学习效率,从而降低有效训练与评估所需的模型规模和数据量。在这些环境中,小模型、小数据集和小代理的工作策略可能可迁移至复杂环境中的大模型。为此,我们聚焦两点:一、最小化语言数据集的噪声与复杂度;二、保持关键文本分布特征。不同于以往方法,我们提出一个数据精炼流水线,通过消除噪声、缩减词汇量并保留类型特定模式(如书籍、对话、代码等)。利用大模型实施该流程,构建了一套轻量级训练与评估数据集:7100万条的Leaner-Pretrain、700万条的Leaner-Instruct、用于评估语言能力的Leaner-Glue,以及用于测试指令遵循能力的Leaner-Eval。实验表明,轻量预训练显著提升学习效率。在不同语言粒度下,基于简化数据训练的小模型在指令遵循任务中表现优于原始数据训练的模型。此外,Leaner-Pretrain与传统大模型训练集对齐,支持对学习目标、模型架构与训练技术如何影响语言建模与下游任务性能进行资源优化分析。代码与数据已开源于https://github.com/EmpathYang/TinyHelen.git。

原文摘要 · Abstract (English)

Training language models (LMs) and their application agents is increasingly costly due to large datasets and models, making test failures difficult to bear. Simplified language environments serve as primordial training and testing grounds, retaining essential commonsense and communication skills but in a more digestible form, potentially enhancing the learning efficiency of LMs, and thus reducing the required model size and data volume for effective training and evaluation. In these simplified language environments, workable strategies for small models, datasets, and agents may be adaptable to larger models, datasets, and agents in complex language environments. To create such environments, we focus on two aspects: i) minimizing language dataset noise and complexity, and ii) preserving the essential text distribution characteristics. Unlike previous methods, we propose a pipeline to refine text data by eliminating noise, minimizing vocabulary, and maintaining genre-specific patterns (e.g., for books, conversation, code, etc.). Implementing this pipeline with large LMs, we have created a leaner suite of LM training and evaluation datasets: 71M Leaner-Pretrain, 7M Leaner-Instruct, Leaner-Glue for assessing linguistic proficiency, and Leaner-Eval for testing instruction-following ability. Our experiments show that leaner pre-training boosts LM learning efficiency. Tiny LMs trained on these datasets outperform those trained on original datasets in instruction-following across different language granularity levels. Moreover, the Leaner-Pretrain dataset's alignment with conventional large LM training sets enables resource-optimized analysis of how learning objectives, model architectures, and training techniques impact performance on language modeling and downstream tasks. Our code and datasets are available at https://github.com/EmpathYang/TinyHelen.git.

小模型训练数据精简指令遵循高效训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。