A KNOWLEDGE DISTILLATION METHOD BASED ON IQE ATTENTION MECHANISM FOR TARGET RECOGNITION IN SAR IMAGERY

被引:3
作者
Wang, Jielei [1 ]
Jiang, Ting [1 ]
Cui, Zongyong [1 ]
Cao, Zongjie [1 ]
Cao, Changjie [1 ]
机构
[1] Univ Elect Sci & Technol China, Sch Informat & Commun Engn, Chengdu, Peoples R China
来源
2022 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2022) | 2022年
基金
中国国家自然科学基金;
关键词
Deep convolutional neural network; SAR-ATR; Knowledge distillation; Image quality evaluation algorithm; Knowledge Processing Unit (KPU);
D O I
10.1109/IGARSS46834.2022.9883376
中图分类号
P [天文学、地球科学];
学科分类号
07 ;
摘要
The huge computing and storage requirements of deep convolutional neural networks (DCNNs) limit their application on edge computing devices. In this article, we propose an attention mechanism based on the feature map quality evaluation algorithm (IQE). The knowledge distillation method based on the IQE attention mechanism uses the IQE method to identify important knowledge in the pre-trained SAR target recognition deep neural network. Then in the process of knowledge distillation, the lightweight network is forced to focus on the learning of important knowledge. Through this mechanism, the method proposed in this paper can efficiently transfer the knowledge of the pre-trained SAR target recognition network to the lightweight network, which makes it is possible to deploy the SAR target recognition algorithm on the edge computing platform. Comparison experiments with several commonly used knowledge distillation methods have proved the effectiveness of our proposed method. In addition, we also verified the performance of the lightweight network obtained by our method on the edge platform based on the K210 processor.
引用
收藏
页码:1043 / 1046
页数:4
相关论文
共 5 条
[1]  
Hinton G., 2015, arXiv
[2]   AN IQE CRITERION-BASED METHOD FOR SAR IMAGES CLASSIFICATION NETWORK PRUNING [J].
Wang, Jielei ;
Cui, Zongyong ;
Cao, Zongjie ;
Wang, Hanzeng ;
Cao, Changjie .
2021 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM IGARSS, 2021, :3593-3596
[3]  
Romero A., 2015, ICLR
[4]   Similarity-Preserving Knowledge Distillation [J].
Tung, Frederick ;
Mori, Greg .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :1365-1374
[5]  
Zagoruyko S., 2016, BRIT MACH VIS C, DOI [DOI 10.5244/C.30.87, 10.5244/C.30.87]