FedProIn: Mitigating Client Drift for Learnable Prototypes in Federated Medical Imaging

2026-07-05 · Harsh Kumar, Tarun Kumar Garg, Vaanathi Sundaresan arxiv

Federated learning (FL) is severely hindered by statistical heterogeneity due to variations in scanners, acquisition protocols, and patient populations. Such non-IID data induces client drift during local optimization, leading to unstable convergence and suboptimal global models when parameter-based aggregation is applied. We propose a prototype-based, influence-aware federated learning framework (FedProIn) that uses multiple learnable class prototypes to capture shared semantic structures across heterogeneous clients. We introduce feature divergence loss and prototype contrastive loss to mitigate client drift by decomposing it into feature drift and prototype drift. In addition, we propose a normalized influence aggregation strategy that adaptively weights client prototypes according to their contribution to the global representation, reducing the impact of biased or low-quality updates. Experimental results on two publicly available medical datasets, HAM10000 and Matek-19, demonstrate that FedProIn achieves accuracies of (83.5% IID, 81.1% non-IID) on HAM10000 and (96.2% IID, 95.8% non-IID) on Matek-19, respectively, outperforming existing baselines in both conditions. Our code is available at https://github.com/harsh-kmr/FedProIn.

📄 PDF Abstract BibTeX arXiv:2607.04158

Code (0)

등록된 구현이 없습니다.

Tasks

Federated Learning

FedHPro: Federated Hyper-Prototype Learning via Gradient Matching

2026-05-13 · Huan Wang, Jun Shen, Haoran Li, Zhenyu Yang 외 arxiv

Federated Learning (FL) enables collaborative training of distributed clients while protecting privacy. To enhance generalization capability in FL, prototype-based FL is in the spotlight, since shared global prototypes o…

Contrastive Learning Federated Learning

Federated Class-Incremental Learning with Hierarchical Generative Prototypes

2024-06-04 · Riccardo Salami, Pietro Buzzega, Matteo Mosconi, Mattia Verasani 외

Federated Learning (FL) aims at unburdening the training of deep models by distributing computation across multiple devices (clients) while safeguarding data privacy. On top of that, Federated Continual Learning (FCL) al…

class-incremental learning Class Incremental Learning Continual Learning Federated Learning+1

Dynamic Heterogeneous Federated Learning with Multi-Level Prototypes

2023-12-15 · Shunxin Guo, Hongsong Wang, Xin Geng

Federated learning shows promise as a privacy-preserving collaborative learning technique. Existing heterogeneous federated learning mainly focuses on skewing the label distribution across clients. However, most approach…

Federated Learning Privacy Preserving

FedAli: Personalized Federated Learning with Aligned Prototypes through Optimal Transport

2024-11-15 · Sannara Ek, Kaile Wang, François Portet, Philippe Lalanda 외

Federated Learning (FL) enables collaborative, personalized model training across multiple devices without sharing raw data, making it ideal for pervasive computing applications that optimize user-centric performances in…

Activity Recognition Federated Learning Human Activity Recognition Personalized Federated Learning

Federated Learning Under Temporal Drift -- Mitigating Catastrophic Forgetting via Experience Replay

2026-01-19 · Sahasra Kokkula, Daniel David, Aaditya Baruah arxiv

Federated Learning struggles under temporal concept drift where client data distributions shift over time. We demonstrate that standard FedAvg suffers catastrophic forgetting under seasonal drift on Fashion-MNIST, with a…