TCCU-Net: Transformer and CNN Collaborative Unmixing Network for Hyperspectral Image

被引:15
作者
Chen, Jianfeng [1 ]
Yang, Chen [1 ,2 ]
Zhang, Lan [1 ]
Yang, Linzi [1 ]
Bian, Lifeng [3 ]
Luo, Zijiang [4 ]
Wang, Jihong [1 ,2 ]
机构
[1] Guizhou Univ, Coll Big Data & Informat Engn, Power Syst Engn Res Ctr, Minist Educ, Guiyang 550025, Peoples R China
[2] Guizhou Univ, China State Key Lab Publ Big Data, Guiyang 550025, Peoples R China
[3] Fudan Univ, Frontier Inst Chip & Syst, Shanghai 200433, Peoples R China
[4] Shunde Polytech, Inst Intelligent Mfg, Shunde 528300, Guangdong, Peoples R China
基金
中国国家自然科学基金;
关键词
Transformers; Hyperspectral imaging; Task analysis; Convolutional neural networks; Three-dimensional displays; Head; Transformer cores; CNN; global and local information; hyperspectral image unmixing (HSU); spectral and spatial information; transformer;
D O I
10.1109/JSTARS.2024.3352073
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
In recent years, deep-learning-based hyperspectral unmixing techniques have garnered increasing attention and made significant advancements. However, relying solely on the use of convolutional neural network (CNN) or transformer approaches is insufficient for effectively capturing both global and fine-grained information, thereby compromising the accuracy of unmixing tasks. In order to fully harness the information contained within hyperspectral images, this article explores a dual-stream collaborative network, referred to as TCCU-Net. It end-to-end learns information in four dimensions: spectral, spatial, global, and local, to achieve more effective unmixing. The network comprises two core encoders: one is a transformer encoder, which includes squeeze-launch modules, DSSCR-vision transformer modules, and stripe pooling modules, while the other one is a CNN encoder, which is composed of two-dimensional (2-D) pyramid convolutions and 3-D pyramid convolutions. By fusing the outputs of these two encoders, the semantic gap between the encoder and decoder is bridged, resulting in improved feature mapping and unmixing outcomes. This article extensively evaluates TCCU-Net and seven hyperspectral unmixing methods on four datasets (Samson, Apex, Jasper Ridge, and Synthetic dataset). The experimental results firmly demonstrate that the proposed approach surpasses others in terms of accuracy, holding the potential to effectively address hyperspectral unmixing tasks.
引用
收藏
页码:8073 / 8089
页数:17
相关论文
共 40 条
[1]   Hyperspectral Remote Sensing Data Analysis and Future Challenges [J].
Bioucas-Dias, Jose M. ;
Plaza, Antonio ;
Camps-Valls, Gustavo ;
Scheunders, Paul ;
Nasrabadi, Nasser M. ;
Chanussot, Jocelyn .
IEEE GEOSCIENCE AND REMOTE SENSING MAGAZINE, 2013, 1 (02) :6-36
[2]  
Chen D., 2023, IEEE GEOSCI REMOTE S, V20
[3]   Temporal difference-guided network for hyperspectral image change detection [J].
Chen, Zhonghao ;
Wang, Yuyang ;
Gao, Hongmin ;
Ding, Yao ;
Zhong, Qiqiang ;
Hong, Danfeng ;
Zhang, Bing .
INTERNATIONAL JOURNAL OF REMOTE SENSING, 2023, 44 (19) :6033-6059
[4]   Local aggregation and global attention network for hyperspectral image classification with spectral-induced aligned superpixel segmentation [J].
Chen, Zhonghao ;
Wu, Guoyong ;
Gao, Hongmin ;
Ding, Yao ;
Hong, Danfeng ;
Zhang, Bing .
EXPERT SYSTEMS WITH APPLICATIONS, 2023, 232
[5]   Global to Local: A Hierarchical Detection Algorithm for Hyperspectral Image Target Detection [J].
Chen, Zhonghao ;
Lu, Zhengtao ;
Gao, Hongmin ;
Zhang, Yiyan ;
Zhao, Jia ;
Hong, Danfeng ;
Zhang, Bing .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[6]   Hyperspectral and LiDAR Data Fusion: Outcome of the 2013 GRSS Data Fusion Contest [J].
Debes, Christian ;
Merentitis, Andreas ;
Heremans, Roel ;
Hahn, Juergen ;
Frangiadakis, Nikolaos ;
van Kasteren, Tim ;
Liao, Wenzhi ;
Bellens, Rik ;
Pizurica, Aleksandra ;
Gautama, Sidharta ;
Philips, Wilfried ;
Prasad, Saurabh ;
Du, Qian ;
Pacifici, Fabio .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2014, 7 (06) :2405-2418
[7]  
Devlin J, 2019, 2019 CONFERENCE OF THE NORTH AMERICAN CHAPTER OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS: HUMAN LANGUAGE TECHNOLOGIES (NAACL HLT 2019), VOL. 1, P4171
[8]   Joint Bayesian Endmember Extraction and Linear Unmixing for Hyperspectral Imagery [J].
Dobigeon, Nicolas ;
Moussaoui, Said ;
Coulon, Martial ;
Tourneret, Jean-Yves ;
Hero, Alfred O. .
IEEE TRANSACTIONS ON SIGNAL PROCESSING, 2009, 57 (11) :4355-4368
[9]  
Dosovitskiy A, 2021, Arxiv, DOI arXiv:2010.11929
[10]   Hyperspectral Image Unmixing With Endmember Bundles and Group Sparsity Inducing Mixed Norms [J].
Drumetz, Lucas ;
Meyer, Travis R. ;
Chanussot, Jocelyn ;
Bertozzi, Andrea L. ;
Jutten, Christian .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2019, 28 (07) :3435-3450