A novel graph-attention based multimodal fusion network for joint classification of hyperspectral image and LiDAR data

被引:31
作者
Cai, Jianghui [1 ,2 ]
Zhang, Min [1 ]
Yang, Haifeng [1 ,3 ]
He, Yanting [1 ]
Yang, Yuqing [1 ]
Shi, Chenhui [1 ]
Zhao, Xujun [1 ,3 ]
Xun, Yaling [1 ]
机构
[1] Taiyuan Univ Sci & Technol TYUST, Sch Comp Sci & Technol, Taiyuan 030024, Shanxi, Peoples R China
[2] North Univ China NUC, Sch Data Sci & Technol, Taiyuan 030051, Shanxi, Peoples R China
[3] Shanxi Key Lab Big Data Anal & Parallel Comp, Taiyuan 030024, Shanxi, Peoples R China
关键词
Multi-source data classification; Graph-attention based fusion; Hyperspectral image; Parameter sharing;
D O I
10.1016/j.eswa.2024.123587
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The joint classification of hyperspectral image (HSI) and Light Detection and Ranging (LiDAR) data can provide complementary information for each other, which has become a prominent topic in the field of remote sensing. Nevertheless, the common CNN -based fusion techniques still suffer from the following drawbacks. (1) Most of these models omit the correlation and complementarity between different data sources and always fail to model the long-distance dependencies of spectral information well. (2) Simply splicing the multi -source feature embeddings overlooks the deep semantic relationships among them. To tackle these issues, we propose a novel graph -attention based multimodal fusion network (GAMF). Specifically, it employs three major components, including an HSI-LiDAR feature extractor, a graph -attention based fusion module and a classification module. In the feature extraction module, we consider the correlation and complementarity between multi -sensor data by parameter sharing and employ Gaussian tokenization for feature transformation additionally. To address the problem of long-distance dependencies, the deep fusion module utilizes modality -specific tokens to construct an undirected weighted graph, which is essentially a heterogeneous graph. And the deep semantic relationships between them are exploited utilizing a graph -attention based fusion framework. At the end, two fully connected layers classify the fused embeddings. Experiment evaluations on several benchmark HSI-LiDAR datasets (Trento, University of Houston 2013 and MUUFL) show that GAMF achieves more accurate prediction results than some state-of-the-art baselines. The code is available at https://github.com/tyust-dayu/GAMF.
引用
收藏
页数:16
相关论文
共 50 条
[1]   Hyperspectral Image Classification-Traditional to Deep Models: A Survey for Future Prospects [J].
Ahmad, Muhammad ;
Shabbir, Sidrah ;
Roy, Swalpa Kumar ;
Hong, Danfeng ;
Wu, Xin ;
Yao, Jing ;
Khan, Adil Mehmood ;
Mazzara, Manuel ;
Distefano, Salvatore ;
Chanussot, Jocelyn .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2022, 15 :968-999
[2]   A Fast and Compact 3-D CNN for Hyperspectral Image Classification [J].
Ahmad, Muhammad ;
Khan, Adil Mehmood ;
Mazzara, Manuel ;
Distefano, Salvatore ;
Ali, Mohsin ;
Sarfraz, Muhammad Shahzad .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
[3]   Multimodal Earth observation data fusion: Graph-based approach in shared latent space [J].
Arun, P., V ;
Sadeh, R. ;
Avneri, A. ;
Tubul, Y. ;
Camino, C. ;
Buddhiraju, K. M. ;
Porwal, A. ;
Lati, R. N. ;
Zarco-Tejada, P. J. ;
Peleg, Z. ;
Herrmann, I .
INFORMATION FUSION, 2022, 78 :20-39
[4]   ARIS: A Noise Insensitive Data Pre-Processing Scheme for Data Reduction Using Influence Space [J].
Cai, Jianghui ;
Yang, Yuqing ;
Yang, Haifeng ;
Zhao, Xujun ;
Hao, Jing .
ACM TRANSACTIONS ON KNOWLEDGE DISCOVERY FROM DATA, 2022, 16 (06)
[5]   Hypergraph-Structured Autoencoder for Unsupervised and Semisupervised Classification of Hyperspectral Image [J].
Cai, Yaoming ;
Zhang, Zijia ;
Cai, Zhihua ;
Liu, Xiaobo ;
Jiang, Xinwei .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
[6]   BS-Nets: An End-to-End Framework for Band Selection of Hyperspectral Image [J].
Cai, Yaoming ;
Liu, Xiaobo ;
Cai, Zhihua .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2020, 58 (03) :1969-1984
[7]  
Chakraborty T, 2021, Arxiv, DOI [arXiv:2104.00341, DOI 10.48550/ARXIV.2104.00341]
[8]   Automatic Design of Convolutional Neural Network for Hyperspectral Image Classification [J].
Chen, Yushi ;
Zhu, Kaiqiang ;
Zhu, Lin ;
He, Xin ;
Ghamisi, Pedram ;
Benediktsson, Jon Atli .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2019, 57 (09) :7048-7066
[9]   Semi-Supervised Locality Preserving Dense Graph Neural Network With ARMA Filters and Context-Aware Learning for Hyperspectral Image Classification [J].
Ding, Yao ;
Zhao, Xiaofeng ;
Zhang, Zhili ;
Cai, Wei ;
Yang, Nengjun ;
Zhan, Ying .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[10]   Weighted Feature Fusion of Convolutional Neural Network and Graph Attention Network for Hyperspectral Image Classification [J].
Dong, Yanni ;
Liu, Quanwei ;
Du, Bo ;
Zhang, Liangpei .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2022, 31 :1559-1572