动态选择数据子集,加速大模型训练且不降性能
GRACE: A Dynamic Coreset Selection Framework for Large Language Model Optimization
- 基于图传播与梯度重要性动态构建数据核心集
- 在三个基准上提升训练效率并保持下游性能
- 适合大规模语言模型训练优化场景
大语言模型(LLMs)在自然语言理解与生成方面表现出色,但其庞大的参数量和复杂的Transformer架构导致训练资源消耗巨大、计算复杂度高,难以在大规模数据集上高效优化。为降低训练成本同时保持性能,研究者探索了核心集选择技术,旨在从全量训练数据中挑选小而具代表性的子集以加速训练。然而,现有方法无法适应大模型训练的动态特性,且在模型规模下可扩展性差。为此,本文提出一种图引导的自适应动态核心集选择框架GRACE。GRACE通过结合表示多样性与基于梯度的重要性度量,动态构建并更新核心集,确保数据的丰富性与信息量。为降低频繁更新的开销,采用k-NN图传播机制,仅选择性地更新得分与嵌入表示,适应训练过程中的动态变化。在三个基准上的大量实验表明,GRACE显著提升了训练效率,并在多种大模型与任务中保持或提升下游性能。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have demonstrated remarkable capabilities in natural language understanding and generation. However, their immense number of parameters and complex transformer-based architectures result in significant resource demands and computational complexity during training, making it challenging to optimize them efficiently on large datasets. To reduce training costs while preserving performance, researchers have investigated coreset selection techniques, which aim to identify small, representative subsets of the entire training dataset to accelerate LLM training. However, existing coreset selection methods fail to adapt to the dynamic nature of LLM training and often struggle with scalability for models of this size. To address these limitations, we propose a graph-guided adaptive and dynamic coreset selection framework for LLMs, namely GRACE. GRACE dynamically constructs and updates coresets by combining representation diversity with gradient-based importance metrics, ensuring both informativeness and efficiency. To mitigate the computational cost of frequent updates, GRACE leverages a $k$-NN graph-based propagation mechanism and selectively updates scores and embeddings, adapting to evolving training dynamics. Extensive experiments on three benchmarks demonstrate that GRACE significantly improves training efficiency and downstream performance across diverse LLMs and tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。