Effectiveness of Decentralized Federated Learning Algorithms in Healthcare: A Case Study on Cancer Classification

Deep learning-based medical image analysis is an effective and precise method for identifying various cancer types. However, due to concerns over patient privacy, sharing diagnostic images across medical facilities is typically not permitted. Federated learning (FL) tries to construct a shared model...

Full description

Saved in:
Bibliographic Details
Published in:Electronics (Basel) 2022-12, Vol.11 (24), p.4117
Main Authors: Subramanian, Malliga, Rajasekar, Vani, V. E., Sathishkumar, Shanmugavadivel, Kogilavani, Nandhini, P. S
Format: Article
Language:English
Subjects:
Algorithms
Artificial intelligence
Autonomous vehicles
Cancer
Collaboration
Data mining
Datasets
Deep learning
Diagnosis
Diagnostic imaging
Environment models
Experiments
Federated learning
Heterogeneity
Image analysis
Image classification
Machine learning
Medical imaging
Optimization
Optimization techniques
Performance evaluation
Privacy
Citations: Items that this one cites
Items that cite this one
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Deep learning-based medical image analysis is an effective and precise method for identifying various cancer types. However, due to concerns over patient privacy, sharing diagnostic images across medical facilities is typically not permitted. Federated learning (FL) tries to construct a shared model across dispersed clients under such privacy-preserving constraints. Although there is a good chance of success, dealing with non-IID (non-independent and identical distribution) client data, which is a typical circumstance in real-world FL tasks, is still difficult for FL. We use two FL algorithms, FedAvg and FedProx, to manage client heterogeneity and non-IID data in a federated setting. A heterogeneous data split of the cancer datasets with three different forms of cancer—cervical, lung, and colon—is used to validate the efficacy of the FL. In addition, since hyperparameter optimization presents new difficulties in an FL setting, we also examine the impact of various hyperparameter values. We use Bayesian optimization to fine-tune the hyperparameters and identify the appropriate values in order to increase performance. Furthermore, we investigate the hyperparameter optimization in both local and global models of the FL environment. Through a series of experiments, we find that FedProx outperforms FedAvg in scenarios with significant levels of heterogeneity.
ISSN:2079-9292
2079-9292
DOI:10.3390/electronics11244117