Performance evaluation of cluster-based federated machine learning

Karim Asif Sattar,Uthman Baroudi
DOI: https://doi.org/10.1007/s00521-024-09487-3
2024-02-19
Neural Computing and Applications
Abstract:Federated Learning (FL) is a collaborative training method for machine learning (ML) that aggregates model weights from multiple participants during the training phase. The learning phase of machine learning techniques is distributed, in which each participating device trains a model using its local data set and sends model weights to a centralized node. The central node aggregates weights and sends the updated weights back to devices. The process continues until a specific threshold is reached such accuracy, response time. In this paper, we present a performance evaluation of FL in a clustering-based multi-hop network to simulate the effect of the dynamic environment on the accuracy of the global model. It is observed that a minimum number of participating nodes is required within a cluster to maintain a high level of global accuracy. A global threshold value needs to be defined to maintain high global accuracy and avoid degradation of model performance.
computer science, artificial intelligence
What problem does this paper attempt to address?