Optimizing Proximity Strategy for Federated Learning Node Selection in the Space-Air-Ground Information Network for Smart Cities

As the Internet of Things (IoT) technology and Artificial Intelligence (AI) technology continue to evolve, many envisaged concepts regarding smart cities are gradually becoming a reality. However, the proliferation of numerous IoT devices in smart cities has led to several challenges. The existing 5G networks are incapable of meeting the requirements of these devices in terms of channel capacity and network coverage. Additionally, traditional cloud-based centralized machine learning methods fail to ensure the privacy of user data. At this juncture, Space-Air-Ground Information Network, along with Federated Learning (FL), are perceived as viable solutions to address these issues. This paper focuses on addressing FL challenges in smart cities using the Space-Air-Ground Information Network. Here, data distribution heterogeneity leads to increased federated training time and higher energy costs. The paper begins by analyzing the reasons for the non-independent and non-identically distributed (NON-IID) data collected by devices in this scenario. Subsequently, from the perspective of device selection, the paper proposes a node selection model based on near-edge strategy optimization, termed "Low Node Selection in Federated Learning" (LCNSFL). Finally, the LCNSFL algorithm is compared with federated av-eraging algorithms based on random selection strategies and the FedProx algorithm. Experimental results demonstrate that the federated learning model aided by the LCNSFL algorithm achieves the target accuracy with fewer communication rounds, considerably reducing the required training time and energy costs compared to the other two algorithms. © 2024 IEEE.