Efficient Pipeline Collaborative DNN Inference in Resource-Constrained UAV Swarm

Recent advancements in unmanned aerial vehicle (UAV) technology have propelled the popularity of edge intelligence (EI) applications with deep learning in UAV swarm. Nevertheless, the high computational demands of deep neural networks (DNNs) conflict with the limited computing power and battery capacity of UAV. Furthermore, many UAV applications require real-time performance such as object detection and recognition. In this paper, we study how to achieve fast DNN inference in UAV swarm by the collaboration of multiple UAVs, and formulate the problem of minimizing the completion time of a series of arriving DNN inference tasks, under memory and energy constraints. To solve the aforementioned challenging problem with combinatorial explosion, we propose an efficient solution exploiting deep reinforcement learning (DRL) with action space simplification to find the allocation strategy of each DNN inference task within a resource-constrained UAV swarm. Simulation results validate the effectiveness of the proposed solution compared to five benchmark algorithms.