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 [J].
Cao, Zeyu ;
Li, Xiaorun ;
Feng, Yueming ;
Chen, Shuhan ;
Xia, Chaoqun ;
Zhao, Liaoying .
NEUROCOMPUTING, 2021, 460 :71-83
[2]  
Caron M, 2020, ADV NEUR IN, V33
[3]   Emerging Properties in Self-Supervised Vision Transformers [J].
Caron, Mathilde ;
Touvron, Hugo ;
Misra, Ishan ;
Jegou, Herve ;
Mairal, Julien ;
Bojanowski, Piotr ;
Joulin, Armand .
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 [J].
Chen, Ning ;
Yue, Jun ;
Fang, Leyuan ;
Xia, Shaobo .
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 [J].
Chen, Ning ;
Fang, Leyuan ;
Xia, Yang ;
Xia, Shaobo ;
Liu, Hui ;
Yue, Jun .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62 :1-14
[6]  
Chen T., 2020, NeurIPS 2020, P22243, DOI 10.48550/arXiv.2006.10029
[7]  
Chen T., 2020, INT C MACHINE LEARNI, P1597
[8]  
Chen XL, 2020, Arxiv, DOI arXiv:2003.04297
[9]   Exploring Simple Siamese Representation Learning [J].
Chen, Xinlei ;
He, Kaiming .
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 [J].
Chen, Xinlei ;
Xie, Saining ;
He, Kaiming .
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, :9620-9629