计算机工程与应用 ›› 2021, Vol. 57 ›› Issue (21): 224-233.DOI: 10.3778/j.issn.1002-8331.2007-0080

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

利用同质区特性的高光谱图像迁移学习分类

周绍光,吴昊,赵婵娟,陈仁喜   

  1. 河海大学 地球科学与工程学院,南京 211100
  • 出版日期:2021-11-01 发布日期:2021-11-04

Transfer Learning for Hyperspectral Image Classification Using Homogeneous Area Characteristics

ZHOU Shaoguang, WU Hao, ZHAO Chanjuan, CHEN Renxi   

  1. School of Earth Sciences and Engineering, Hohai University, Nanjing 211100, China
  • Online:2021-11-01 Published:2021-11-04

摘要:

深度卷积神经网络能充分利用特征间的内在联系,提高高光谱影像的可分性,近年来受到了广泛关注。但是,训练深度网络模型对大量标记样本的需求限制了此类方法的应用。将迁移学习思想引入遥感影像分类以减少对标记样本数量的需求。具体研究目标图像中每类只有一个标记样本的情况。通过对目标图像分割得到的同质区扩增目标域的训练样本数量,在此基础上运用深度孪生卷积神经网络减少源域图像与目标域图像的分布差异,实现对目标高光谱图像的最终分类。实验结果表明:同质区和孪生卷积网络的结合可提高半监督迁移学习分类的效果,较好地解决跨区域的高光谱图像分类问题。

关键词: 高光谱图像分类, 孪生神经网络, 图像分割, 同质区, 单样本学习, 半监督迁移学习

Abstract:

Deep convolutional neural networks can fully utilize the inner connections between features which improve the separability of the hyperspectral images and have been widely studied in recent years. However, the need for an abundant labeled sample for training deep network models limits the application of such methods. To cut the need for the number of labeled samples, the technology of transfer learning is adopted for remote sensing classification in this paper. The one-shot problem is concerned in which only a single sample of each class is available. Data augmentation is carried out, based on homogenous regions obtained by image segmentation. Using the deep Siamese CNN trained by augmentation samples, this paper can cut the divergence between the distributions of the source domain and target domain, and classify the target image. Experiment results based on cross-region hyperspectral image datasets show the effect of the combination of deep Siamese CNN and homogenous areas on semi-supervised transfer learning.

Key words: hyperspectral image classification, siamese neural network, image segmentation, homogeneous area, one-shot learning, semi-supervised transfer learning