Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Data Quality Control in Federated Instruction-tuning of Large Language Models

About

Federated Learning (FL) enables privacy-preserving collaborative instruction tuning of large language models (LLMs) by leveraging massively distributed data. However, the decentralized nature of FL exacerbates data quality challenges, as local clients lack global visibility to filter noisy or low-quality samples before training. To resolve this issue, we propose FedDQC, a novel federated instruction tuning framework with dynamic data quality control. Our approach introduces two key innovations. First, we propose instruction-response alignment (IRA), an efficient client-side metric for quality evaluation requiring only low-cost inference. We validate that higher-IRA data corresponds to more relevant and easier-to-learn question-answer pairs. Second, mirroring the human easy-to-hard knowledge acquisition process, we design a quality-aware hierarchical FL training framework, where the LLM is progressively fine-tuned from high- to low-IRA data in a collaborative manner. The framework also supports adaptive data quality assessment at each hierarchy, enabling dynamic adjustments throughout the training process. Extensive experiments on synthetic and real-world datasets show that our method significantly improves LLM performance on mixed-quality data in FL.

Yaxin Du, Rui Ye, Fengting Yuchi, Wanru Zhao, Jingjing Qu, Yanfeng Wang, Siheng Chen• 2024

Related benchmarks

TaskDatasetResultRank
Algebraic Question AnsweringAQUA-RAT Synthetic NIID 1.0 (test)
Accuracy28
7
Medical Question AnsweringPubMedQA Synthetic IID 1.0 (test)
Accuracy75.1
7
Medical Question AnsweringPubMedQA Synthetic NIID 1.0 (test)
Accuracy75.1
7
Molecular Science InstructionsMol-Instructions Synthetic IID 1.0 (test)
BertScore0.819
7
Molecular Science InstructionsMol-Instructions Synthetic NIID 1.0 (test)
BertScore0.824
7
Algebraic Question AnsweringAQUA-RAT Synthetic IID 1.0 (test)
Accuracy29
7
Financial Question AnsweringFIQA Synthetic IID 1.0 (test)
Win Rate72.1
6
Financial Question AnsweringFIQA Synthetic NIID 1.0 (test)
Win Rate82.1
6
Instruction FollowingFed-WildChat Real Dataset 1.0 (test)
MT-Bench Score4.78
6
Showing 9 of 9 rows

Other info

Code

Follow for update