计算机工程与应用 ›› 2019, Vol. 55 ›› Issue (17): 191-198.DOI: 10.3778/j.issn.1002-8331.1904-0077

• 图形图像处理 • 上一篇    下一篇

基于伪标签深度学习的高光谱影像半监督分类

刘丽丽,周绍光,赵婵娟,丁倩   

  1. 河海大学 地球科学与工程学院,南京 211100
  • 出版日期:2019-09-01 发布日期:2019-08-30

Semi-Supervised Classification of Hyperspectral Image Based on Pseudo-Label Deep Learning

LIU Lili, ZHOU Shaoguang, ZHAO Chanjuan, DING Qian   

  1. College of Earth Sciences and Engineering, Hehai University, Nanjing 211100, China
  • Online:2019-09-01 Published:2019-08-30

摘要: 研究了一种仅利用少量标记点训练深度卷积神经网络并对高光谱影像进行分类的方法。以图像分割获得的同质区增加训练样本数目;借助这些增加的样本训练初始分类器并预测所有未知点的初始类别;将每一初始类别聚集为适当的类簇,以类簇号作为伪标签对深度卷积网进行预训练;最后利用经过同质区增加的训练样本精调预训练深度卷积网。实验结果证明新方法可以在仅用少量实际标记样本的情况下成功地训练深度卷积网,对高光谱数据进行有效分类。

关键词: 卷积神经网络(CNN), 伪标签, 半监督分类, 高光谱影像

Abstract: In view of the problem of needing a large number of training samples for the depth network, this paper proposes a classification method that deep networks are effectively trained by limited labeled data. The homogeneous region increases the number of training samples. The initial classifier is trained by the increased training samples and predicts the initial category of all unknown pixels. Each initial category is aggregated into the appropriate cluster and cluster labels as a pseudo label. Hyperspectral data together with their pseudo labels are used to pre-train a deep convolutional neural network. Finally, the pre-trained deep convolution network is fine-tuned by the limited training samples added by homogenous regions. Experimental results with real hyperspectral image data sets demonstrate that the new method can successfully train deep convolutional networks with a small number of actual labeled samples and efficiently classify hyperspectral data.

Key words: Convolutional Neural Network(CNN), pseudo labels, semi-supervised classification, hyperspectral imagery