Towards Energy-Aware Federated Learning via MARL: A Dual-Selection Approach for Model and Client

2024-05-13 · Jun Xia, Yi Zhang, Yiyu Shi

Although Federated Learning (FL) is promising in knowledge sharing for heterogeneous Artificial Intelligence of Thing (AIoT) devices, their training performance and energy efficacy are severely restricted in practical battery-driven scenarios due to the ``wooden barrel effect'' caused by the mismatch between homogeneous model paradigms and heterogeneous device capability. As a result, due to various kinds of differences among devices, it is hard for existing FL methods to conduct training effectively in energy-constrained scenarios, such as battery constraints of devices. To tackle the above issues, we propose an energy-aware FL framework named DR-FL, which considers the energy constraints in both clients and heterogeneous deep learning models to enable energy-efficient FL. Unlike Vanilla FL, DR-FL adopts our proposed Muti-Agents Reinforcement Learning (MARL)-based dual-selection method, which allows participated devices to make contributions to the global model effectively and adaptively based on their computing capabilities and energy capacities in a MARL-based manner. Experiments conducted with various widely recognized datasets demonstrate that DR-FL has the capability to optimize the exchange of knowledge among diverse models in large-scale AIoT systems while adhering to energy limitations. Additionally, it improves the performance of each individual heterogeneous device's model.

📄 PDF Abstract BibTeX arXiv:2405.08183

Code (0)

등록된 구현이 없습니다.

Tasks

Federated Learning

REWAFL: Residual Energy and Wireless Aware Participant Selection for Efficient Federated Learning over Mobile Devices

2023-09-24 · Y. Li, X. Qin, J. Geng, R. Chen 외

Participant selection (PS) helps to accelerate federated learning (FL) convergence, which is essential for the practical deployment of FL over mobile devices. While most existing PS approaches focus on improving training…

Federated Learning

Federated Multi-Agent Reinforcement Learning for Privacy-Preserving and Energy-Aware Resource Management in 6G Edge Networks

2025-09-12 · Francisco Javier Esono Nkulu Andong, Qi Min arxiv

As sixth-generation (6G) networks move toward ultra-dense, intelligent edge environments, efficient resource management under stringent privacy, mobility, and energy constraints becomes critical. This paper introduces a …

Multi-agent Reinforcement Learning

Scaling up Energy-Aware Multi-Agent Reinforcement Learning for Mission-Oriented Drone Networks with Individual Reward

2026-05-24 · Changling Li, Ying Li arxiv

Multi-agent reinforcement learning (MARL) has shown wide applicability in collaborative systems such as autonomous driving and smart cities for its ability of learning through interaction. With the recent development of …

Multi-agent Reinforcement Learning Trajectory Planning Autonomous Driving

A Multi-agent Reinforcement Learning Approach for Efficient Client Selection in Federated Learning

2022-01-09 · Sai Qian Zhang, Jieyu Lin, Qi Zhang

Federated learning (FL) is a training technique that enables client devices to jointly learn a shared model by aggregating locally-computed models without exposing their raw data. While most of the existing work focuses …

Federated Learning Multi-agent Reinforcement Learning reinforcement-learning Reinforcement Learning (RL)

Data-Efficient Energy-Aware Participant Selection for UAV-Enabled Federated Learning

2023-08-14 · Youssra Cheriguene, Wael Jaafar, Chaker Abdelaziz Kerrache, Halim Yanikomeroglu 외

Unmanned aerial vehicle (UAV)-enabled edge federated learning (FL) has sparked a rise in research interest as a result of the massive and heterogeneous data collected by UAVs, as well as the privacy concerns related to U…

Federated Learning SSIM