Domain Sensitive Federated Learning with Fisher-Informed Pruning
Chenchen Lin, Wenhao Yuan, Zhengji Xu, Xuehe Wang
Abstract
Federated Learning (FL) serves as a prominent distributed training paradigm, enabling devices to collaboratively train a shared model with local data. However, in practice, clients generally possess data from multiple distributional domains, posing significant challenges to efficiency and generalization. In this paper, we propose a domainsensitive federated pruning framework, FEDFIP, that preserves domain-invariant structures while retaining domainspecific representations. The Domain-Sensitive Fisher Pruning (DSFP) module estimates channel importance per domain via Fisher information, and uploads this signal to the server to obtain a globally shared pruning mask. Given local domain heterogeneity, each client reuses its Fisher information to selectively reactivate domain-specific channels, yielding personalized sparse models that remain structurally aligned yet adapt to local heterogeneity. To further enhance performance, we adopt a Domain-Sensitive Regularization (DSR) module: the server builds domain prototypes from uploaded importance signals and broadcasts them back. Guided by the domain prototypes, we adopt a structure-contrastive loss to strengthen intra-domain consistency and inter-domain discriminability. We propose a structure-aware aggregation algorithm that fuses heterogeneous personalized architectures into a domain-generalized global model. Experiments on multi-domain benchmarks demonstrate that FEDFIP surpasses state-of-the-art FL baselines while substantially shrinking model size.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 926dee8a-2212-4aa2-bfe5-f7f35792efe8Builds on26
- SCAFFOLD: Stochastic Controlled Averaging for Federated LearningSai Praneeth Karimireddy, Satyen Kale, Mehryar Mohri, Sashank J. Reddi et al.ICML 2020 · 3,875 citations
- FedBABU: Toward Enhanced Representation for Federated Image ClassificationJaehoon Oh, Sangmook Kim, Se-Young YunICLR 2022 · 332 citations
- Federated Learning with Label Distribution Skew via Logits CalibrationJie Zhang, Zhiqi Li, Bo Li, Jianghe Xu et al.ICML 2022 · 221 citations
- FedSR: A Simple and Effective Domain Generalization Method for Federated LearningA. Tuan Nguyen, Philip H. S. Torr, Ser Nam LimNeurIPS 2022 · 153 citations
- FedDAT: An Approach for Foundation Model Finetuning in Multi-Modal Heterogeneous Federated LearningHaokun Chen, Yao Zhang, Denis Krompass, Jindong Gu et al.AAAI 2024 · 105 citations
Related papers
- DapperFL: Domain Adaptive Federated Learning with Model Fusion Pruning for Edge DevicesYongzhe Jia, Xuyun Zhang, Hongsheng Hu, Kim-Kwang Raymond Choo et al.NeurIPS 2024 · 14 citations
- Federated Manifold Learning (FML): Tackling Domain Heterogeneity with Structural Knowledge TransferXutong Mu, Yanbiao Ma, Jia Shi, Xueli Geng et al.ICML 2026
- FedDAP: Domain-Aware Prototype Learning for Federated Learning under Domain ShiftHuy Q. Le, Loc X. Nguyen, Yu Qiao, Seong Tae Kim et al.CVPR 2026
- Taming Cross-Domain Representation Variance in Federated Prototype Learning with Heterogeneous Data DomainsLei Wang, Jieming Bian, Letian Zhang, Chen Chen et al.NeurIPS 2024 · 36 citations
- Bayesian Evidence-Driven Prototype Evolution for Federated Domain AdaptationXiaoyang Yi, Li Peng, Yuru Bao, Jian ZhangICLR 2026
