An Efficient and Lightweight CNN Model With Soft Quantification for Ship Detection in SAR Images

被引:26
作者
Yang, Xi [1 ]
Zhang, Jianan [1 ]
Chen, Chengzeng [2 ]
Yang, Dong [3 ]
机构
[1] Xidian Univ, State Key Lab Integrated Serv Networks, Sch Telecommun Engn, Xian 710071, Peoples R China
[2] Beijing Inst Infinite Elect Measurement, Lab Pinghu, Pinghu 314201, Peoples R China
[3] Xian Inst Space Radio Technol, Xian 710100, Peoples R China
来源
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING | 2022年 / 60卷
基金
中国国家自然科学基金;
关键词
Lightweight network; ship detection; soft quantization; synthetic aperture radar (SAR); NEURAL-NETWORK; DISCRIMINATION; YOLO;
D O I
10.1109/TGRS.2022.3186155
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
Convolutional neural networks (CNNs) have been widely used for synthetic aperture radar (SAR) target detection. Typical methods based on CNN have obtained favorable detection accuracy at the cost of high model complexity, and thus are difficult to be directly applied to real-time satellites on board as well as maritime rescue. To deal with this problem, this article proposes an efficient and lightweight target detection network incorporating soft quantization. First, to compensate for the lack of accuracy caused by lightweight networks, a feature fusion module called split bidirectional feature pyramid network is proposed to alleviate the interference of complex background on SAR images. Meanwhile, to adapt the lightweight network and the feature fusion module, a linear transformation module is presented to enhance the linear representation of the model via learnable parameters. Eventually, to make the model size smaller, a soft quantization algorithm is proposed to reduce the accuracy degradation caused by quantization errors. We validate the robustness of the model in several publicly available datasets. Experimental results show that our model achieves 97.0% detection accuracy on SAR ship detection dataset, with a 0.9% accuracy improvement compared to mainstream methods using less than 15x the number of parameters and less than 6x the number of flops.
引用
收藏
页数:13
相关论文
共 58 条
[1]   SAR Target Classification Using the Multikernel-Size Feature Fusion-Based Convolutional Neural Network [J].
Ai, Jiaqiu ;
Mao, Yuxiang ;
Luo, Qiwu ;
Jia, Lu ;
Xing, Mengdao .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[2]   Robust CFAR Ship Detector Based on Bilateral-Trimmed-Statistics of Complex Ocean Scenes in SAR Imagery: A Closed-Form Solution [J].
Ai, Jiaqiu ;
Mao, Yuxiang ;
Luo, Qiwu ;
Xing, Mengdao ;
Jiang, Kai ;
Jia, Lu ;
Yang, Xingming .
IEEE TRANSACTIONS ON AEROSPACE AND ELECTRONIC SYSTEMS, 2021, 57 (03) :1872-1890
[3]   Outliers-Robust CFAR Detector of Gaussian Clutter Based on the Truncated-Maximum-Likelihood- Estimator in SAR Imagery [J].
Ai, Jiaqiu ;
Luo, Qiwu ;
Yang, Xuezhi ;
Yin, Zhiping ;
Xu, Hao .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2020, 21 (05) :2039-2049
[4]   Multi-Scale Rotation-Invariant Haar-Like Feature Integrated CNN-Based Ship Detection Algorithm of Multiple-Target Environment in SAR Imagery [J].
Ai, Jiaqiu ;
Tian, Ruitian ;
Luo, Qiwu ;
Jin, Jing ;
Tang, Bo .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2019, 57 (12) :10070-10087
[5]   An Adaptively Truncated Clutter-Statistics-Based Two-Parameter CFAR Detector in SAR Imagery [J].
Ai, Jiaqiu ;
Yang, Xuezhi ;
Song, Jitao ;
Dong, Zhangyu ;
Jia, Lu ;
Zhou, Fang .
IEEE JOURNAL OF OCEANIC ENGINEERING, 2018, 43 (01) :267-279
[6]  
[Anonymous], 2020, 2020 GAOFEN CHALLENG
[7]  
Bochkovskiy A, 2020, Arxiv, DOI arXiv:2004.10934
[8]   Ship Detection Based on YOLOv2 for SAR Imagery [J].
Chang, Yang-Lang ;
Anagaw, Amare ;
Chang, Lena ;
Wang, Yi Chun ;
Hsiao, Chih-Yu ;
Lee, Wei-Hong .
REMOTE SENSING, 2019, 11 (07)
[9]   You Only Look One-level Feature [J].
Chen, Qiang ;
Wang, Yingming ;
Yang, Tong ;
Zhang, Xiangyu ;
Cheng, Jian ;
Sun, Jian .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :13034-13043
[10]  
Courbariaux M, 2016, Arxiv, DOI arXiv:1602.02830