Subnetwork-based Federated Few-shot semantic seg-mentation of Organ Images
DOI:
Author:
Affiliation:

Tianjin University of Technology

Clc Number:

Fund Project:

The National Natural Science Foundation of China (General Program, Key Program, Major Research Plan)

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    Federated learning, as a distributed learning paradigm, allows multiple medical institutions to collaborate on learning without the need to centralize all client data. However, existing methods pay little attention to more challenging medical image semantic segmentation tasks, especially in the scenario of uneven data category distribution in fed-erated few-shot learning. In this context, we propose a subnetwork-based federated few-shot medical image seg-mentation method. Firstly, individual clients train using local training samples and then upload local model gradients to the server. The server utilizes their respective local model gradients to update the subnetwork maintained on the server and generate aggregation weights for forming personalized model parameters. Through this method, we are able to learn the similarities between different clients to address data heterogeneity issues. In addition, to enhance the communication efficiency between clients and the server, we have also designed a personalized layer aggregation strategy, which only transmits partial layer model parameters during the communication process to improve com-munication efficiency. Finally, we conducted experiments on ABD-MRI and ABD-CT datasets to demonstrate the effectiveness of our method.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:November 16,2023
  • Revised:November 30,2023
  • Adopted:December 14,2023
  • Online:
  • Published: