Slim and Efficient Neural Network Design for Resource-Constrained SAR Target Recognition

被引:42
作者
Chen, Hongyi [1 ]
Zhang, Fan [1 ]
Tang, Bo [2 ]
Yin, Qiang [1 ]
Sun, Xian [3 ]
机构
[1] Beijing Univ Chem Technol, Coll Informat Sci & Technol, Beijing 100029, Peoples R China
[2] Mississippi State Univ, Dept Elect & Comp Engn, Starkville, MS 39759 USA
[3] Chinese Acad Sci, Inst Elect, Beijing 100190, Peoples R China
基金
中国国家自然科学基金;
关键词
deep learning; synthetic aperture radar (SAR); automatic target recognition (ATR); model compression; fast algorithm; ALGORITHM;
D O I
10.3390/rs10101618
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
Deep convolutional neural networks (CNN) have been recently applied to synthetic aperture radar (SAR) for automatic target recognition (ATR) and have achieved state-of-the-art results with significantly improved recognition performance. However, the training period of deep CNN is long, and the size of the network is huge, sometimes reaching hundreds of megabytes. These two factors of deep CNN hinders its practical implementation and deployment in real-time SAR platforms that are typically resource-constrained. To address this challenge, this paper presents three strategies of network compression and acceleration to decrease computing and memory resource dependencies while maintaining a competitive accuracy. First, we introduce a new weight-based network pruning and adaptive architecture squeezing method to reduce the network storage and the time of inference and training process, meanwhile maintain a balance between compression ratio and classification accuracy. Then we employ weight quantization and coding to compress the network storage space. Due to the fact that the amount of calculation is mainly reflected in the convolution layer, a fast approach for pruned convolutional layers is proposed to reduce the number of multiplication by exploiting the sparsity in the activation inputs and weights. Experimental results show that the convolutional neural networks for SAR-ATR can be compressed by without loss of accuracy, and the number of multiplication can be reduced by. Combining these strategies, we can easily load the network in resource-constrained platforms, speed up the inference process to get the results in real-time or even retrain a more suitable network with new image data in a specific situation.
引用
收藏
页数:15
相关论文
共 38 条
[1]  
[Anonymous], J ELECT ENG
[2]  
[Anonymous], P SPIE
[3]  
[Anonymous], 2014, 14126115 ARXIV
[4]  
[Anonymous], J SYST ENG ELECT
[5]  
[Anonymous], 2010, P EUR C SYNTH AP RAD
[6]  
[Anonymous], 180206367 ARXIV
[7]  
[Anonymous], 150404788 ARXIV
[8]  
[Anonymous], 2015, 150602626 ARXIV
[9]  
Bentes C, 2015, INT GEOSCI REMOTE SE, P3703, DOI 10.1109/IGARSS.2015.7326627
[10]  
Chang-Shun Liu, 2015, 2015 IEEE Sensors. Proceedings, P1, DOI 10.1109/ICSENS.2015.7370538