Neural network architecture search and model compression for fast prediction of UAS traffic density

Zhenhang Zhang, Chen Luo, M. Cenk Gursoy, Qinru Qiu, Carlos Caicedo, Adrian Solomon, Franco Basti

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

The number of daily small Unmanned Aircraft Systems (sUAS) operations in uncontrolled low altitude airspace is expected to reach into the millions in the future. This makes UAS traffic density prediction a critical and challenging problem. In an Unmanned Aircraft System Traffic Management (UTM) framework, an accurate traffic prediction model is the key to manage congestion in very dense areas and allow us to evaluate different mission and resource provision plans in search for the best solution. In our previous work, a deep neural network (DNN) model has been proposed that predicts the instantaneous traffic density based on mission schedule information. However, one of the main drawbacks of the DNN is the high computational cost, which prevents us from applying the model to search for the best mission plan or best locations of launching and landing zones, because it requires exponentially large numbers of predictions based on different input combinations. In this paper, we aim to reduce the complexity of the neural network model. A neural architecture optimization framework that searches for the best compression ratio for each layer is developed. Overall, we are able to reduce the size of the traffic prediction model by 50%. Furthermore, because the pruning adds more regularization on the model and reduces the potential of overfitting, the compressed model also achieves small improvements in the prediction accuracy.

Original languageEnglish (US)
Title of host publication2021 Integrated Communications Navigation and Surveillance Conference, ICNS 2021
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9781665435840
DOIs
StatePublished - Apr 20 2021
Event2021 Integrated Communications Navigation and Surveillance Conference, ICNS 2021 - Dulles, United States
Duration: Apr 19 2021Apr 23 2021

Publication series

NameIntegrated Communications, Navigation and Surveillance Conference, ICNS
Volume2021-April
ISSN (Print)2155-4943
ISSN (Electronic)2155-4951

Conference

Conference2021 Integrated Communications Navigation and Surveillance Conference, ICNS 2021
Country/TerritoryUnited States
CityDulles
Period4/19/214/23/21

ASJC Scopus subject areas

  • Computer Graphics and Computer-Aided Design
  • Computer Vision and Pattern Recognition
  • Human-Computer Interaction

Fingerprint

Dive into the research topics of 'Neural network architecture search and model compression for fast prediction of UAS traffic density'. Together they form a unique fingerprint.

Cite this