Federated Semi-supervised Learning for Medical Image Segmentation with Intra-client and Inter-client Consistency
Yubin Zheng, Peng Tang, Tianjie Ju, Hao Wang, Weidong Qiu, Jagath Chandana Rajapakse · 2024
Medical image segmentation plays a vital role in medical image analysis. However, it is impractical to build a large-scale centralized segmentation dataset due to the privacy of medical images. Federated learning (FL) aims to train a shared model of isolated clients without local data exchange which aligns well with the scarcity and privacy characteristics of medical images. Moreover, there is a large amount of unlabeled data in clients due to the difficulty in annotating medical images. Federated semi-supervised learning (FSSL) can leverage the unlabeled data of clients to improve the performance of the global model. Many existing FSSL methods apply the complicated semi-supervised learning protocols and some of them neglect the problem of data heterogeneity in FL. In this paper, we propose a novel federated semi-supervised learning framework for medical image segmentation incorporating intra-client and inter-client consistency learning. The intra-client consistency learning can introduce global data noise in data augmentation which can improve the generalization ability of the model and reduce the impact of data heterogeneity. The inter-client consistency learning is proposed to expand the feature search space and learn the ensemble knowledge of different clients. The two consistency learning mechanisms are achieved with the assistance of a Variational Autoencoder (VAE) trained collaboratively by clients. The experimental results illustrate that our method outperforms the state-of-the-art methods under different FSSL settings. The code is available at https://github.com/zyb98/FV2IC.