Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Ziqiao Weng, Weidong Cai, Bo Zhou• 2025

Related benchmarks

TaskDatasetResultRank
Autism Spectrum Disorder ClassificationFedASD Local (test)
AUC (Europe)79.8
17
Skin lesion classificationFedSkin Local alpha=1 (test)
Accuracy88.41
17
Skin lesion classificationFedSkin Global alpha=1 (test)
Mean Classification Score84.59
17
Skin lesion classificationFedSkin alpha=0.5 (test)
Metric M189.27
17
Skin lesion classificationFedSkin Global alpha=0.5 (test)
Mean Metric83.64
17
Autism Spectrum Disorder ClassificationFedASD Global (test)
Mean AUC67.55
17
10-class classificationFedCIFAR-10 Local Dir(0.5) (test)
M1 AUC97.99
17
10-class classificationFedCIFAR-10 Global Dir(0.5) (test)
Mean AUC98.19
17
Showing 8 of 8 rows

Other info

Follow for update