Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Explainable Token-level Noise Filtering for LLM Fine-tuning Datasets

About

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.

Yuchen Yang, Wenze Lin, Enhao Huang, Zhixuan Chu, Hongbin Zhou, Lan Tao, Yiming Li, Zhan Qin, Kui Ren• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningGSM8K (test)
Accuracy71.8
954
Commonsense ReasoningHellaSwag
HellaSwag Accuracy54.05
711
Mathematical ReasoningMATH 500
Accuracy77.2
442
Question AnsweringPubMedQA (test)
Accuracy55.7
170
Code GenerationHumanEval
HumanEval Score73.61
128
Financial Question AnsweringFiQA
Accuracy45.6
85
General Capability EvaluationGeneral Capability Suite MMLU, GSM8K, HumanEval, IFEval
Common Average Score69.84
39
General Capability EvaluationGeneral Capability Suite ARC-C, HellaSwag, MMLU, GSM8K
ARC-C Accuracy52.56
27
Science Question AnsweringARC-C
Accuracy (ARC-C)48.56
25
Multi-task Language UnderstandingMMLU
Accuracy55.8
15
Showing 10 of 12 rows

Other info

Follow for update