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 条
  • [41] Dual-Branch Fourier-Mixing Transformer Network for Hyperspectral Target Detection
    Jiao, Jinyue
    Gong, Zhiqiang
    Zhong, Ping
    REMOTE SENSING, 2023, 15 (19)
  • [42] DeepfakeUCL: Deepfake Detection via Unsupervised Contrastive Learning
    Fung, Sheldon
    Lu, Xuequan
    Zhang, Chao
    Li, Chang-Tsun
    2021 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2021,
  • [43] Recursive Automatic Target Generation Process for Unsupervised Hyperspectral Target Detection
    Gao, Cheng
    Chang, Chein-I
    2014 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS), 2014, : 3598 - 3601
  • [44] Siamese comparative transformer-based network for unsupervised landmark detection
    Zhao, Can
    Wu, Tao
    Zhang, Jianlin
    Xu, Zhiyong
    Li, Meihui
    Liu, Dongxu
    PLOS ONE, 2024, 19 (12):
  • [45] Unsupervised discriminative projection based on contrastive learning
    Yang, Jingwen
    Zhang, Hongjie
    Zhou, Ruojin
    Hao, Zhuangzhuang
    Jing, Ling
    KNOWLEDGE-BASED SYSTEMS, 2024, 301
  • [46] A Multi-Stage Transformer Network for Image Dehazing Based on Contrastive Learning
    Gao F.
    Ji S.
    Guo J.
    Hou J.
    Ouyang C.
    Yang B.
    Hsi-An Chiao Tung Ta Hsueh/Journal of Xi'an Jiaotong University, 2023, 57 (01): : 195 - 210
  • [47] Hyperspectral Imagery Classification Based on Contrastive Learning
    Hou, Sikang
    Shi, Hongye
    Cao, Xianghai
    Zhang, Xiaohua
    Jiao, Licheng
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [48] Multiple-Instance Metric Learning Network for Hyperspectral Target Detection
    Yang, Bo
    He, Yi
    Jiao, Changzhe
    Pan, Xiao
    Wang, Guozhen
    Wang, Lei
    Wu, Jinjian
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [49] Transformer-Based Zero-Shot Detection via Contrastive Learning
    Liu, Wei
    Chen, Hui
    Ma, Yongqiang
    Wang, Jianji
    Zheng, Nanning
    ARTIFICIAL INTELLIGENCE APPLICATIONS AND INNOVATIONS, AIAI 2022, PART I, 2022, 646 : 316 - 327
  • [50] Distribution Network Anomaly Detection Based on Graph Contrastive Learning
    Feng, Mingjun
    Liu, Caiyun
    Sun, Yan
    Wu, Yidong
    Li, Bo
    JOURNAL OF SIGNAL PROCESSING SYSTEMS FOR SIGNAL IMAGE AND VIDEO TECHNOLOGY, 2024, 96 (10): : 541 - 554