Communication-Efficient Federated Learning for LEO Satellite Networks Integrated with HAPs Using Hybrid NOMA-OFDM

Mohamed Elmahallawy,Tie Luo,Khaled Ramadan
2024-02-16
Abstract:Space AI has become increasingly important and sometimes even necessary for government, businesses, and society. An active research topic under this mission is integrating federated learning (FL) with satellite communications (SatCom) so that numerous low Earth orbit (LEO) satellites can collaboratively train a machine learning model. However, the special communication environment of SatCom leads to a very slow FL training process up to days and weeks. This paper proposes NomaFedHAP, a novel FL-SatCom approach tailored to LEO satellites, that (1) utilizes high-altitude platforms (HAPs) as distributed parameter servers (PS) to enhance satellite visibility, and (2) introduces non-orthogonal multiple access (NOMA) into LEO to enable fast and bandwidth-efficient model transmissions. In addition, NomaFedHAP includes (3) a new communication topology that exploits HAPs to bridge satellites among different orbits to mitigate the Doppler shift, and (4) a new FL model aggregation scheme that optimally balances models between different orbits and shells. Moreover, we (5) derive a closed-form expression of the outage probability for satellites in near and far shells, as well as for the entire system. Our extensive simulations have validated the mathematical analysis and demonstrated the superior performance of NomaFedHAP in achieving fast and efficient FL model convergence with high accuracy as compared to the state-of-the-art.
Machine Learning,Artificial Intelligence,Distributed, Parallel, and Cluster Computing
What problem does this paper attempt to address?
The paper primarily addresses the communication efficiency issues faced by Federated Learning (FL) in the integration of Low Earth Orbit (LEO) satellite networks and High Altitude Platforms (HAPs). Specifically, the paper aims to solve the following key problems: 1. **Addressing the slow training problem of Federated Learning in satellite communications**: Due to the unique environment of satellite communications, traditional Federated Learning methods encounter very slow training processes in LEO satellite networks, which may take days or even weeks to complete a full training cycle. 2. **Improving communication efficiency using Non-Orthogonal Multiple Access (NOMA)**: A new Federated Learning framework, NomaFedHAP, is proposed. This framework combines NOMA technology, allowing LEO satellites on different orbital shells to concurrently transmit models, thereby significantly improving bandwidth utilization and model transmission speed. 3. **Introducing High Altitude Platforms as distributed parameter servers**: To enhance satellite visibility and reduce the impact of Doppler shifts, the scheme utilizes multiple HAPs as distributed parameter servers. These platforms collaborate to improve the overall communication efficiency of the system. 4. **Optimizing the model aggregation strategy**: A new model aggregation scheme is proposed, which balances the number of models between different orbits, avoids skewed global models, and ensures the quality of the overall model. 5. **Improving the communication topology**: A new communication topology is designed, where HAPs not only serve as parameter servers but also act as relay nodes for satellites between different orbits, effectively mitigating the Doppler shift problem. In summary, the goal of the paper is to significantly enhance the training speed and efficiency of Federated Learning in LEO satellite networks through a series of technological innovations, including the introduction of NOMA, the utilization of HAPs, and improved model aggregation and communication strategies, thereby achieving faster and more efficient model convergence.