Negative Samples Mining Matters: Reconsidering Hyperspectral Image Classification With Contrastive Learning

被引:0
作者
Liu, Hui [1 ,2 ]
Huang, Chenjia [1 ]
Chen, Ning [3 ]
Xie, Tao [1 ]
Lu, Mingyue [4 ]
Huang, Zhou [3 ]
机构
[1] Nanjing Univ Informat Sci & Technol, Sch Remote Sensing & Geomat Engn, Nanjing 210044, Peoples R China
[2] Deqing Acad Satellite Applicat, Lab Target Microwave Properties, Huzhou 313000, Peoples R China
[3] Peking Univ, Inst Remote Sensing & Geog Informat Syst, Beijing 100871, Peoples R China
[4] Nanjing Univ Informat Sci & Technol, Sch Geog Sci, Nanjing 210044, Peoples R China
来源
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING | 2024年 / 62卷
基金
中国国家自然科学基金;
关键词
Contrastive learning; Hyperspectral imaging; Feature extraction; Training; Accuracy; Transformers; Optimization; Information science; Data mining; Semisupervised learning; deep neural network; hyperspectral image (HSI) classification; negative samples mining; NETWORKS; AUTOENCODER;
D O I
10.1109/TGRS.2024.3491074
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
In recent years, there have been significant advancements in hyperspectral image (HSI) classification methods using contrastive learning. However, these methods often fail to effectively screen and mine negative samples during the construction of contrastive learning pairs. This oversight introduces negative samples that belong to the same class as the positive samples, thereby limiting the effectiveness of contrastive learning. To address this issue, we propose a novel HSI classification framework based on contrastive learning that flexibly supports the selection and mining of negative samples. Specifically, before training the contrastive learning task, we use pseudolabel information to guide the mining of negative samples, eliminating those with pseudolabels of the same class as the anchor samples. This approach strengthens the alignment between the optimization directions of the contrastive learning task and the classification task. In addition, we improve the contrastive learning process by introducing a controlled mixture of hard and easy negative samples, which enhances the accuracy of HSI classification. The pivotal characteristic of this study lies in enhancing the effectiveness of HSI classification based on contrastive learning through effective filtering and mining of negative samples. Experimental comparisons across multiple public datasets demonstrate the superiority of our proposed method over state-of-the-art algorithms.
引用
收藏
页数:16
相关论文
共 66 条
  • [1] ContrastNet: Unsupervised feature learning by autoencoder and prototypical contrastive learning for hyperspectral imagery classification
    Cao, Zeyu
    Li, Xiaorun
    Feng, Yueming
    Chen, Shuhan
    Xia, Chaoqun
    Zhao, Liaoying
    [J]. NEUROCOMPUTING, 2021, 460 : 71 - 83
  • [2] Caron M, 2020, ADV NEUR IN, V33
  • [3] Emerging Properties in Self-Supervised Vision Transformers
    Caron, Mathilde
    Touvron, Hugo
    Misra, Ishan
    Jegou, Herve
    Mairal, Julien
    Bojanowski, Piotr
    Joulin, Armand
    [J]. 2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, : 9630 - 9640
  • [4] SpectralDiff: A Generative Framework for Hyperspectral Image Classification With Diffusion Models
    Chen, Ning
    Yue, Jun
    Fang, Leyuan
    Xia, Shaobo
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [5] Spectral Query Spatial: Revisiting the Role of Center Pixel in Transformer for Hyperspectral Image Classification
    Chen, Ning
    Fang, Leyuan
    Xia, Yang
    Xia, Shaobo
    Liu, Hui
    Yue, Jun
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62 : 1 - 14
  • [6] Inducing metallicity in graphene nanoribbons via zero-mode superlattices
    Rizzo, Daniel J.
    Veber, Gregory
    Jiang, Jingwei
    McCurdy, Ryan
    Cao, Ting
    Bronner, Christopher
    Chen, Ting
    Louie, Steven G.
    Fischer, Felix R.
    Crommie, Michael F.
    [J]. SCIENCE, 2020, 369 (6511) : 1597 - +
  • [7] Chen Ting, 2020, Advances in neural information processing systems
  • [8] Chen XL, 2020, Arxiv, DOI arXiv:2003.04297
  • [9] Exploring Simple Siamese Representation Learning
    Chen, Xinlei
    He, Kaiming
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 15745 - 15753
  • [10] An Empirical Study of Training Self-Supervised Vision Transformers
    Chen, Xinlei
    Xie, Saining
    He, Kaiming
    [J]. 2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, : 9620 - 9629