A heterogeneity-aware federated edge learning framework with RIS-empowered wireless aggregation
摘要
As an promising distributed edge learning architecture, federated edge learning (FEEL) facilitates multi-party cooperative model training while preserving data locality. However, the practical deployment of FEEL encounters significant challenges, primarily due to heterogeneity in data distribution and the presence of noisy wireless channels. To tackle these challenges, this paper presents a robust solution for the FEEL system, which integrates a weighted aggregation algorithm based on the index of data completeness to mitigate the adverse effects of heterogeneous data distribution. Concurrently, this solution employs reconfigurable intelligent surface (RIS)-enabled over-the-air computation to optimize communication efficiency and enhance model convergence in heterogeneous wireless environments. Within this framework, we rigorously derive the convergence behavior of the presented FEEL system by taking into account the effects of noisy wireless aggregation and data heterogeneity. Furthermore, we formulate a unified wireless resource optimization problem aimed at minimizing the training loss by jointly optimizing transceiver beamforming and RIS phase shifts. Simulation results demonstrate that the proposed design achieves significant performance improvements compared to several baseline methods.