Dynamic Resource Optimization for Adaptive Federated Learning Empowered by Reconfigurable Intelligent Surfaces

• Published in: ICASSP 2022 - 2022 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP) pp. 4083 - 4087
• Main Authors: Battiloro, Claudio, Merluzzi, Mattia, Di Lorenzo, Paolo, Barbarossa, Sergio
• Format: Conference Proceeding
• Language: English
• Published: IEEE 23.05.2022
• Subjects: Adaptive federated learning; Collaborative work; Dynamic scheduling; Lyapunov optimization; Performance evaluation; Reconfigurable intelligent surfaces; resource allocation; Task analysis; Training; Wireless communication
• ISSN: 2379-190X
• DOI: 10.1109/ICASSP43922.2022.9746891

The aim of this work is to propose a novel dynamic resource allocation strategy for adaptive Federated Learning (FL), in the context of beyond 5G networks endowed with Reconfigurable Intelligent Surfaces (RISs). Due to time-varying wireless channel conditions, communication resources (e.g., set of transmitting devices, transmit powers, bits), computation parameters (e.g., CPU cycles at devices and at server) and RISs reflectivity must be optimized in each communication round, in order to strike the best trade-off between power, latency, and performance of the FL task. Hinging on Lyapunov stochastic optimization, we devise an online strategy able to dynamically allocate these resources, while controlling learning performance in a fully data-driven fashion. Numerical simulations implement distributed training of deep convolutional neural networks, illustrating the effectiveness of the proposed FL strategy endowed with multiple reconfigurable intelligent surfaces.