Reset filters

Search publications


By keyword
By department

No publications found.

 

Trust-Augmented Deep Reinforcement Learning for Federated Learning Client Selection

Authors: Rjoub GWahab OABentahar JCohen RBataineh AS


Affiliations

1 Concordia Institute for Information Systems Engineering, Concordia University, 1455 De Maisonneuve Blvd. W.2, Montreal, H3G 1M8 Quebec Canada.
2 Department of Computer Science and Engineering, Université du Québec en Outaouais, 101, Saint-Jean-Bosco, C.P. 1250, succursale Hull, Gatineau, J8X 3X7 Quebec Canada.
3 David R. Cheriton School of Computer Science, University of Waterloo, 200 University Avenue West, Waterloo, N2L 3G1 ON Canada.

Description

In the context of distributed machine learning, the concept of federated learning (FL) has emerged as a solution to the privacy concerns that users have about sharing their own data with a third-party server. FL allows a group of users (often referred to as clients) to locally train a single machine learning model on their devices without sharing their raw data. One of the main challenges in FL is how to select the most appropriate clients to participate in the training of a certain task. In this paper, we address this challenge and propose a trust-based deep reinforcement learning approach to select the most adequate clients in terms of resource consumption and training time. On top of the client selection mechanism, we embed a transfer learning approach to handle the scarcity of data in some regions and compensate potential lack of learning at some servers. We apply our solution in the healthcare domain in a COVID-19 detection scenario over IoT devices. In the considered scenario, edge servers collaborate with IoT devices to train a COVID-19 detection model using FL without having to share any raw confidential data. Experiments conducted on a real-world COVID-19 dataset reveal that our solution achieves a good trade-off between detection accuracy and model execution time compared to existing approaches.


Keywords: COVID-19 detectionDeep reinforcement learningEdge computingFederated learningInternet of things (IoT)Transfer learning


Links

PubMed: https://pubmed.ncbi.nlm.nih.gov/35875592/

DOI: 10.1007/s10796-022-10307-z