Exploration and Exploitation in Federated Learning to Exclude Clients with Poisoned Data

Shadha Tabatabai, Ihab Mohammed, Basheer Qolomany, Abdullatif Albaseer, Kashif Ahmad, Mohamed Abdallah, Ala Al-Fuqaha

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

Federated Learning (FL) is one of the hot research topics, and it utilizes Machine Learning (ML) in a distributed manner without directly accessing private data on clients. How-ever, FL faces many challenges, including the difficulty to obtain high accuracy, high communication cost between clients and the server, and security attacks related to adversarial ML. To tackle these three challenges, we propose an FL algorithm inspired by evolutionary techniques. The proposed algorithm groups clients randomly in many clusters, each with a model selected randomly to explore the performance of different models. The clusters are then trained in a repetitive process where the worst performing cluster is removed in each iteration until one cluster remains. In each iteration, some clients are expelled from clusters either due to using poisoned data or low performance. The surviving clients are exploited in the next iteration. The remaining cluster with surviving clients is then used for training the best FL model (i.e., remaining FL model). Communication cost is reduced since fewer clients are used in the final training of the FL model. To evaluate the performance of the proposed algorithm, we conduct a number of experiments using FEMNIST dataset and compare the result against the random FL algorithm. The experimental results show that the proposed algorithm outperforms the baseline algorithm in terms of accuracy, communication cost, and security.

Original languageEnglish (US)
Title of host publication2022 International Wireless Communications and Mobile Computing, IWCMC 2022
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages407-412
Number of pages6
ISBN (Electronic)9781665467490
DOIs
StatePublished - 2022
Event18th IEEE International Wireless Communications and Mobile Computing, IWCMC 2022 - Dubrovnik, Croatia
Duration: May 30 2022Jun 3 2022

Publication series

Name2022 International Wireless Communications and Mobile Computing, IWCMC 2022

Conference

Conference18th IEEE International Wireless Communications and Mobile Computing, IWCMC 2022
Country/TerritoryCroatia
CityDubrovnik
Period5/30/226/3/22

Keywords

  • CNNs
  • Deep Learning
  • Distributed ML
  • Edge Computing
  • Federated Learning
  • Internet of Things
  • Security

ASJC Scopus subject areas

  • Computer Networks and Communications
  • Signal Processing
  • Instrumentation

Fingerprint

Dive into the research topics of 'Exploration and Exploitation in Federated Learning to Exclude Clients with Poisoned Data'. Together they form a unique fingerprint.

Cite this