AdaptFedDS: adaptive federated learning edge devices selection by using a light-dual reinforcement learning approach
摘要
With the growing volume of data collected on devices and the growing importance of training models directly on these devices, Federated Learning (FL) was introduced. FL enables devices to train a shared model without exchanging raw data while respecting the privacy of data. However, FL faces some challenges including bandwidth limitations and device heterogeneity, hindering simultaneous updates on all devices. In this paper, a lightweight dual-reinforcement learning-based method is proposed for the optimal selection of participating devices in the learning process, as well as determining their level of participation. The proposed mechanism selects the most appropriate devices during communication rounds to improve validation accuracy and minimize rounds. Also, it determines the participation level of each selected device in every round based on their characteristics. According to the experiments, the proposed method can decrease latency compared to FedAvg and FLASH algorithms, while also elevating accuracy.