Smart client selection strategies for enhanced federated learning in digital healthcare applications

Sachin D N,Annappa B,Sateesh Ambesange
DOI: https://doi.org/10.1007/s11042-024-19403-5
IF: 2.577
2024-05-22
Multimedia Tools and Applications
Abstract:Federated Learning (FL) trains AI models in healthcare without sharing patient data. FL computes client models locally and combines them to create a global model. However, involving all clients is impractical due to resource limitations. Random selection of a subset of clients in each FL round can pose challenges for resource-limited devices, leading to longer processing times and potential training failures. To tackle these obstacles, this research proposes a novel strategy for FL that treats each training round as a client selection process to improve the efficiency and effectiveness of FL in healthcare applications, where data privacy is paramount. The approach begins by calculating the uncertainty value for each client, which quantifies the contribution of the client's data to the overall model. Clients are then ranked based on their uncertainty values, and those with higher loss values are given a higher probability of participating in the training process. The experimental outcomes clearly show that the proposed strategy effectively makes 1.3x training faster, and 30% lowers communication expenses, conserves computational resources, and enhances model performance when contrasted with random client selection.
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering
What problem does this paper attempt to address?