Laser & Optoelectronics Progress, Volume. 61, Issue 12, 1237001(2024)

Dimensionality Reduction Algorithm for Hyperspectral Image Based on Self-Supervised Learning

Zheng Zhou1,3, Yu Yang2、*, Gan Zhang4, Libing Xu3, Mingqing Wang1,3, and Qibing Zhu2
Author Affiliations
  • 1Wuxi Nine Cosmos Technology Co., Ltd., Wuxi 214072, Jiangsu, China
  • 2Key Laboratory of Advanced Process Control for Light Industry, Ministry of Education, School of Internet of Things Engineering, Jiangnan University, Wuxi 214122, Jiangsu, China
  • 3Department of Earth System Science, Tsinghua University, Beijing 100084, China
  • 4School of Electronic and Electrical Engineering, Zhengzhou University of Science and Technology, Zhengzhou 450052, Henan, China
  • show less

    To address the low hyperspectral image (HSI) pixel classification accuracy caused by few number of labeled training samples and high-dimensional spectral data, this study proposes a self-supervised learning-based feature extraction method to extract low-dimensional features representing crucial information of HSI data. First, an unsupervised data augmentation was used to expand the HSI training dataset. Then, the feature extraction module constructed by an external attention module was trained using an extended training dataset under a self-supervised framework. The self-affinity features between bands of a single sample and the potential correlation between different samples were extracted under the supervision of the intrinsic co-occurrence attributes of the spectral data. Finally, the trained feature extraction module was applied to reduce the dimension of raw HSI data, and the low-dimensional features were input to the subsequent classifier to classify the HSI pixels. The feasibility and effectiveness of the proposed method were evaluated through quantitative evaluation of dimensionality reduction results on Indian Pines, Salinas, and Pavia University datasets. The experimental results show that the feature extraction module generated using the proposed method can fully extract the spatial-spectral features from the original spectra. The proposed method is insensitive to the size of the training set and is suitable for small-sized HSI data.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Zheng Zhou, Yu Yang, Gan Zhang, Libing Xu, Mingqing Wang, Qibing Zhu. Dimensionality Reduction Algorithm for Hyperspectral Image Based on Self-Supervised Learning[J]. Laser & Optoelectronics Progress, 2024, 61(12): 1237001

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Digital Image Processing

    Received: Jul. 3, 2023

    Accepted: Aug. 22, 2023

    Published Online: Jun. 5, 2024

    The Author Email: Yu Yang (yangxiangyu1168@163.com)

    DOI:10.3788/LOP231646

    CSTR:32186.14.LOP231646

    Topics