通过训练动态识别高质量数据,实现跨私有域协作训练的高效数据筛选。
CLUES: Collaborative High-Quality Data Selection for LLMs via Training Dynamics
- 基于训练动态相似性筛选数据,利用梯度内积迹衡量数据质量。
- 在医疗、多语言、金融等异构数据上,优于现有数据选择方法。
- 支持联邦学习与模型合并,适用于数据不共享的协作场景。
近期研究强调了大规模语言模型(LLMs)中数据质量的重要性。然而,在数据孤岛间无法直接共享的协作环境中,自动化数据质量控制面临独特挑战。本文提出一种基于训练动态的数据质量控制新方法:高质量数据更可能与基准数据集保持相似的训练动态。通过计算私有数据与基准数据的样本级梯度,并以累积内积的迹作为数据质量指标,实现跨私有域的高质量数据筛选。在服务器端通过模型合并或联邦学习进行集中式模型更新,支持协作训练。此外,我们设计了针对异构领域数据的协作环境专用评估方案。实验表明,在医疗、多语言及金融等多个私有域数据集上,使用本方法筛选的高质量数据进行微调,性能普遍优于其他数据选择方法。代码已开源:github.com/Ryan0v0/CLUES。
原文摘要 · Abstract (English)
Recent research has highlighted the importance of data quality in scaling large language models (LLMs). However, automated data quality control faces unique challenges in collaborative settings where sharing is not allowed directly between data silos. To tackle this issue, this paper proposes a novel data quality control technique based on the notion of data influence on the training dynamics of LLMs, that high quality data are more likely to have similar training dynamics to the anchor dataset. We then leverage the influence of the training dynamics to select high-quality data from different private domains, with centralized model updates on the server side in a collaborative training fashion by either model merging or federated learning. As for the data quality indicator, we compute the per-sample gradients with respect to the private data and the anchor dataset, and use the trace of the accumulated inner products as a measurement of data quality. In addition, we develop a quality control evaluation tailored for collaborative settings with heterogeneous domain data. Experiments show that training on the high-quality data selected by our method can often outperform other data selection methods for collaborative fine-tuning of LLMs, across diverse private domain datasets, in medical, multilingual and financial settings. Our code is released at github.com/Ryan0v0/CLUES.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。