计算机科学 ›› 2020, Vol. 47 ›› Issue (5): 124-128.doi: 10.11896/jsjkx.190400136

• 计算机图形学&多媒体 • 上一篇    下一篇

基于小样本学习的SAR图像识别

汪航1, 陈晓2, 田晟兆1, 陈端兵1,3   

  1. 1 电子科技大学大数据研究中心 成都611731
    2 陆军参谋部信息保障室 北京100042
    3 电子科技大学数字文化与传媒研究中心 成都611731
  • 收稿日期:2019-04-25 出版日期:2020-05-15 发布日期:2020-05-19
  • 通讯作者: 陈端兵(dbchen@uestc.edu.cn)
  • 作者简介:cnwanghhh@163.com
  • 基金资助:
    国家自然科学基金(61673085,61433014);国家重点研发计划(2017YFC1601005)

SAR Image Recognition Based on Few-shot Learning

WANG Hang1, CHEN Xiao2, TIAN Sheng-zhao1, CHEN Duan-bing1,3   

  1. 1 Big Data Research Center,University of Electronic Science and Technology of China,Chengdu 611731,China
    2 Information Assurance Office of Army Staff,Beijing 100042,China
    3 Center for Digitized Culture and Media,University of Electronic Science and Technology of China,Chengdu 611731,China
  • Received:2019-04-25 Online:2020-05-15 Published:2020-05-19
  • About author:WANG Hang,born in 1997,postgra-duate.His main research interests include big data and cloud computing.
    CHEN Duan-bing,born in 1971,professor.His main research interests include big data mining,complex network,information spreading and recommending.
  • Supported by:
    This work was supported by the National Natural Science Foundation of China (61673085,61433014) and National Key Research and Development Project (2017YFC1601005).

摘要: 深度学习已成为图像识别领域的一个研究热点。与传统图像识别方法不同,深度学习从大量数据中自动学习特征,并且具有强大的自学习能力和高效的特征表达能力。但在小样本条件下,传统的深度学习方法如卷积神经网络难以学习到有效的特征,造成图像识别的准确率较低。因此,提出一种新的小样本条件下的图像识别算法用于解决SAR图像的分类识别。该算法以卷积神经网络为基础,结合自编码器,形成深度卷积自编码网络结构。首先对图像进行预处理,使用2D Gabor滤波增强图像,在此基础上对模型进行训练,最后构建图像分类模型。该算法设计的网络结构能自动学习并提取小样本图像中的有效特征,进而提高识别准确率。在MSTAR数据集的10类目标分类中,选择训练集数据中10%的样本作为新的训练数据,其余数据为验证数据,并且,测试数据在卷积神经网络中的识别准确率为76.38%,而在提出的卷积自编码结构中的识别准确率达到了88.09%。实验结果表明,提出的算法在小样本图像识别中比卷积神经网络模型更加有效。

关键词: 卷积神经网络, 深度学习, 小样本学习, 自编码器

Abstract: Deep learning has become a research hotspot in the field of image recognition.Different from traditional image recognition methods,deep learning is to automatically learn features from a large amount data and has a strong ability of feature learning and representation.However,under the condition of small samples,the traditional deep learning methods such as convolutional neural network are difficult to learn effective features,resulting in low image recognition accuracy.Thus,a new image recognition algorithm under small samples was proposed to solve the classification and recognition of SAR images.On the basis of convolutional neural network,it combines convolution operation with autoencoder to form a deep convolutional autoencoder network structure.The algorithm firstly preprocesses the image and enhances the image using 2D Gabor filter,and thentrains the model,finally,constructsthe image classification model.The proposed model can automatically learn and extract effective features from small sample images,and improve the recognition accuracy.On 10 categories of target classification of MSTAR data set,10% samples from the training data were selected as new training data,the rest were valid data,and the recognition accuracy of the test data in the convolutional neural network is 76.38%,while that in the proposed convolutional autoencoder is 88.09%.Experimental results show that the proposed algorithm is more effective than convolutional neural network in small sample image recognition.

Key words: Autoencoder, Convolutional neural network, Deep learning, Few-shot learning

中图分类号: 

  • TP301.6
