arXiv:2506.24120cs.LGcs.AI2025-06

选分布更均匀的数据,能加速大模型训练并提升效果。

Data Uniformity Improves Training Efficiency and More, with a Convergence Framework Beyond the NTK Regime

  • 通过最大化数据点间最小距离提升分布均匀性。
  • 均匀数据使训练更快,且误差随最小距离增大而降低。
  • 理论适用于Transformer等结构,适合做模型优化的研究者。

数据选择在数据驱动决策中至关重要,尤其在大语言模型中,通常依赖具体任务。虽然数据质量与多样性已被广泛研究,但尚不清楚是否存在其他可量化且通用的数据选择原则以持续提升复杂任务的性能。本文证明:选择更均匀分布的数据可提高训练效率并增强性能。具体而言,我们发现更均匀(偏差更小)的分布会增大数据点间的最小成对距离 $h_{ ext{min}}$,并证明 $h_{ ext{min}}$ 越小,梯度下降(GD)的训练动态越慢。此外,理论上表明神经网络的近似误差随 $h_{ ext{min}}$ 增大而减小。我们提出一个超越神经正切核(NTK)范式的收敛框架,适用于包括Transformer在内的多种架构,无需假设Lipschitz光滑性。该框架还为残差连接和函数复合在深层网络中的使用提供了理论支持。最后,在不同优化策略、模型规模与训练数据集上进行的全面实验表明,通过最大化成对距离选择数据,能显著加速训练,并在各类数据集上达到相当或更优的性能。代码与数据集见:https://github.com/SafeRL-Lab/data-uniformity。

原文摘要 · Abstract (English)

Data selection plays a crucial role in data-driven decision-making, including in large language models (LLMs), and is typically task-dependent. Properties such as data quality and diversity have been extensively studied and are known to enhance model performance. However, it remains unclear whether there exist other quantitative and general principles of data selection that can consistently improve performance, especially for complicated tasks. In this paper, we demonstrate that selecting more uniformly distributed data can improve training efficiency while enhancing performance. Specifically, we establish that more uniform (less biased) distribution leads to a larger minimum pairwise distance between data points, denoted by $h_{\min}$, and prove that a smaller $h_{\min}$ can slow down the training dynamics of gradient descent (GD). Moreover, we theoretically show that the approximation error of neural networks decreases as $h_{\min}$ increases. Our analysis introduces a convergence framework for GD beyond the Neural Tangent Kernel (NTK) regime, applicable to a broad class of architectures, including transformers, without requiring Lipschitz smoothness. This framework further provides theoretical justification for the use of residual connection and function composition in deep neural architectures. In the end, we conduct comprehensive experiments for supervised fine-tuning across various settings, including different optimization strategies, model sizes, and training datasets. The results consistently demonstrate that selecting data by maximizing pairwise distance significantly accelerates training and achieves comparable or better performance in LLMs across diverse datasets. Code and Datasets are available at the link: https://github.com/SafeRL-Lab/data-uniformity.

数据选择训练加速大模型优化理论分析

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