AdaptiveFL: Adaptive Heterogeneous Federated Learning for Resource-Constrained AIoT Systems

Autor:	Jia, Chentao, Hu, Ming, Chen, Zekai, Yang, Yanxin, Xie, Xiaofei, Liu, Yang, Chen, Mingsong
Rok vydání:	2023
Předmět:	Computer Science - Machine Learning Computer Science - Distributed Parallel and Cluster Computing
Druh dokumentu:	Working Paper
DOI:	10.1145/3649329.3655917
Popis:	Although Federated Learning (FL) is promising to enable collaborative learning among Artificial Intelligence of Things (AIoT) devices, it suffers from the problem of low classification performance due to various heterogeneity factors (e.g., computing capacity, memory size) of devices and uncertain operating environments. To address these issues, this paper introduces an effective FL approach named AdaptiveFL based on a novel fine-grained width-wise model pruning strategy, which can generate various heterogeneous local models for heterogeneous AIoT devices. By using our proposed reinforcement learning-based device selection mechanism, AdaptiveFL can adaptively dispatch suitable heterogeneous models to corresponding AIoT devices on the fly based on their available resources for local training. Experimental results show that, compared to state-of-the-art methods, AdaptiveFL can achieve up to 16.83% inference improvements for both IID and non-IID scenarios. Comment: This paper has been accepted by DAC2024
Databáze:	arXiv
Externí odkaz:	http://arxiv.org/abs/2311.13166 Zobrazit plný text záznamu View this record from Arxiv