SAR Target Image Classification Based on Transfer Learning and Model Compression

被引:62
|
作者
Zhong, Chengliang [1 ]
Mu, Xiaodong [1 ]
He, Xiangchen [2 ]
Wang, Jiaxin [2 ]
Zhu, Ming [1 ]
机构
[1] Xian Res Inst Hitech, Comp Dept, Xian 710025, Shaanxi, Peoples R China
[2] Beijing Inst Remote Sensing Technol, Res Ctr Simulat Algorithm, Beijing 100039, Peoples R China
基金
中国国家自然科学基金;
关键词
Convolutional neural networks (CNNs); model compression; synthetic aperture radar (SAR); transfer learning;
D O I
10.1109/LGRS.2018.2876378
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
When convolutional neural networks (CNNs) are applied to the synthetic aperture radar (SAR) image classification, they are prone to overfitting due to scarce SAR image data, and CNNs require a large amount of storage and long computing time, so it is difficult to deploy them on resource constrained devices. This letter proposes a simple and feasible approach that can effectively solve these problems. First, the convolutional layers of the pretrained model on the Image Net data set are transferred, and a new convolutional layer and global pooling layer are added afterward. Then, fine-tuning is performed on the new network from the SAR image data set. Finally, a filter based pruning method is used on the convolutional layers to obtain a compact network. Compared with the all-convolutional network (A-ConvNets) which is the state-of-the-art method on the moving and stationary target acquisition and recognition data set, our method achieves about 3.6 x speedup during forward propagation and 3.7 x compression of the parameters, with only a 1.42% decrease in the accuracy.
引用
收藏
页码:412 / 416
页数:5
相关论文
共 50 条
  • [1] SAR Target Classification Based on Radar Image Luminance Analysis by Deep Learning
    Zhu, Hongliang
    Wang, Weiye
    Leung, Rocky
    [J]. IEEE SENSORS LETTERS, 2020, 4 (03)
  • [2] Convolution neural network SAR image target recognition based on transfer learning
    Chen Lifu
    Wu Hong
    Cui Xianliang
    Guo Zhenghua
    Jia Zhiwei
    [J]. CHINESE SPACE SCIENCE AND TECHNOLOGY, 2018, 38 (06) : 45 - 51
  • [3] Polarimetric SAR Image Classification Based on Discriminative Dictionary Learning Model
    Sang, Cheng Wei
    Sun, Hong
    [J]. MIPPR 2017: REMOTE SENSING IMAGE PROCESSING, GEOGRAPHIC INFORMATION SYSTEMS, AND OTHER APPLICATIONS, 2018, 10611
  • [4] An Image Classification Model for Natural Scenes Based on Model Transfer Learning
    Chen, Yancang
    Zhang, Song
    Tao, Yerong
    Zhai, Shuxin
    [J]. 2024 5TH INTERNATIONAL CONFERENCE ON GEOLOGY, MAPPING AND REMOTE SENSING, ICGMRS 2024, 2024, : 108 - 111
  • [5] Deep Learning Based Compression with Classification Model on CMOS Image Sensors
    Palani, Vinayagam
    Thanarajan, Tamilvizhi
    Krishnamurthy, Anand
    Rajendran, Surendran
    [J]. TRAITEMENT DU SIGNAL, 2023, 40 (03) : 1163 - 1170
  • [6] CNN Model Visualization Method for SAR Image Target Classification
    Miaoge L.
    Bo C.
    Dongsheng W.
    Hongwei L.
    [J]. Journal of Radars, 2024, 13 (02): : 359 - 373
  • [7] An image classification model based on transfer learning for ulcerative proctitis
    Feng Zeng
    Xingcun Li
    Xiaoheng Deng
    Lan Yao
    Guanghui Lian
    [J]. Multimedia Systems, 2021, 27 : 627 - 636
  • [8] An image classification model based on transfer learning for ulcerative proctitis
    Zeng, Feng
    Li, Xingcun
    Deng, Xiaoheng
    Yao, Lan
    Lian, Guanghui
    [J]. MULTIMEDIA SYSTEMS, 2021, 27 (04) : 627 - 636
  • [9] Polarimetric SAR Image Classification Based on Deep Learning and Hierarchical Semantic Model
    Shi J.-F.
    Liu F.
    Lin Y.-H.
    Liu L.
    [J]. Shi, Jun-Fei (shijunfei3@126.com), 1600, Science Press (43): : 215 - 226
  • [10] Target Classification for Single-Channel SAR Images Based on Transfer Learning With Subaperture Decomposition
    Wang, Zhaocheng
    Fu, Xiaoya
    Xia, Kewen
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19