Dynamic selectout and voting-based federated learning for enhanced medical image analysis

Iqbal, Saeed and Qureshi, Adnan N. and Alhussein, Musaed and Aurangzeb, Khursheed and Mahmood, Atif and Razalli Azzuhri, Saaidal (2025) Dynamic selectout and voting-based federated learning for enhanced medical image analysis. Machine Learning: Science and Technology, 6 (1). 015010. ISSN 2632-2153, DOI https://doi.org/10.1088/2632-2153/ada0a6.

Full text not available from this repository.
Official URL: https://doi.org/10.1088/2632-2153/ada0a6

Abstract

Federated learning (FL) is a promising technique for training machine learning models on distributed, privacy-aware datasets. Nevertheless, FL faces difficulties with agent/client participation, model performance, and the heterogeneous nature of networked data sources when it comes to distributed healthcare systems. When these agents work together in the system, it is imperative to tackle the complexities of distributed deep learning. We suggest a novel approach that uses a voting mechanism and dynamic SelectOut inside the FL framework to address these problems. Local medical imaging datasets frequently show diversity in distribution and data imbalances. In certain situations, traditional FL techniques like FedProx and federated averaging, which depend on data size to weight contributions, might not be the optimal choice. In order to improve parameter aggregation and client selection unpredictability and increase the model's adaptability to imbalanced and heterogeneous datasets, our proposed FedVoteNet model introduces SelectOut techniques based on voting methodology. Based on how much their local performance has improved from the last communication cycle, we arbitrarily remove clients. Additionally eliminated are clients whose model weights when combined with the global model adversely affect its performance. Our method is further enhanced by the inclusion of a voting mechanism. At the conclusion of each communication cycle, clients that improve both their local performance and their contribution to the global model are awarded higher voting values. This encourages more significant and effective contributions from clients by providing incentives for them to actively increase the diversity of their training data. We assess our approach on a dataset of medical images, including magnetic resonance imaging scans, and find that the FL model performs noticeably better (F1 Score = 0.968, Sensitivity = 0.977, Specificity = 0.945, and AUC = 0.950). The voting system and the dynamic SelectOut algorithms improve the convergence of the FL model and successfully handle the difficulties presented by uneven and heterogeneous datasets. To sum up, our proposed approach uses voting and dynamic SelectOut techniques to improve FL performance on a variety of uneven, distributed, and varied datasets. This strategy has a lot of potential to improve FL across a range of applications, especially those that prioritize data privacy, diversity, and performance.

Item Type: Article
Funders: Research Center of the College of Computer and Information Sciences, King Saud Universityhttp://dx.doi.org/10.13039/501100004745 (RSPD2025R947), King Saud University
Uncontrolled Keywords: federated learning; dynamic SelectOut mechanisms; voting system; heterogeneous datasets; model performance; data privacy
Subjects: Q Science > QA Mathematics > QA75 Electronic computers. Computer science
Divisions: Faculty of Computer Science & Information Technology > Department of Computer System & Technology
Depositing User: Ms. Juhaida Abd Rahim
Date Deposited: 17 Mar 2025 01:47
Last Modified: 17 Mar 2025 01:47
URI: http://eprints.um.edu.my/id/eprint/47692

Actions (login required)

View Item View Item