提出Olmix框架,解决语言模型训练中数据混合的动态调整难题。
Olmix: A Framework for Data Mixing Throughout LM Development
- 通过实证研究确定高效数据混合的设计关键点。
- 在5次真实数据集更新中,仅用26%计算量达到全重算性能。
- 适合需要持续迭代训练的大规模语言模型团队使用。
数据混合——即不同领域数据的比例分配——是训练语言模型的核心问题。现有方法在实际开发中表现不足。本文提出Olmix框架,解决两大挑战:其一,系统性地研究了混合方法的设计空间,识别出影响效果的关键设计选择;其二,针对实际开发中数据集频繁增删、拆分和修订导致领域集合动态变化的问题,提出混合比例复用机制,仅对受影响领域重新计算比例,其余沿用历史结果。在模拟真实开发流程的五次领域集更新中,该机制相比每次全量重算节省74%计算资源,且下游任务性能比不混合提升11.6%。
原文摘要 · Abstract (English)
Data mixing -- determining the ratios of data from different domains -- is a first-order concern for training language models (LMs). While existing mixing methods show promise, they fall short when applied during real-world LM development. We present Olmix, a framework that addresses two such challenges. First, the configuration space for developing a mixing method is not well understood -- design choices across existing methods lack justification or consensus and overlook practical issues like data constraints. We conduct a comprehensive empirical study of this space, identifying which design choices lead to a strong mixing method. Second, in practice, the domain set evolves throughout LM development as datasets are added, removed, partitioned, and revised -- a problem setting largely unaddressed by existing works, which assume fixed domains. We study how to efficiently recompute the mixture after the domain set is updated, leveraging information from past mixtures. We introduce mixture reuse, a mechanism that reuses existing ratios and recomputes ratios only for domains affected by the update. Over a sequence of five domain-set updates mirroring real-world LM development, mixture reuse matches the performance of fully recomputing the mix after each update with 74% less compute and improves over training without mixing by 11.6% on downstream tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。