An Unsupervised Momentum Contrastive Learning Based Transformer Network for Hyperspectral Target Detection

被引:11
|
作者
Wang, Yulei [1 ]
Chen, Xi [1 ]
Zhao, Enyu [1 ]
Zhao, Chunhui [2 ]
Song, Meiping [1 ]
Yu, Chunyan [1 ]
机构
[1] Dalian Maritime Univ, Informat Sci & Technol Coll, Dalian 116026, Peoples R China
[2] Harbin Engn Univ, Coll Informat & Commun Engn, Harbin 150009, Peoples R China
基金
中国国家自然科学基金;
关键词
Hyperspectral imagery (HSI); momentum contrastive learning; target detection; transformer; unsupervised learning; ORTHOGONAL SUBSPACE PROJECTION; SPECTRAL MATCHED-FILTER; REPRESENTATION;
D O I
10.1109/JSTARS.2024.3387985
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Hyperspectral target detection plays a pivotal role in various civil and military applications. Although recent advancements in deep learning have largely embraced supervised learning approaches, they often hindered by the limited availability of labeled data. Unsupervised learning, therefore, emerges as a promising alternative, yet its potential has not been fully realized in current methodologies. This article proposes an innovative unsupervised learning framework employing a momentum contrastive learning-based transformer network specifically tailored for hyperspectral target detection. The proposed approach innovatively combines transformer-based encoder and momentum encoder networks to enhance feature extraction capabilities, adeptly capturing both local spectral details and long-range spectral dependencies through the novel overlapping spectral patch embedding and a cross-token feedforward layer. This dual-encoder design significantly improves the model's ability to discern relevant spectral features amidst complex backgrounds. Through unsupervised momentum contrastive learning, a dynamically updated queue of negative sample features is utilized so that the model can demonstrate superior spectral discriminability. This is further bolstered by a unique background suppression mechanism leveraging nonlinear transformations of cosine similarity detection results, with two nonlinearly pull-up operations, significantly enhancing target detection sensitivity, where the nonlinearly operations are the exponential function with its normalization and the power function with its normalization, respectively. Comparative analysis against seven state-of-the-art hyperspectral target detection methods across four real hyperspectral images demonstrates the effectiveness of the proposed method for hyperspectral target detection, with an increase in detection accuracy and a competitive computational efficiency. An extensive ablation study further validates the critical components of the proposed framework, confirming its comprehensive capability and applicability in hyperspectral target detection scenarios.
引用
收藏
页码:9053 / 9068
页数:16
相关论文
共 50 条
  • [1] Unsupervised contrastive learning based transformer for lung nodule detection
    Niu, Chuang
    Wang, Ge
    PHYSICS IN MEDICINE AND BIOLOGY, 2022, 67 (20):
  • [2] CONTRASTIVE LEARNING FOR HYPERSPECTRAL TARGET DETECTION
    Chen, Xi
    Wang, Yulei
    Che, Zongwei
    Zhu, Liyu
    Song, Meiping
    Yu, Haoyang
    2022 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2022), 2022, : 887 - 890
  • [3] HTDFormer: Hyperspectral Target Detection Based on Transformer With Distributed Learning
    Li, Yunsong
    Qin, Haonan
    Xie, Weiying
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [4] HTDFormer: Hyperspectral Target Detection Based on Transformer With Distributed Learning
    Li, Yunsong
    Qin, Haonan
    Xie, Weiying
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [5] Contrastive Learning Based on Transformer for Hyperspectral Image Classification
    Hu, Xiang
    Li, Teng
    Zhou, Tong
    Liu, Yu
    Peng, Yuanxi
    APPLIED SCIENCES-BASEL, 2021, 11 (18):
  • [6] ConVaT: A Variational Generative Transformer With Momentum Contrastive Learning for Hyperspectral Image Classification
    Liang, Miaomiao
    Liu, Zuo
    Dong, Jian
    Yu, Lingjuan
    Yu, Xiangchun
    Li, Jun
    Jiao, Licheng
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2024, 21 : 1 - 5
  • [7] Siamese Transformer Network for Hyperspectral Image Target Detection
    Rao, Weiqiang
    Gao, Lianru
    Qu, Ying
    Sun, Xu
    Zhang, Bing
    Chanussot, Jocelyn
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [8] Triplet Spectralwise Transformer Network for Hyperspectral Target Detection
    Jiao, Jinyue
    Gong, Zhiqiang
    Zhong, Ping
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [9] Unsupervised Transformer Boundary Autoencoder Network for Hyperspectral Image Change Detection
    Liu, Song
    Li, Haiwei
    Wang, Feifei
    Chen, Junyu
    Zhang, Geng
    Song, Liyao
    Hu, Bingliang
    REMOTE SENSING, 2023, 15 (07)
  • [10] Unsupervised transfer learning for target detection from hyperspectral images
    Du, Bo
    Zhang, Liangpei
    Tao, Dacheng
    Zhang, Dengyi
    NEUROCOMPUTING, 2013, 120 : 72 - 82