Enhancing Inference Time and Memory Utilization for Machine Learning in Resource Constrained Internet of Battlefield Things
ARTIFICIAL INTELLIGENCE AND MACHINE LEARNING FOR MULTI-DOMAIN OPERATIONS APPLICATIONS IV(2022)
Abstract
Convolution Neural Network (CNN) models have demonstrated remarkable success for many applications including computer vision in recent years. However, the increase in accuracy achieved so far is at the cost of memory and complex computations. The CNN models are growing deeper and wider making it difficult to be fit in a single device that has limited resources. Moreover, the inference time of such models is large enough to be unfit to apply for the real-time mission critical applications like the Internet of Battlefield Things (IoBT). In IoBT, where the unmanned aerial vehicles (UAVs) flying in the battlefield zone and capturing images, require accurate learning and immediate inference. It becomes problematic if the learning model does not fit in a single UAV when it has limited resources. Considering the aforementioned issues, in this paper, we study a formal approach to improve the inference time and memory utilization in resource constrained IoBT. We consider that multiple UAVs are involved in the inference process in which we apply spatially parallel convolution and pooling operations for all the convolution layers and pooling layers as well as model parallelism for fully connected (FC) layers. Finally, we present the numerical results for varying number of participating UAVs, input data/image sizes, and communication speeds.
MoreTranslated text
Key words
Convolution Neural Network, Computer Vision, Internet of Battlefield, Unmanned Aerial Vehicles, Spatially Parallel Convolution, Model Parallelism
AI Read Science
Must-Reading Tree
Example
![](https://originalfileserver.aminer.cn/sys/aminer/pubs/mrt_preview.jpeg)
Generate MRT to find the research sequence of this paper
Chat Paper
Summary is being generated by the instructions you defined