Self-Correlation Network With Triple Contrastive Learning for Hyperspectral Image Classification With Noisy Labels

被引:1
作者
Sarpong, Kwabena [1 ]
Awrangjeb, Mohammad [1 ]
Islam, Md. Saiful [2 ]
Helmy, Islam [1 ]
机构
[1] Griffith Univ, Sch Informat & Commun Technol, Nathan, Qld 4111, Australia
[2] Univ Newcastle, Coll Engn Sci & Environm, Sch Informat & Phys Sci, Callaghan, NSW 2308, Australia
关键词
Noise measurement; Noise; Feature extraction; Accuracy; Training; Contrastive learning; Classification algorithms; Transformers; Hyperspectral imaging; Deep learning; Contrastive learning (CL); convolutional neural network (CNN); hyperspectral image classification (HIC); self-correlation network; RESIDUAL NETWORK;
D O I
10.1109/JSTARS.2025.3543764
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Data quality is essential for training deep learning models, and recently, the challenge of noisy labels in hyperspectral image (HSI) classification has attracted considerable attention. However, current deep learning approaches typically employ conventional convolution methods that treat all spatial frequency components uniformly, neglecting the exploration of feature-dependent knowledge, significantly affecting learning with noisy labels. Consequently, these methods perform poorly in scenarios with a high noisy-to-clean sample ratio. To address the above drawback, we propose an end-to-end self-correlation framework with triple contrastive learning (SCTCL) for HSI classification with noisy labels. Our SCTCL harnesses maximizing the similarities of the positive pairs of the HSI features by defining cluster-, instance-, and structure-level learnings representing a contrastive loss. First, we construct HSI data pairs through weak and strong data augmentations. Then, we propose a cross-convolutional with a self-correlation network (ConvSCNet) module to extract spatial-spectral feature representation from all augmented samples. Subsequently, we employ instance- and cluster-level contrastive learnings to project the feature matrix in row and column spaces to minimize negative and maximize positive pairs. Furthermore, we incorporate structure-level representation learning to address inconsistencies across different projections. By doing so, we mitigate the classifier from overfitting to noisy labels. We conducted experiments on five publicly available HSI datasets with various noisy-to-clean sample ratios. We consider both symmetric and asymmetric noises. The classification results prove that the proposed SCTCL performs excellently in training HSI with a limited clean sample compared to the state-of-the-art methods.
引用
收藏
页码:7166 / 7188
页数:23
相关论文
共 77 条
[1]   Image classification with deep learning in the presence of noisy labels: A survey [J].
Algan, Gorkem ;
Ulusoy, Ilkay .
KNOWLEDGE-BASED SYSTEMS, 2021, 215
[2]   Vision Transformer With Contrastive Learning for Remote Sensing Image Scene Classification [J].
Bi, Meiqiao ;
Wang, Minghua ;
Li, Zhi ;
Hong, Danfeng .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2023, 16 :738-749
[3]  
Chen T., 2020, PMLR, P1597
[4]   Drop an Octave: Reducing Spatial Redundancy in Convolutional Neural Networks with Octave Convolution [J].
Chen, Yunpeng ;
Fan, Haoqi ;
Xu, Bing ;
Yan, Zhicheng ;
Kalantidis, Yannis ;
Rohrbach, Marcus ;
Yan, Shuicheng ;
Feng, Jiashi .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :3434-3443
[5]   Deep Feature Extraction and Classification of Hyperspectral Images Based on Convolutional Neural Networks [J].
Chen, Yushi ;
Jiang, Hanlu ;
Li, Chunyang ;
Jia, Xiuping ;
Ghamisi, Pedram .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2016, 54 (10) :6232-6251
[6]  
Cheng J., 2017, P INT C MACH LEARN, P1789
[7]   Learning Dual Geometric Low-Rank Structure for Semisupervised Hyperspectral Image Classification [J].
Feng, Zhixi ;
Yang, Shuyuan ;
Wang, Min ;
Jiao, Lichen .
IEEE TRANSACTIONS ON CYBERNETICS, 2021, 51 (01) :346-358
[8]   Advances in Hyperspectral Image and Signal Processing A comprehensive overview of the state of the art [J].
Ghamisi, Pedram ;
Yokoya, Naoto ;
Li, Jun ;
Liao, Wenzhi ;
Liu, Sicong ;
Plaza, Javier ;
Rasti, Behnood ;
Plaza, Antonio .
IEEE GEOSCIENCE AND REMOTE SENSING MAGAZINE, 2017, 5 (04) :37-78
[9]  
Han B., 2020, arXiv
[10]   Co-teaching: Robust Training of Deep Neural Networks with Extremely Noisy Labels [J].
Han, Bo ;
Yao, Quanming ;
Yu, Xingrui ;
Niu, Gang ;
Xu, Miao ;
Hu, Weihua ;
Tsang, Ivor W. ;
Sugiyama, Masashi .
ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 31 (NIPS 2018), 2018, 31