通过多目标分块优化数据布局,显著提升长文本大模型训练效果。
DataSculpt: Crafting Data Landscapes for Long-Context LLMs through Multi-Objective Partitioning
- 将数据组织建模为多目标优化问题,分粗细两阶段聚类与拼接。
- 在检索增强、摘要、阅读理解等任务上分别提升18.09%~21.27%。
- 适合需要强化长文本处理能力的LLM训练场景。
近年来,大语言模型(LLMs)在多项任务中展现出显著进步,其中长上下文能力尤为关键。提升长上下文性能的核心在于有效整合多领域数据并优化训练时的上下文窗口。通过大量实验分析,我们识别出三个核心挑战:跨领域的长文档稀缺、上下文窗口的有效构建以及大规模数据集的高效组织。为此,我们提出DataSculpt,一种专为长上下文训练设计的数据管理框架。该框架将训练数据组织建模为多目标组合优化问题,关注相关性、同质性、完整性与效率等属性。采用从粗到细的方法:先基于语义相似性聚类(粗粒度),再在每个簇内进行多目标贪心搜索,评估并拼接文档形成不同上下文窗口(细粒度)。全面评估表明,DataSculpt显著提升长上下文训练表现,在检索增强、摘要、阅读理解任务上分别提升18.09%、21.23%、21.27%,代码补全提升3.81%,同时整体模型性能提升4.88%。
原文摘要 · Abstract (English)
In recent years, Large Language Models (LLMs) have demonstrated significant improvements across a variety of tasks, one of which is the long-context capability. The key to improving long-context performance lies in effective data organization and management strategies that integrate data from multiple domains and optimize the context window during training. Through extensive experimental analysis, we identified three key challenges in designing effective data management strategies that enable the model to achieve long-context capability without sacrificing performance in other tasks: (1) a shortage of long documents across multiple domains, (2) effective construction of context windows, and (3) efficient organization of large-scale datasets. To address these challenges, we introduce DataSculpt, a novel data management framework designed for long-context training. We first formulate the organization of training data as a multi-objective combinatorial optimization problem, focusing on attributes including relevance, homogeneity, integrity, and efficiency. Specifically, our approach utilizes a coarse-to-fine methodology to optimize training data organization both efficiently and effectively. We begin by clustering the data based on semantic similarity (coarse), followed by a multi-objective greedy search within each cluster to score and concatenate documents into various context windows (fine). Our comprehensive evaluations demonstrate that DataSculpt significantly enhances long-context training performance, resulting in improvements of 18.09% in retrieval augmentation, 21.23% in summarization, 21.27% in reading comprehension, and a 3.81% increase in code completion, while also maintaining overall model proficiency with a 4.88% improvement.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。