Joint Self-Organizing Maps and Knowledge Distillation-Based Communication-Efficient Federated Learning for Resource-Constrained UAV-IoT Systems

Gad Gad,Aya Farrag,Ahmed Aboulfotouh,Khaled Bedda,Zubair Md Fadlullah,Mostafa M. Fouda
DOI: https://doi.org/10.1109/jiot.2023.3349295
IF: 10.6
2024-01-01
IEEE Internet of Things Journal
Abstract:The adoption of IoT and monitoring devices in 5G and beyond networks has been widespread. Unmanned Aerial vehicles (UAVs) have shown success in connecting rural and remote areas due to the high cost of deploying infrastructures like cellular network base stations and optical fiber connections in vast landscapes with sparse populations. The constrained energy of UAVs results in limited coverage area and flight time, which in turn reduces the potential of UAVs to provide task-oriented wireless communication links. In this paper, we explore path optimization and transmission organization algorithms to minimize flight time and extend the range of UAVs performing collaborative Federated Learning (FL) among geographically dispersed nodes communicating through wireless connections offered by UAVs coupled with Device-to-Device (D2D) networks. The UAV orchestrates federated learning (FL) between spatially scattered homes via LoRa wireless communication. We formulate the drone path optimization as a Traveling Salesman Problem (TSP) and employ Self-Organizing Maps (SOM) for path planning. Additionally, Knowledge Distillation (KD)-based FL is used to reduce communication overhead for the resource-constrained UAV-IoT system. Experimental results demonstrate SOM’s ability to represent the topological structure of nodes and produce a cost-efficient Hamiltonian cycle, from which the drone path is derived. Our results demonstrate the communication efficiency and utility of KD-based FL compared to model-based FL methods. The proposed hybrid solution enables energy-constrained UAVs to perform FL over large areas leveraging a shared dataset for KD and a SOM-based path optimization algorithm.
computer science, information systems,telecommunications,engineering, electrical & electronic
What problem does this paper attempt to address?