FedSKD: Aggregation-free Model-heterogeneous Federated Learning via Multi-dimensional Similarity Knowledge Distillation for Medical Image Classification
About
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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Autism Spectrum Disorder Classification | FedASD Local (test) | AUC (Europe)79.8 | 17 | |
| Skin lesion classification | FedSkin Local alpha=1 (test) | Accuracy88.41 | 17 | |
| Skin lesion classification | FedSkin Global alpha=1 (test) | Mean Classification Score84.59 | 17 | |
| Skin lesion classification | FedSkin alpha=0.5 (test) | Metric M189.27 | 17 | |
| Skin lesion classification | FedSkin Global alpha=0.5 (test) | Mean Metric83.64 | 17 | |
| Autism Spectrum Disorder Classification | FedASD Global (test) | Mean AUC67.55 | 17 | |
| 10-class classification | FedCIFAR-10 Local Dir(0.5) (test) | M1 AUC97.99 | 17 | |
| 10-class classification | FedCIFAR-10 Global Dir(0.5) (test) | Mean AUC98.19 | 17 |