Federated Learning (FL) has attracted considerable attention due to its ability to train machine learning models collaboratively without sharing raw data. However, the computational limitations of devices involved in FL often lead to slow performance and delays in the training process. To address these challenges, we propose a novel approach leveraging a Deep Reinforcement Learning (DRL) technique to adjust the training intensities of participating edge devices (EDs) dynamically. Additionally, we integrate Bayesian Optimization (BO) techniques to optimize the allocation of local training iterations based on the computational capabilities of each client device. The proposed approach enhances FL performance by effectively managing the training workload across distributed edge devices, thereby improving the convergence of the global model. Through extensive simulations, we demonstrate that the proposed approach mitigates the computational constraints of individual devices and facilitates faster model convergence in FL scenarios.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

An Adaptive Federated Deep Reinforcement Learning with Bayesian Optimization for Data Analysis in Edge Networks

  • Parvati Viswanathan,
  • Neha Singh,
  • Mainak Adhikari

摘要

Federated Learning (FL) has attracted considerable attention due to its ability to train machine learning models collaboratively without sharing raw data. However, the computational limitations of devices involved in FL often lead to slow performance and delays in the training process. To address these challenges, we propose a novel approach leveraging a Deep Reinforcement Learning (DRL) technique to adjust the training intensities of participating edge devices (EDs) dynamically. Additionally, we integrate Bayesian Optimization (BO) techniques to optimize the allocation of local training iterations based on the computational capabilities of each client device. The proposed approach enhances FL performance by effectively managing the training workload across distributed edge devices, thereby improving the convergence of the global model. Through extensive simulations, we demonstrate that the proposed approach mitigates the computational constraints of individual devices and facilitates faster model convergence in FL scenarios.