多尺度特征衔接的高光谱分类网络
A classification network of hyperspectral images with multi-scale feature fusion
-
摘要: 针对高光谱图像分类过程中难以有效提取多尺度特征和姿态信息容易丢失的问题,该文提出了一种多尺度特征衔接高光谱分类网络(hierarchical multi-scale concatenation net,HMC-Net)。首先,HMC-Net利用多尺度卷积核并行计算以提取多层次特征,同时引入1×1卷积核降低输入输出维度,平衡计算复杂度,从而在不显著增加总体计算负担的前提下,实现高效的特征提取;接着,采用独立的胶囊网络并行处理各尺度特征,即通过动态路由改进最大池化,增强特征的平移不变性以减少姿态信息丢失;最后,通过concatenate操作衔接整合不同尺度的特征图,从而实现对高光谱图像分类过程中多层次信息的精确解析。对比实验结果表明: HMC-Net在肯尼迪航天中心数据集、帕维亚大学数据集和萨利纳斯数据集上整体精度分别达到了94%,98%和99%,与最新的高光谱分类模型相比有明显性能优势,验证了该文所提模型的有效性。Abstract: The classification of hyperspectral images faces challenges like ineffective extraction of multi-scale features and easy loss of pose information. Considering these challenges, this study proposed a classification network of hyperspectral images with multi-scale feature fusion-the hierarchical multi-scale concatenation net (HMC-Net). Initially, multi-scale convolution kernels were applied for parallel computing to extract multi-level features. Meanwhile, the 1×1 convolutional kernels were employed to reduce input-output dimensions, balancing computational complexity. These operations enabled efficient feature extraction without significantly increasing the overall computational burden. Subsequently, independent capsule networks were used for parallel processing of features at various scales. The max pooling was improved via dynamic routing to enhance the translation invariance of features, thereby reducing the loss of pose information. Finally, the concatenate operation integrated feature maps of different scales, thereby achieving a precise analysis of multi-level information in the classification of hyperspectral images. Comparative experimental results demonstrate that the HMC-Net achieved an overall accuracy of 94%, 98%, and 99% on the Kennedy Space Center, University of Pavia, and Salinas datasets, respectively. Compared to the latest classification model of hyperspectral images, the HMC-Net exhibited significant performance advantages, validating its effectiveness.
下载: