Performance evaluation of cluster-based federated machine learning

Karim Asif Sattar*, Uthman Baroudi

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

1 Scopus citations

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.

Original languageEnglish
Pages (from-to)7657-7668
Number of pages12
JournalNeural Computing and Applications
Volume36
Issue number14
DOIs
StatePublished - May 2024

Bibliographical note

Publisher Copyright:
© The Author(s), under exclusive licence to Springer-Verlag London Ltd., part of Springer Nature 2024.

Keywords

  • Federated machine learning
  • Internet of Things (IoT)
  • K-means clustering
  • Machine learning
  • Mobile ad-hoc networks (MANETs)
  • NS3
  • Selective participating nodes

ASJC Scopus subject areas

  • Software
  • Artificial Intelligence

Fingerprint

Dive into the research topics of 'Performance evaluation of cluster-based federated machine learning'. Together they form a unique fingerprint.

Cite this