Empowering the Edge Intelligence by Air-Ground Integrated Federated Learning in 6G Networks

Ubiquitous intelligence has been widely recognized as a critical vision of the future sixth generation (6G) networks, which implies the intelligence over the whole network from the core to the edge including end devices. Nevertheless, fulfilling such vision, particularly the intelligence at the edge, is extremely challenging, due to the limited resources of edge devices as well as the ubiquitous coverage envisioned by 6G. To empower the edge intelligence, in this article, we propose a novel framework called AGIFL (Air-Ground Integrated Federated Learning), which organically integrates air-ground integrated networks and federated learning (FL). In the AGIFL, leveraging the flexible on-demand 3D deployment of aerial nodes such as unmanned aerial vehicles (UAVs), all the nodes can collaboratively train an effective learning model by FL. We also conduct a case study to evaluate the effect of two different deployment schemes of the UAV over the learning and network performance. Last but not the least, we highlight several technical challenges and future research directions in the AGIFL.

[1]  Qiang Yang,et al.  Federated Deep Reinforcement Learning , 2019, 1901.08277.

[2]  Walid Saad,et al.  A Vision of 6G Wireless Systems: Applications, Trends, Technologies, and Open Research Problems , 2019, IEEE Network.

[3]  Mehdi Bennis,et al.  Communication-Efficient Massive UAV Online Path Control: Federated Learning Meets Mean-Field Game Theory , 2020, IEEE Transactions on Communications.

[4]  Blaise Agüera y Arcas,et al.  Communication-Efficient Learning of Deep Networks from Decentralized Data , 2016, AISTATS.

[5]  Albert Y. Zomaya,et al.  Federated Learning over Wireless Networks: Optimization Model Design and Analysis , 2019, IEEE INFOCOM 2019 - IEEE Conference on Computer Communications.

[6]  Xiang Li,et al.  On the Convergence of FedAvg on Non-IID Data , 2019, ICLR.

[7]  Wenchao Xu,et al.  Air-Ground Integrated Mobile Edge Networks: Architecture, Challenges, and Opportunities , 2018, IEEE Communications Magazine.

[8]  Hubert Eichner,et al.  Towards Federated Learning at Scale: System Design , 2019, SysML.

[9]  Rui Zhang,et al.  Energy-Efficient UAV Communication With Trajectory Optimization , 2016, IEEE Transactions on Wireless Communications.

[10]  Chunyan Miao,et al.  Towards Federated Learning in UAV-Enabled Internet of Vehicles: A Multi-Dimensional Contract-Matching Approach , 2020, IEEE Transactions on Intelligent Transportation Systems.

[11]  Kin K. Leung,et al.  Adaptive Federated Learning in Resource Constrained Edge Computing Systems , 2018, IEEE Journal on Selected Areas in Communications.

[12]  BOUZIANE BRIK,et al.  Federated Learning for UAVs-Enabled Wireless Networks: Use Cases, Challenges, and Open Problems , 2020, IEEE Access.

[13]  Jun Zhao,et al.  Artificial-Intelligence-Enabled Intelligent 6G Networks , 2020, IEEE Network.

[14]  Omid Semiari,et al.  Federated Learning in the Sky: Joint Power Allocation and Scheduling with UAV Swarms , 2020, ICC 2020 - 2020 IEEE International Conference on Communications (ICC).

[15]  Peter Richtárik,et al.  Federated Learning: Strategies for Improving Communication Efficiency , 2016, ArXiv.