FedFMS: Exploring Federated Foundation Models for Medical Image Segmentation

Yuxi Liu, Guibo Luo, Yuesheng Zhu
3/8/2024
eess.IVcs.CVcs.DC

Official code: LIU-YUXI/FedFMS (72 stars, last commit 2025-02-16). 1 more implementation below.

Abstract

Medical image segmentation is crucial for clinical diagnosis. The Segmentation Anything Model (SAM) serves as a powerful foundation model for visual segmentation and can be adapted for medical image segmentation. However, medical imaging data typically contain privacy-sensitive information, making it challenging to train foundation models with centralized storage and sharing. To date, there are few foundation models tailored for medical image deployment within the federated learning framework, and the segmentation performance, as well as the efficiency of communication and training, remain unexplored. In response to these issues, we developed Federated Foundation models for Medical image Segmentation (FedFMS), which includes the Federated SAM (FedSAM) and a communication and training-efficient Federated SAM with Medical SAM Adapter (FedMSA). Comprehensive experiments on diverse datasets are conducted to investigate the performance disparities between centralized training and federated learning across various configurations of FedFMS. The experiments revealed that FedFMS could achieve performance comparable to models trained via centralized training methods while maintaining privacy. Furthermore, FedMSA demonstrated the potential to enhance communication and training efficiency. Our model implementation codes are available at https://github.com/LIU-YUXI/FedFMS.

View on arXivView PDF

Code Implementations(2)

LIU-YUXI/FedFMSOfficial100%

[MICCAI2024] "FedFMS: Exploring Federated Foundation Models for Medical Image Segmentation". A framework for fine-tuning SAM (Segment Anything) in the federated learning paradigm for medical image segmentation.

727Shell, PythonJun 24, 20231 years ago
LMIAPC/FednnU-NetCommunity100%
Apache-2.0

Federated learning nnU-Net code, based on FedAvg.

70Shell, PythonJun 25, 20242 years agoApache-2.0

Similar Papers

Cite this paper

@article{liu2024fedfms,
  title  = {FedFMS: Exploring Federated Foundation Models for Medical Image Segmentation},
  author = {Yuxi Liu and Guibo Luo and Yuesheng Zhu},
  year   = {2024},
  eprint = {2403.05408},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2403.05408v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2403.05408.svg)](https://codeofpaper.com/paper/2403.05408)

Discussion