Over-the-Air Federated Learning in Digital Twins Empowered UAV Swarms

The development of Unmanned Aerial Vehicles (UAVs) offers new prospects for emerging applications in the Industrial Internet of Things (IIoT) networks. With the assistance of Digital Twin (DT), a real-time understanding of physical entities can be constructed for dynamic perception and decision-making. However, DT modeling requires distributed data aggregation, resulting in privacy disclosure and communication burden. Therefore, we propose the digital twin edge network by integrating the DT technology and edge computing, which leverages an over-the-air computation enabled federated learning architecture for an efficient and secure DT model construction. Specifically, we propose a heterogeneity-aware and energy-conscious device scheduling mechanism, considering the update importance, channel condition, and computation capacity based on a probabilistic scheduling framework. To enhance energy efficiency, we introduce a virtual queue to track the difference between the cumulative energy consumption and budget. Additionally, we design a low-complexity scheduling algorithm to solve the optimization problem. Simulation results validate the superiority of our proposed mechanism in improving the test accuracy and energy efficiency in a heterogeneous and energy-constrained environment. Moreover, the proposed mechanism demonstrates significant advantages when employed to highly heterogeneous datasets, and exhibits a certain level of robustness to mapping errors arising from the utilization of DT technique.