T. Garg

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

This work proposes a prototype-based, influence-aware federated learning framework (FedProIn) that uses multiple learnable class prototypes to capture shared semantic structures across heterogeneous clients and introduces feature divergence loss and prototype contrastive loss to mitigate client drift by decomposing it into feature drift and prototype drift.