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 article 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. This article begins by analyzing the reasons for the nonindependent and nonidentically distributed (Non-IID) data collected by devices in this scenario. Subsequently, from the perspective of device selection, this article proposes a node selection model based on near-edge strategy optimization, termed "low node selection in FL" (LCNSFL). Finally, the LCNSFL algorithm is compared with federated averaging algorithms based on random selection strategies and the FedProx algorithm. Experimental results demonstrate that the FL 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.

키워드