On-Demand Model and Client Deployment in Federated Learning with Deep Reinforcement Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Chahoud, Mario, Sami, Hani, Mourad, Azzam, Otrok, Hadi, Bentahar, Jamal, Guizani, Mohsen
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909200360144896
author Chahoud, Mario
Sami, Hani
Mourad, Azzam
Otrok, Hadi
Bentahar, Jamal
Guizani, Mohsen
author_facet Chahoud, Mario
Sami, Hani
Mourad, Azzam
Otrok, Hadi
Bentahar, Jamal
Guizani, Mohsen
contents In Federated Learning (FL), the limited accessibility of data from diverse locations and user types poses a significant challenge due to restricted user participation. Expanding client access and diversifying data enhance models by incorporating diverse perspectives, thereby enhancing adaptability. However, challenges arise in dynamic and mobile environments where certain devices may become inaccessible as FL clients, impacting data availability and client selection methods. To address this, we propose an On-Demand solution, deploying new clients using Docker Containers on-the-fly. Our On-Demand solution, employing Deep Reinforcement Learning (DRL), targets client availability and selection, while considering data shifts, and container deployment complexities. It employs an autonomous end-to-end solution for handling model deployment and client selection. The DRL strategy uses a Markov Decision Process (MDP) framework, with a Master Learner and a Joiner Learner. The designed cost functions represent the complexity of the dynamic client deployment and selection. Simulated tests show that our architecture can easily adjust to changes in the environment and respond to On-Demand requests. This underscores its ability to improve client availability, capability, accuracy, and learning efficiency, surpassing heuristic and tabular reinforcement learning solutions.
format Preprint
id arxiv_https___arxiv_org_abs_2405_07175
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle On-Demand Model and Client Deployment in Federated Learning with Deep Reinforcement Learning
Chahoud, Mario
Sami, Hani
Mourad, Azzam
Otrok, Hadi
Bentahar, Jamal
Guizani, Mohsen
Machine Learning
In Federated Learning (FL), the limited accessibility of data from diverse locations and user types poses a significant challenge due to restricted user participation. Expanding client access and diversifying data enhance models by incorporating diverse perspectives, thereby enhancing adaptability. However, challenges arise in dynamic and mobile environments where certain devices may become inaccessible as FL clients, impacting data availability and client selection methods. To address this, we propose an On-Demand solution, deploying new clients using Docker Containers on-the-fly. Our On-Demand solution, employing Deep Reinforcement Learning (DRL), targets client availability and selection, while considering data shifts, and container deployment complexities. It employs an autonomous end-to-end solution for handling model deployment and client selection. The DRL strategy uses a Markov Decision Process (MDP) framework, with a Master Learner and a Joiner Learner. The designed cost functions represent the complexity of the dynamic client deployment and selection. Simulated tests show that our architecture can easily adjust to changes in the environment and respond to On-Demand requests. This underscores its ability to improve client availability, capability, accuracy, and learning efficiency, surpassing heuristic and tabular reinforcement learning solutions.
title On-Demand Model and Client Deployment in Federated Learning with Deep Reinforcement Learning
topic Machine Learning
url https://arxiv.org/abs/2405.07175