[1]FENG J,LIU L,CAO X,et al.Marginal Stacked Autoencoder With Adaptively-Spatial Regularization for Hyperspectral Image Classification [J].IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing,2018,11(9):3297-3311.
[2]MOCANU C,MOCANU E.One-Shot Learning using Mixture of Variational Autoencoders:a Generalization Learning approach [C]//Proceedings of the 17th International Conference on Autonomous Agents and MultiAgent Systems.Stockholm,Sweden,2018:2016-2018.
[3]YANG Q,SUN F.Small sample learning with high order contractive auto-encoders and application in SAR images[J].Science China Information Sciences,2018,61(9):271-273.
[4]WANG X,MU S,SHI A,et al.A Stacked Denoising Autoenco-der Based on Supervised Pre-training[C]//Smart Innovations in Communication and Computational Sciences.Singapore:Sprin-ger,2019:139-146.
[5]WANG H,CHEN S,XU F,et al.Application of deep-learning algorithms to MSTAR data[C]//2015 IEEE International Geoscience and Remote Sensing Symposium (IGARSS).Milan,2015:3743-3745.
[6]ZAIED S,TOUMI A,KHENCHAF A.Target classificationusing convolutional deep learning and auto-encoder models[C]//2018 4th International Conference on Advanced Technologies for Signal and Image Processing (ATSIP).Sousse,2018:1-6.
[7]ZHAO F.SAR Image Target Recognition Based on Deep Lear-ning Network[D].Xi'an:Xidian University,2017.
[8]JIA Z,GUANGCHANG D,FENG C,et al.A Deep LearningFusion Recognition Method Based On SAR Image Data[J].Procedia Computer Science,2019,147:533-541.
[9]CHEN Y,MENG H,WEN X,et al.Classification methods of a small sample target object in the sky based on the higher layer visualizing feature and transfer learning deep networks[J].EURASIP Journal on Wireless Communications and Networking,2018,2018(1):127.
[10]LADES M,VORBRUGGEN J C,BUHMANN J.Distortion invariant object recognition in the dynamic link architecture[J].IEEE Transaction on Computer,1993,42(3):300-311.
[11]JONES J P,PALMER L A.An evaluation of the two-dimensional Gabor filter model of simple receptive fields in cat striate cortex[J].Journal of Neurophysiology,1987,58(6):1233-1258.
[12]MOU H.Face Recognition Method Based on Gabor Features[D].Wuhan:South-Central University for Nationalities,2013.
[13]NI W,YAN W,WU J,et al.2D Gabor Filter Enhancing andAdaptiveThresholding for MSTAR Image[J].Opto-Electronic Engineering,2013,40(3):87-93.
[14] SIMONYAN K,ZISSERMAN A.Very Deep Convolutional Networks for Large-Scale Image Recognition[J].arXiv:14096.
[1] 饶志双, 贾真, 张凡, 李天瑞.
基于Key-Value关联记忆网络的知识图谱问答方法
Key-Value Relational Memory Networks for Question Answering over Knowledge Graph
计算机科学, 2022, 49(9): 202-207. https://doi.org/10.11896/jsjkx.220300277
[2] 汤凌韬, 王迪, 张鲁飞, 刘盛云.
基于安全多方计算和差分隐私的联邦学习方案
Federated Learning Scheme Based on Secure Multi-party Computation and Differential Privacy
计算机科学, 2022, 49(9): 297-305. https://doi.org/10.11896/jsjkx.210800108
[3] 王冠宇, 钟婷, 冯宇, 周帆.
基于矢量量化编码的协同过滤推荐方法
Collaborative Filtering Recommendation Method Based on Vector Quantization Coding
计算机科学, 2022, 49(9): 48-54. https://doi.org/10.11896/jsjkx.210700109
[4] 周乐员, 张剑华, 袁甜甜, 陈胜勇.
多层注意力机制融合的序列到序列中国连续手语识别和翻译
Sequence-to-Sequence Chinese Continuous Sign Language Recognition and Translation with Multi- layer Attention Mechanism Fusion
计算机科学, 2022, 49(9): 155-161. https://doi.org/10.11896/jsjkx.210800026
[5] 徐涌鑫, 赵俊峰, 王亚沙, 谢冰, 杨恺.
时序知识图谱表示学习
Temporal Knowledge Graph Representation Learning
计算机科学, 2022, 49(9): 162-171. https://doi.org/10.11896/jsjkx.220500204
[6] 李宗民, 张玉鹏, 刘玉杰, 李华.
基于可变形图卷积的点云表征学习
Deformable Graph Convolutional Networks Based Point Cloud Representation Learning
计算机科学, 2022, 49(8): 273-278. https://doi.org/10.11896/jsjkx.210900023
[7] 王剑, 彭雨琦, 赵宇斐, 杨健.
基于深度学习的社交网络舆情信息抽取方法综述
Survey of Social Network Public Opinion Information Extraction Based on Deep Learning
计算机科学, 2022, 49(8): 279-293. https://doi.org/10.11896/jsjkx.220300099
[8] 郝志荣, 陈龙, 黄嘉成.
面向文本分类的类别区分式通用对抗攻击方法
Class Discriminative Universal Adversarial Attack for Text Classification
计算机科学, 2022, 49(8): 323-329. https://doi.org/10.11896/jsjkx.220200077
[9] 姜梦函, 李邵梅, 郑洪浩, 张建朋.
基于改进位置编码的谣言检测模型
Rumor Detection Model Based on Improved Position Embedding
计算机科学, 2022, 49(8): 330-335. https://doi.org/10.11896/jsjkx.210600046
[10] 陈泳全, 姜瑛.
基于卷积神经网络的APP用户行为分析方法
Analysis Method of APP User Behavior Based on Convolutional Neural Network
计算机科学, 2022, 49(8): 78-85. https://doi.org/10.11896/jsjkx.210700121
[11] 朱承璋, 黄嘉儿, 肖亚龙, 王晗, 邹北骥.
基于注意力机制的医学影像深度哈希检索算法
Deep Hash Retrieval Algorithm for Medical Images Based on Attention Mechanism
计算机科学, 2022, 49(8): 113-119. https://doi.org/10.11896/jsjkx.210700153
[12] 孙奇, 吉根林, 张杰.
基于非局部注意力生成对抗网络的视频异常事件检测方法
Non-local Attention Based Generative Adversarial Network for Video Abnormal Event Detection
计算机科学, 2022, 49(8): 172-177. https://doi.org/10.11896/jsjkx.210600061
[13] 檀莹莹, 王俊丽, 张超波.
基于图卷积神经网络的文本分类方法研究综述
Review of Text Classification Methods Based on Graph Convolutional Network
计算机科学, 2022, 49(8): 205-216. https://doi.org/10.11896/jsjkx.210800064
[14] 胡艳羽, 赵龙, 董祥军.
一种用于癌症分类的两阶段深度特征选择提取算法
Two-stage Deep Feature Selection Extraction Algorithm for Cancer Classification
计算机科学, 2022, 49(7): 73-78. https://doi.org/10.11896/jsjkx.210500092
[15] 张颖涛, 张杰, 张睿, 张文强.
全局信息引导的真实图像风格迁移
Photorealistic Style Transfer Guided by Global Information
计算机科学, 2022, 49(7): 100-105. https://doi.org/10.11896/jsjkx.210600036
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!