English

Federated Medical Image Classification under Class and Domain Imbalance exploiting Synthetic Sample Generation

Computer Vision and Pattern Recognition 2026-04-30 v1

Abstract

Exploiting deep learning in medical imaging faces critical challenges, including strict privacy constraints, heterogeneous imaging devices with varying acquisition properties, and class imbalance due to the uneven prevalence of pathologies. In this work, we propose FedSSG, a novel Federated Learning framework that addresses domain shifts caused by diverse imaging devices while mitigating the under-representation of rare pathologies. The key contribution is a strategy for generating synthetic samples and distributing them across clients to improve coverage of both underrepresented pathologies and imaging devices. Experimental results demonstrate that our approach significantly enhances model performance and generalization across heterogeneous institutions, with minimal computational overhead at the client side.

Keywords

Cite

@article{arxiv.2604.26324,
  title  = {Federated Medical Image Classification under Class and Domain Imbalance exploiting Synthetic Sample Generation},
  author = {Martina Pavan and Matteo Caligiuri and Francesco Barbato and Pietro Zanuttigh},
  journal= {arXiv preprint arXiv:2604.26324},
  year   = {2026}
}

Comments

Accepted at ICPR 2026, 13 pages, 3 figures, 5 tables

R2 v1 2026-07-01T12:40:33.783Z