计算机科学
人工智能
自编码
模式识别(心理学)
高光谱成像
深度学习
特征提取
主成分分析
构造(python库)
特征学习
样品(材料)
监督学习
人工神经网络
机器学习
化学
色谱法
程序设计语言
作者
Minghao Zhu,H Wang,Yuebo Meng,Zhe Shan,Zongfang Ma
标识
DOI:10.1007/978-981-99-8462-6_39
摘要
Deep learning methods have made significant progress in the field of hyperspectral image (HSI) classification. However, these methods often rely on a large number of labeled samples, parameters, and computational resources to achieve state-of-the-art performance, which limits their applicability. To address these issues, this paper proposes a lightweight multiview mask contrastive network (LMCN) for HSI classification under small-sample conditions. Considering the influence of irrelevant bands, we construct two views in an HSI scene using band selection and principal component analysis (PCA). To enhance instance discriminability, we propose a combination of self-supervised mask learning and contrastive learning in the design of LMCN. Specifically, we train corresponding masked autoencoders using the obtained views and utilize the feature extraction part of the autoencoder as an augmentation function, conducting unsupervised training through contrastive learning. To reduce the number of parameters, we employ lightweight Transformer modules to construct the autoencoder. Experimental results demonstrate the superiority of this approach over several advanced supervised learning methods and few-shot learning methods under small-sample conditions. Furthermore, this method exhibits lower computational costs. Our code is available at https://github.com/Winkness/LMCN.git .
科研通智能强力驱动
Strongly Powered by AbleSci AI