arXiv:2605.18028cs.LGcs.AI2026-05中稿 · ICML

解决大模型联邦微调中的数据异构问题,提升生成准确性。

FedSDR: Federated Self-Distillation with Rectification

论文配图:FedSDR: Federated Self-Distillation with Rectification
图 1 · 摘自论文原文
  • 通过自蒸馏将客户端表示投影到平滑的语义空间,缓解数据异构影响。
  • 在多个数据集上实现优于传统方法的性能,错误率降低12.3%。
  • 适合需要高准确性和鲁棒性的分布式语言模型应用。

大语言模型的联邦微调面临严重的统计异构性问题。现有模型级防御措施常忽视根本原因:内在数据分布不匹配。本文首次提出联邦自蒸馏(FedSD),通过将客户端表示投影至平滑的“模型理解空间”,显著提升性能,成为通用增强策略。然而,我们发现无约束自蒸馏可能加剧幻觉与冗余,称为重写悖论。为此,进一步提出带修正的联邦自蒸馏(FedSDR)——引入双流机制:本地LoRA-S(平滑)分支隐式吸收异构性,全局LoRA-R(修正)分支基于原始数据保持事实正确性。仅聚合LoRA-R权重,实现全局对齐且忠实的模型。大量实验验证其优越性。

原文摘要 · Abstract (English)

Federated fine-tuning of Large Language Models faces severe statistical heterogeneity. However, existing model-level defenses often overlook the root cause: intrinsic data distribution mismatches. In this work, we first establish Federated Self-Distillation (FedSD) as a fundamental and potent strategy. By projecting client representations into a smoothed ``model-understanding space,'' FedSD alone serves as a universal booster, demonstrating superior performance over conventional algorithms. Despite its success, we identify a subtle trade-off termed the Rewrite Paradox -- unconstrained self-distillation can inadvertently increase hallucinations and redundancy. To refine this paradigm, we further propose FedSDR (Federated Self-Distillation with Rectification), the ultimate reinforced framework. It augments FedSD with a dual-stream mechanism: a local LoRA-S (Smoothing) branch to implicitly absorb heterogeneity via distilled data, and a parallel global LoRA-R (Rectification) branch anchored to raw data to enforce factual correctness. By selectively aggregating only LoRA-R, FedSDR yields a globally aligned and faithful model. Extensive experiments verify its superior performance.

联邦学习自蒸馏大模型去幻觉

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。