Federated Contrastive Learning for Volumetric Medical Image Segmentation
Not on the map. Why?
Abstract (arXiv)
Supervised deep learning needs a large amount of labeled data to achieve high performance. However, in medical imaging analysis, each site may only have a limited amount of data and labels, which makes learning ineffective. Federated learning (FL) can help in this regard by learning a shared model while keeping training data local for privacy. Traditional FL requires fully-labeled data for training, which is inconvenient or sometimes infeasible to obtain due to high labeling cost and the requirement of expertise. Contrastive learning (CL), as a self-supervised learning approach, can effectively learn from unlabeled data to pre-train a neural network encoder, followed by fine-tuning for downstream tasks with limited annotations. However, when adopting CL in FL, the limited data diversity on each client makes federated contrastive learning (FCL) ineffective. In this work, we propose an FCL framework for volumetric medical image segmentation with limited annotations. More specifically, we exchange the features in the FCL pre-training process such that diverse contrastive data are provided to each site for effective local CL while keeping raw data private. Based on the exchanged features, global structural matching further leverages the structural similarity to align local features to the remote ones such that a unified feature space can be learned among different sites. Experiments on a cardiac MRI dataset show the proposed framework substantially improves the segmentation performance compared with state-of-the-art techniques.
Key facts. Cited by 33 papers in the map (data through 14 Oct 2026). Builds on 25 of its 34 references that are in the map. First posted on arXiv 23 Apr 2022.
Builds on
25 of 34 references.
References · 25
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
- 2019Advances and Open Problems in Federated Learningcited by 3,046
- 2018Federated Optimization in Heterogeneous Networkscited by 2,987
- 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,857
- 2016V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentationcited by 2,292
- 2018Unsupervised Feature Learning via Non-Parametric Instance-level Discriminationcited by 1,845
- 2018Unsupervised Representation Learning by Predicting Image Rotationscited by 1,591
- 2020Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hyperspherecited by 1,405
- 2020Big Self-Supervised Models are Strong Semi-Supervised Learnerscited by 1,380
13 more
- 2019Contrastive Multiview Codingcited by 1,301
- 2018Federated Learning with Non-IID Datacited by 1,108
- 2019Self-Supervised Learning of Pretext-Invariant Representationscited by 808
- 2020The Future of Digital Health with Federated Learningcited by 633
- 2019Federated Machine Learning: Concept and Applicationscited by 412
- 2020Contrastive learning of global and local features for medical image segmentation with limited annotationscited by 309
- 2020Threats to Federated Learning: A Surveycited by 178
- 2018Multi-Institutional Deep Learning Modeling Without Sharing Patient Data: A Feasibility Study on Brain Tumor Segmentationcited by 149
- 2020Federated Unsupervised Representation Learningcited by 93
- 2021Positional Contrastive Learning for Volumetric Medical Image Segmentationcited by 56
- 2017Automatic Brain Tumor Detection and Segmentation Using U-Net Based Fully Convolutional Networkscited by 42
- 2021Enabling On-Device Self-Supervised Contrastive Learning With Selective Data Contrastcited by 21
- 2022Federated Contrastive Learning for Dermatological Disease Diagnosis via On-device Learningcited by 21
Built on by
Citing papers per year: 2022: 10; 2023: 12; 2024: 7; 2025: 4.
33 papers.
Sort:
By area:
- 2023Federated Learning for Medical Image Analysis: A Surveycited by 78
- 2022Dissecting Self-Supervised Learning Methods for Surgical Computer Visioncited by 48
- 2022Label-Efficient Self-Supervised Federated Learning for Tackling Data Heterogeneity in Medical Imagingcited by 39
- 2023AutoReP: Automatic ReLU Replacement for Fast Private Network Inferencecited by 29
- 2023LinGCN: Structural Linearized Graph Convolutional Network for Homomorphically Encrypted Inferencecited by 27
- 2022FedMix: Mixed Supervised Federated Learning for Medical Image Segmentationcited by 23
- 2022Distributed Contrastive Learning for Medical Image Segmentationcited by 16
- 2022Hypernetwork-based Personalized Federated Learning for Multi-Institutional CT Imagingcited by 14
- 2022Dynamic Bank Learning for Semi-supervised Federated Image Diagnosis with Class Imbalancecited by 13
- 2023RRNet: Towards ReLU-Reduced Neural Network for Two-party Computation Based Private Inferencecited by 12
- 2023A Review of Predictive and Contrastive Self-supervised Learning for Medical Imagescited by 10
- 2023On-Device Unsupervised Image Segmentationcited by 10
21 more
- 2023Cross-Modal Vertical Federated Learning for MRI Reconstructioncited by 9
- 2022Contrastive Brain Network Learning via Hierarchical Signed Graph Pooling Modelcited by 8
- 2024A Comprehensive Survey of Federated Transfer Learning: Challenges, Methods and Applicationscited by 8
- 2022Self-Supervised-RCNN for Medical Image Segmentation with Limited Data Annotationcited by 7
- 2023QuMoS: A Framework for Preserving Security of Quantum Machine Learning Modelcited by 6
- 2022Achieving Fairness in Dermatological Disease Diagnosis through Automatic Weight Adjusting Federated Learning and Personalizationcited by 4
- 2024All-in-One Tuning and Structural Pruning for Domain-Specific LLMscited by 4
- 2025FROG: Fair Removal on Graphscited by 4
- 2024Interpretable Spatio-Temporal Embedding for Brain Structural-Effective Network with Ordinary Differential Equationcited by 3
- 2023A Survey of the Impact of Self-Supervised Pretraining for Diagnostic Tasks with Radiological Imagescited by 2
- 2023Edge-InversionNet: Enabling Efficient Inference of InversionNet on Edge Devicescited by 2
- 2024Key Information Retrieval to Classify the Unstructured Data Content of Preferential Trade Agreementscited by 1
- 2025Bridging Contrastive Learning and Domain Adaptation: Theoretical Perspective and Practical Applicationcited by 1
- 2025FedDEAP: Adaptive Dual-Prompt Tuning for Multi-Domain Federated Learningcited by 1
- 2022Joint Embedding of 2D and 3D Networks for Medical Image Anomaly Detectioncited by 0
- 2023Semi-Supervised Relational Contrastive Learningcited by 0
- 2023Tackling Heterogeneity in Medical Federated learning via Vision Transformerscited by 0
- 2024Membership Information Leakage in Federated Contrastive Learningcited by 0
- 2024Anatomical 3D Style Transfer Enabling Efficient Federated Learning with Extremely Low Communication Costscited by 0
- 2024Federated Black-Box Adaptation for Semantic Segmentationcited by 0
- 2025Deep Learning Approaches for Medical Imaging Under Varying Degrees of Label Availability: A Comprehensive Surveycited by 0
- 2025FedDEAP: Adaptive Dual-Prompt Tuning for Multi-Domain Federated Learning21 Oct 2025
- 2025Deep Learning Approaches for Medical Imaging Under Varying Degrees of Label Availability: A Comprehensive Survey15 Apr 2025
- 2025FROG: Fair Removal on Graphs23 Mar 2025
- 2025Bridging Contrastive Learning and Domain Adaptation: Theoretical Perspective and Practical Application28 Jan 2025
- 2024All-in-One Tuning and Structural Pruning for Domain-Specific LLMs19 Dec 2024
- 2024Federated Black-Box Adaptation for Semantic Segmentation31 Oct 2024
- 2024Anatomical 3D Style Transfer Enabling Efficient Federated Learning with Extremely Low Communication Costs26 Oct 2024
- 2024Interpretable Spatio-Temporal Embedding for Brain Structural-Effective Network with Ordinary Differential Equation21 May 2024
- 2024Membership Information Leakage in Federated Contrastive Learning6 Mar 2024
- 2024A Comprehensive Survey of Federated Transfer Learning: Challenges, Methods and Applications3 Mar 2024
- 2024Key Information Retrieval to Classify the Unstructured Data Content of Preferential Trade Agreements23 Jan 2024
- 2023Edge-InversionNet: Enabling Efficient Inference of InversionNet on Edge Devices14 Oct 2023
21 more
- 2023Tackling Heterogeneity in Medical Federated learning via Vision Transformers13 Oct 2023
- 2023LinGCN: Structural Linearized Graph Convolutional Network for Homomorphically Encrypted Inference25 Sep 2023
- 2023A Survey of the Impact of Self-Supervised Pretraining for Diagnostic Tasks with Radiological Images5 Sep 2023
- 2023AutoReP: Automatic ReLU Replacement for Fast Private Network Inference20 Aug 2023
- 2023Federated Learning for Medical Image Analysis: A Survey9 Jun 2023
- 2023Cross-Modal Vertical Federated Learning for MRI Reconstruction5 Jun 2023
- 2023QuMoS: A Framework for Preserving Security of Quantum Machine Learning Model23 Apr 2023
- 2023Semi-Supervised Relational Contrastive Learning11 Apr 2023
- 2023On-Device Unsupervised Image Segmentation24 Feb 2023
- 2023A Review of Predictive and Contrastive Self-supervised Learning for Medical Images10 Feb 2023
- 2023RRNet: Towards ReLU-Reduced Neural Network for Two-party Computation Based Private Inference5 Feb 2023
- 2022Joint Embedding of 2D and 3D Networks for Medical Image Anomaly Detection21 Dec 2022
- 2022Achieving Fairness in Dermatological Disease Diagnosis through Automatic Weight Adjusting Federated Learning and Personalization23 Aug 2022
- 2022Distributed Contrastive Learning for Medical Image Segmentation7 Aug 2022
- 2022Self-Supervised-RCNN for Medical Image Segmentation with Limited Data Annotation17 Jul 2022
- 2022Contrastive Brain Network Learning via Hierarchical Signed Graph Pooling Model14 Jul 2022
- 2022Dissecting Self-Supervised Learning Methods for Surgical Computer Vision1 Jul 2022
- 2022Dynamic Bank Learning for Semi-supervised Federated Image Diagnosis with Class Imbalance27 Jun 2022
- 2022Hypernetwork-based Personalized Federated Learning for Multi-Institutional CT Imaging8 Jun 2022
- 2022Label-Efficient Self-Supervised Federated Learning for Tackling Data Heterogeneity in Medical Imaging17 May 2022
- 2022FedMix: Mixed Supervised Federated Learning for Medical Image Segmentation4 May 2022
- 2025FROG: Fair Removal on Graphscited by 4
- 2024Membership Information Leakage in Federated Contrastive Learningcited by 0
- 2025FROG: Fair Removal on Graphs23 Mar 2025
- 2024Membership Information Leakage in Federated Contrastive Learning6 Mar 2024
- 2023Federated Learning for Medical Image Analysis: A Surveycited by 78
- 2022Dissecting Self-Supervised Learning Methods for Surgical Computer Visioncited by 48
- 2022Label-Efficient Self-Supervised Federated Learning for Tackling Data Heterogeneity in Medical Imagingcited by 39
- 2023AutoReP: Automatic ReLU Replacement for Fast Private Network Inferencecited by 29
- 2023LinGCN: Structural Linearized Graph Convolutional Network for Homomorphically Encrypted Inferencecited by 27
- 2022FedMix: Mixed Supervised Federated Learning for Medical Image Segmentationcited by 23
- 2022Distributed Contrastive Learning for Medical Image Segmentationcited by 16
- 2022Hypernetwork-based Personalized Federated Learning for Multi-Institutional CT Imagingcited by 14
- 2022Dynamic Bank Learning for Semi-supervised Federated Image Diagnosis with Class Imbalancecited by 13
- 2023RRNet: Towards ReLU-Reduced Neural Network for Two-party Computation Based Private Inferencecited by 12
- 2023A Review of Predictive and Contrastive Self-supervised Learning for Medical Imagescited by 10
- 2023On-Device Unsupervised Image Segmentationcited by 10
12 most cited of 31 All 31 (JSON)
- 2025FedDEAP: Adaptive Dual-Prompt Tuning for Multi-Domain Federated Learning21 Oct 2025
- 2025Deep Learning Approaches for Medical Imaging Under Varying Degrees of Label Availability: A Comprehensive Survey15 Apr 2025
- 2025Bridging Contrastive Learning and Domain Adaptation: Theoretical Perspective and Practical Application28 Jan 2025
- 2024All-in-One Tuning and Structural Pruning for Domain-Specific LLMs19 Dec 2024
- 2024Federated Black-Box Adaptation for Semantic Segmentation31 Oct 2024
- 2024Anatomical 3D Style Transfer Enabling Efficient Federated Learning with Extremely Low Communication Costs26 Oct 2024
- 2024Interpretable Spatio-Temporal Embedding for Brain Structural-Effective Network with Ordinary Differential Equation21 May 2024
- 2024A Comprehensive Survey of Federated Transfer Learning: Challenges, Methods and Applications3 Mar 2024
- 2024Key Information Retrieval to Classify the Unstructured Data Content of Preferential Trade Agreements23 Jan 2024
- 2023Edge-InversionNet: Enabling Efficient Inference of InversionNet on Edge Devices14 Oct 2023
- 2023Tackling Heterogeneity in Medical Federated learning via Vision Transformers13 Oct 2023
- 2023LinGCN: Structural Linearized Graph Convolutional Network for Homomorphically Encrypted Inference25 Sep 2023
12 newest of 31 All 31 (JSON)