An Unsupervised Band Selection Method via Contrastive Learning for Hyperspectral Images

被引:1
作者
Li, Xiaorun [1 ]
Liu, Yufei [1 ]
Hua, Ziqiang [1 ]
Chen, Shuhan [1 ]
机构
[1] Zhejiang Univ, Coll Elect Engn, Hangzhou 310027, Peoples R China
关键词
unsupervised band selection; contrastive learning; hyperspectral image; attention mechanism;
D O I
10.3390/rs15235495
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
Band selection (BS) is an efficacious approach to reduce hyperspectral information redundancy while preserving the physical meaning of hyperspectral images (HSIs). Recently, deep learning-based BS methods have received widespread interest due to their ability to model the nonlinear relationship between bands, with existing methods typically relying on generative algorithms. However, the process of generating images with pixel-level detail required by generative algorithm-based BS methods is computationally expensive. To alleviate this issue, we propose a contrastive learning-based unsupervised BS architecture, termed ContrastBS, in this article. With the help of contrastive learning, the proposed architecture avoids the costly generation step in pixel space by learning to distinguish data at the abstract semantic level of the feature space. Specifically, ContrastBS combines an attention mechanism with contrastive learning to extract the importance of each band. Furthermore, we design a novel loss function, which is able to constrain the symmetric loss while ensuring attention to the most valuable bands, for the contrastive learning-based BS network. Experimental results indicate that ContrastBS has excellent classification performance and competitive time cost compared to the comparison methods.
引用
收藏
页数:17
相关论文
共 50 条
[11]   Unsupervised Image Enhancement via Contrastive Learning [J].
Li, Di ;
Rahardja, Susanto .
2024 IEEE INTERNATIONAL SYMPOSIUM ON CIRCUITS AND SYSTEMS, ISCAS 2024, 2024,
[12]   A New Band Selection Method for Hyperspectral Images based on Constrained Optimization [J].
Gharaati, Elahe ;
Nasri, Mehdi .
2015 7TH CONFERENCE ON INFORMATION AND KNOWLEDGE TECHNOLOGY (IKT), 2015,
[13]   AN UNSUPERVISED BAND SELECTION METHOD BASED ON TRIPLE CONSTRAINTS AND ATTENTION NETWORK FOR HYPERSPECTRAL IMAGERY [J].
Liu, Yufei ;
Li, Xiaorun .
IGARSS 2023 - 2023 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM, 2023, :5395-5398
[14]   Clustering based Band Selection for Hyperspectral Images [J].
Datta, Aloke ;
Ghosh, Susmita ;
Ghosh, Ashish .
PROCEEDINGS OF THE 2012 INTERNATIONAL CONFERENCE ON COMMUNICATIONS, DEVICES AND INTELLIGENT SYSTEMS (CODLS), 2012, :101-104
[15]   A Diversified Multi-objective Particle Swarm Optimization Algorithm for Unsupervised Band Selection of Hyperspectral Images [J].
Zhang, Yuze ;
Li, Lingjie ;
Xiao, Zhijiao ;
Lin, Qiuzhen ;
Ming, Zhong .
ADVANCES IN SWARM INTELLIGENCE, ICSI 2023, PT I, 2023, 13968 :464-475
[16]   Spatial-Spectral Hypergraph-Based Unsupervised Band Selection for Hyperspectral Remote Sensing Images [J].
Ma, Zhenyu ;
Yang, Bin .
IEEE SENSORS JOURNAL, 2024, 24 (17) :27870-27882
[17]   Unsupervised sound localization via iterative contrastive learning [J].
Lin, Yan-Bo ;
Tseng, Hung-Yu ;
Lee, Hsin-Ying ;
Lin, Yen-Yu ;
Yang, Ming-Hsuan .
COMPUTER VISION AND IMAGE UNDERSTANDING, 2023, 227
[18]   ANT COLONY OPTIMIZATION FOR SUPERVISED AND UNSUPERVISED HYPERSPECTRAL BAND SELECTION [J].
Gao, Jianwei ;
Du, Qian ;
Gao, Lianru ;
Sun, Xu ;
Wu, Yuanfeng ;
Zhang, Bing .
2013 5TH WORKSHOP ON HYPERSPECTRAL IMAGE AND SIGNAL PROCESSING: EVOLUTION IN REMOTE SENSING (WHISPERS), 2013,
[19]   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
[20]   Greedy Unsupervised Hyperspectral Image Band Selection Method Based on Variable Precision Rough Set [J].
Chen Jing ;
Zhang Zhenxing .
ACTA PHOTONICA SINICA, 2021, 50 (02)