通过解析令牌级噪声,提升大模型微调数据质量
Explainable Token-level Noise Filtering for LLM Fine-tuning Datasets
- 将令牌贡献分解为推理重要性、知识新颖性、任务相关性三属性
- 在数学、代码、医学任务上最高提升13.7%性能
- 可解释性强,适合数据清洗与模型优化研究者
大语言模型在众多应用中取得显著进展,微调是将其适配到具体下游任务的关键步骤。然而,当前微调数据集多以句子为单位设计,与模型的令牌级优化机制不匹配,引入了令牌级噪声,影响最终性能。本文提出XTF框架,将令牌对微调过程的复杂贡献解耦为三个明确属性:推理重要性、知识新颖性和任务相关性,通过评分方法评估并掩码噪声令牌的梯度,从而优化微调效果。我们在数学、代码和医学三个代表性任务上,对7个主流大模型进行了广泛实验。结果表明,相较于常规微调,XTF可使下游性能提升最高达13.7%。本工作强调了令牌级数据优化的重要性,并展示了基于属性分解策略在解释复杂训练机制方面的潜力。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have seen remarkable advancements, achieving state-of-the-art results in diverse applications. Fine-tuning, an important step for adapting LLMs to specific downstream tasks, typically involves further training on corresponding datasets. However, a fundamental discrepancy exists between current fine-tuning datasets and the token-level optimization mechanism of LLMs: most datasets are designed at the sentence-level, which introduces token-level noise, causing negative influence to final performance. In this paper, we propose XTF, an explainable token-level noise filtering framework. XTF decomposes the complex and subtle contributions of token-level data to the fine-tuning process into three distinct and explicit attributes (reasoning importance, knowledge novelty, and task relevance), which can be assessed using scoring methods, and then masks the gradients of selected noisy tokens accordingly to optimize the performance of fine-tuned LLMs. We conduct extensive experiments on three representative downstream tasks (math, code and medicine) across 7 mainstream LLMs. The results demonstrate that XTF can significantly improve downstream performance by up to 13.7% compared to regular fine-tuning. Our work highlights the importance of token-level dataset optimization, and demonstrates the potential of strategies based on attribute decomposition for explaining complex training mechanisms.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。