Dealing With Heterogeneous 3D MR Knee Images: A Federated Few-Shot
Learning Method With Dual Knowledge Distillation
- URL: http://arxiv.org/abs/2303.14357v2
- Date: Tue, 18 Apr 2023 02:37:56 GMT
- Title: Dealing With Heterogeneous 3D MR Knee Images: A Federated Few-Shot
Learning Method With Dual Knowledge Distillation
- Authors: Xiaoxiao He, Chaowei Tan, Bo Liu, Liping Si, Weiwu Yao, Liang Zhao, Di
Liu, Qilong Zhangli, Qi Chang, Kang Li and Dimitris N. Metaxas
- Abstract summary: Federated learning enables training between clients without aggregating data.
Clinical institutions do not have enough supervised data for training locally.
Large institutions have the resources to compile data repositories with high-resolution images and labels.
- Score: 39.40515099843844
- License: http://creativecommons.org/licenses/by-nc-sa/4.0/
- Abstract: Federated Learning has gained popularity among medical institutions since it
enables collaborative training between clients (e.g., hospitals) without
aggregating data. However, due to the high cost associated with creating
annotations, especially for large 3D image datasets, clinical institutions do
not have enough supervised data for training locally. Thus, the performance of
the collaborative model is subpar under limited supervision. On the other hand,
large institutions have the resources to compile data repositories with
high-resolution images and labels. Therefore, individual clients can utilize
the knowledge acquired in the public data repositories to mitigate the shortage
of private annotated images. In this paper, we propose a federated few-shot
learning method with dual knowledge distillation. This method allows joint
training with limited annotations across clients without jeopardizing privacy.
The supervised learning of the proposed method extracts features from limited
labeled data in each client, while the unsupervised data is used to distill
both feature and response-based knowledge from a national data repository to
further improve the accuracy of the collaborative model and reduce the
communication cost. Extensive evaluations are conducted on 3D magnetic
resonance knee images from a private clinical dataset. Our proposed method
shows superior performance and less training time than other semi-supervised
federated learning methods. Codes and additional visualization results are
available at https://github.com/hexiaoxiao-cs/fedml-knee.
Related papers
- CL3: A Collaborative Learning Framework for the Medical Data Ensuring Data Privacy in the Hyperconnected Environment [1.223961905359498]
In a hyperconnected environment, medical institutions are concerned with data privacy when sharing and transmitting sensitive patient information.
A collaborative learning framework, including transfer, federated, and incremental learning, can generate efficient, secure, and scalable models.
This study aims to address the detection of COVID-19 using chest X-ray images through a proposed collaborative learning framework called CL3.
arXiv Detail & Related papers (2024-10-10T13:29:12Z) - Federated Semi-supervised Learning for Medical Image Segmentation with intra-client and inter-client Consistency [10.16245019262119]
Federated learning aims to train a shared model of isolated clients without local data exchange.
In this work, we propose a novel federated semi-supervised learning framework for medical image segmentation.
arXiv Detail & Related papers (2024-03-19T12:52:38Z) - Medical Federated Model with Mixture of Personalized and Sharing
Components [31.068735334318088]
We propose a new personalized framework of federated learning to handle the problem.
It successfully yields personalized models based on awareness of similarity between local data.
Also, we propose an effective method to reduce the computational cost, which improves computation efficiency significantly.
arXiv Detail & Related papers (2023-06-26T07:50:32Z) - Federated Learning with Privacy-Preserving Ensemble Attention
Distillation [63.39442596910485]
Federated Learning (FL) is a machine learning paradigm where many local nodes collaboratively train a central model while keeping the training data decentralized.
We propose a privacy-preserving FL framework leveraging unlabeled public data for one-way offline knowledge distillation.
Our technique uses decentralized and heterogeneous local data like existing FL approaches, but more importantly, it significantly reduces the risk of privacy leakage.
arXiv Detail & Related papers (2022-10-16T06:44:46Z) - Distributed Contrastive Learning for Medical Image Segmentation [16.3860181959878]
Supervised deep learning needs a large amount of labeled data to achieve high performance.
In medical imaging analysis, each site may only have a limited amount of data and labels, which makes learning ineffective.
We propose two federated self-supervised learning frameworks for medical image segmentation with limited annotations.
arXiv Detail & Related papers (2022-08-07T20:47:05Z) - Few-Shot Non-Parametric Learning with Deep Latent Variable Model [50.746273235463754]
We propose Non-Parametric learning by Compression with Latent Variables (NPC-LV)
NPC-LV is a learning framework for any dataset with abundant unlabeled data but very few labeled ones.
We show that NPC-LV outperforms supervised methods on all three datasets on image classification in low data regime.
arXiv Detail & Related papers (2022-06-23T09:35:03Z) - Federated Contrastive Learning for Volumetric Medical Image Segmentation [16.3860181959878]
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.
In this work, we propose an FCL framework for volumetric medical image segmentation with limited annotations.
arXiv Detail & Related papers (2022-04-23T03:47:23Z) - Federated Cycling (FedCy): Semi-supervised Federated Learning of
Surgical Phases [57.90226879210227]
FedCy is a semi-supervised learning (FSSL) method that combines FL and self-supervised learning to exploit a decentralized dataset of both labeled and unlabeled videos.
We demonstrate significant performance gains over state-of-the-art FSSL methods on the task of automatic recognition of surgical phases.
arXiv Detail & Related papers (2022-03-14T17:44:53Z) - Federated Semi-supervised Medical Image Classification via Inter-client
Relation Matching [58.26619456972598]
Federated learning (FL) has emerged with increasing popularity to collaborate distributed medical institutions for training deep networks.
This paper studies a practical yet challenging FL problem, named textitFederated Semi-supervised Learning (FSSL)
We present a novel approach for this problem, which improves over traditional consistency regularization mechanism with a new inter-client relation matching scheme.
arXiv Detail & Related papers (2021-06-16T07:58:00Z) - CosSGD: Nonlinear Quantization for Communication-efficient Federated
Learning [62.65937719264881]
Federated learning facilitates learning across clients without transferring local data on these clients to a central server.
We propose a nonlinear quantization for compressed gradient descent, which can be easily utilized in federated learning.
Our system significantly reduces the communication cost by up to three orders of magnitude, while maintaining convergence and accuracy of the training process.
arXiv Detail & Related papers (2020-12-15T12:20:28Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.