面向小样本SAR图像识别的自注意力多尺度特征融合网络

Self-attention multiscale feature fusion network for small sample SAR image recognition

  • 摘要: 合成孔径雷达(Synthetic Aperture Radar,SAR)图像标签难以大量获取,存在着大量小样本SAR数据集。SAR图像充满着散斑噪声,直接将卷积神经网络(Convolutional Neural Network,CNN)应用在小样本SAR数据集上难以提取有效特征。针对以上问题,本文提出了一种面向小样本SAR图像识别的自注意力多尺度特征融合网络。首先,将自注意力机制与幽灵模块相结合构建自注意力幽灵模块,并利用该模块替代经典的卷积操作提取SAR图像特征。其次,在网络中添加通道混洗单元以构建多尺度信息融合支路。最后,引入知识蒸馏对设计的网络进行压缩,进一步控制网络参数量。实验结果表明,本文方法在不同工作条件下采集的MSTAR数据集上具有出色的识别性能,在构建的小样本SAR数据集上也表现出良好的鲁棒性。

     

    Abstract: Synthetic Aperture Radar (SAR) image tags were difficult to obtain in large quantities, and there were large numbers of small sample SAR datasets. SAR image was full of scattered noise, and it was difficult to extract effective features by directly applying Convolutional Neural Network (CNN) on small sample SAR datasets. To addressed the above problems, this paper proposed a self-attention multiscale feature fusion network for small sample SAR image recognition. Firstly, the self-attention mechanism was combined with the ghost module to construct the self-attention ghost module, and the module was used to replace the classic convolution operation to extract SAR image features. Secondly, a channel shuffle unit was added to the network to construct multiscale information fusion branchs. Finally, knowledge distillation was introduced to compress the designed network and further control the number of network parameters. The experimental results show that the method in this paper has excellent recognition performance on MSTAR datasets collected under different operating conditions and also exhibits good robustness on small sample SAR datasets constructed.

     

/

返回文章
返回