FedSKD: Aggregation-free Model-heterogeneous Federated Learning using Multi-dimensional Similarity Knowledge Distillation
Weng, Ziqiao, Cai, Weidong, Zhou, Bo
–arXiv.org Artificial Intelligence
Federated learning (FL) enables privacy-preserving collaborative model training without direct data sharing. Model-heterogeneous FL (MHFL) extends this paradigm by allowing clients to train personalized models with heterogeneous architectures tailored to their computational resources and application-specific needs. However, existing MHFL methods predominantly rely on centralized aggregation, which introduces scalability and efficiency bottlenecks, or impose restrictions requiring partially identical model architectures across clients. While peer-to-peer (P2P) FL removes server dependence, it suffers from model drift and knowledge dilution, limiting its effectiveness in heterogeneous settings. To address these challenges, we propose FedSKD, a novel MHFL framework that facilitates direct knowledge exchange through round-robin model circulation, eliminating the need for centralized aggregation while allowing fully heterogeneous model architectures across clients. FedSKD's key innovation lies in multi-dimensional similarity knowledge distillation, which enables bidirectional cross-client knowledge transfer at batch, pixel/voxel, and region levels for heterogeneous models in FL. This approach mitigates catastrophic forgetting and model drift through progressive reinforcement and distribution alignment while preserving model heterogeneity. Extensive evaluations on fMRI-based autism spectrum disorder diagnosis and skin lesion classification demonstrate that FedSKD outperforms state-of-the-art heterogeneous and homogeneous FL baselines, achieving superior personalization (client-specific accuracy) and generalization (cross-institutional adaptability). These findings underscore FedSKD's potential as a scalable and robust solution for real-world medical federated learning applications.
arXiv.org Artificial Intelligence
Mar-23-2025
- Country:
- Asia > China (0.04)
- Europe
- Belgium > Flanders
- Flemish Brabant > Leuven (0.04)
- France > Grand Est
- Bas-Rhin > Strasbourg (0.04)
- Germany (0.04)
- Ireland (0.04)
- Netherlands (0.04)
- Switzerland > Zürich
- Zürich (0.14)
- Belgium > Flanders
- North America
- Canada > Quebec
- Montreal (0.04)
- United States > Illinois
- Cook County > Chicago (0.04)
- Canada > Quebec
- Oceania > Australia
- New South Wales > Sydney (0.04)
- Genre:
- Research Report > New Finding (0.88)
- Industry:
- Health & Medicine
- Diagnostic Medicine > Imaging (1.00)
- Health Care Technology (1.00)
- Therapeutic Area
- Dermatology (0.89)
- Neurology > Autism (0.69)
- Health & Medicine
- Technology: