高级检索
肖志云, 蒋家旭, 倪晨. 自适应深层残差3D-CNN高光谱图像快速分类算法[J]. 计算机辅助设计与图形学学报, 2019, 31(11): 2017-2029. DOI: 10.3724/SP.J.1089.2019.17552
引用本文: 肖志云, 蒋家旭, 倪晨. 自适应深层残差3D-CNN高光谱图像快速分类算法[J]. 计算机辅助设计与图形学学报, 2019, 31(11): 2017-2029. DOI: 10.3724/SP.J.1089.2019.17552
Xiao Zhiyun, Jiang Jiaxu, Ni Chen. Spectral-Spatial Classification of Hyperspectral Image Based on Self-Adaptive Deep Residual 3D Convolutional Neural Network[J]. Journal of Computer-Aided Design & Computer Graphics, 2019, 31(11): 2017-2029. DOI: 10.3724/SP.J.1089.2019.17552
Citation: Xiao Zhiyun, Jiang Jiaxu, Ni Chen. Spectral-Spatial Classification of Hyperspectral Image Based on Self-Adaptive Deep Residual 3D Convolutional Neural Network[J]. Journal of Computer-Aided Design & Computer Graphics, 2019, 31(11): 2017-2029. DOI: 10.3724/SP.J.1089.2019.17552

自适应深层残差3D-CNN高光谱图像快速分类算法

Spectral-Spatial Classification of Hyperspectral Image Based on Self-Adaptive Deep Residual 3D Convolutional Neural Network

  • 摘要: 为了实现高光谱图像的快速训练、分类和超参数自适应寻优,提出基于深层残差3D卷积神经网络(3D-CNN)的高光谱图像识别分类算法.由于采用的3D特征提取算法更适合高光谱3D数据结构,使得网络可以快速地从完整的高光谱图像样本中同时提取丰富的空间和光谱特征;此外,通过对高光谱图像样本平面空间方向的旋转和翻转操作进行数据增强的方法;以及运用TPE超参数优化算法对设定的超参数选择空间自适应寻优的方法,都可以有效地提高分类准确率.通过在TensorFlow框架下对Pavia University, Indian Pines和KSC等标准高光谱数据集上的实验结果表明,与SSRN等其他算法相比,文中算法在加深网络结构的同时,提高了分类准确率;与人工设定超参数网络相比,以TPE自适应超参数优化算法优化的网络参数数量减少约一半,训练时间缩短约10%.

     

    Abstract: In this paper, a deep residual 3 D convolutional neural network(3 D-CNN) framework is proposed for hyperspectral images classification in order to realize fast training, classification and parameter optimization. Rich spectral and spatial features can be rapidly extracted from samples of complete hyperspectral images using the Network, because the three-dimensional feature extraction algorithm is more suitable for three-dimensional data structure. In addition, the classification accuracy can be effectively improved by data augmentation which rotates and flips hyperspectral image samples from the spatial direction, and TPE algorithm which adjusts hyper-parameters in an artificial search space. Experimental results on the standard hyperspectral data sets, such as Pavia University, Indian Pines and KSC, show that the proposed framework not only deepens the network but also improves the classification accuracy compared with SSRN and other methods. The network optimized by TPE adaptive hyper-parameter optimization algorithm reduces the number of parameters by half and the training time by about 10%, compared with manual settings.

     

/

返回文章
返回