Classification of Hyperspectral Image Based on Double-Branch Dual-Attention Mechanism Network

被引:379
作者
Li, Rui [1 ]
Zheng, Shunyi [1 ]
Duan, Chenxi [2 ]
Yang, Yang [1 ]
Wang, Xiqi [1 ]
机构
[1] Wuhan Univ, Sch Remote Sensing & Informat Engn, Wuhan 430079, Peoples R China
[2] Wuhan Univ, State Key Lab Informat Engn Surveying Mapping & R, Wuhan 430079, Peoples R China
关键词
hyperspectral image classification; deep learning; channel-wise attention mechanism; spatial-wise attention mechanism; NEURAL-NETWORKS; INFORMATION;
D O I
10.3390/rs12030582
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
In recent years, researchers have paid increasing attention on hyperspectral image (HSI) classification using deep learning methods. To improve the accuracy and reduce the training samples, we propose a double-branch dual-attention mechanism network (DBDA) for HSI classification in this paper. Two branches are designed in DBDA to capture plenty of spectral and spatial features contained in HSI. Furthermore, a channel attention block and a spatial attention block are applied to these two branches respectively, which enables DBDA to refine and optimize the extracted feature maps. A series of experiments on four hyperspectral datasets show that the proposed framework has superior performance to the state-of-the-art algorithm, especially when the training samples are signally lacking.
引用
收藏
页数:25
相关论文
共 53 条
[1]  
[Anonymous], 2018, IEEE T NEUR NET LEAR
[2]  
[Anonymous], 2015, P INT C MACH LEARN
[3]  
[Anonymous], ADV NEURAL INFORM PR
[4]  
[Anonymous], 2018, IEEE T GEOSCI REMOTE, DOI DOI 10.1109/TGRS.2018.2838665
[5]  
[Anonymous], 2014, Neural Information Processing Systems
[6]   Improved Capability in Stone Pine Forest Mapping and Management in Lebanon Using Hyperspectral CHRIS-Proba Data Relative to Landsat ETM [J].
Awad, Mohamad ;
Jomaa, Ihab ;
Arab, Fatima .
PHOTOGRAMMETRIC ENGINEERING AND REMOTE SENSING, 2014, 80 (08) :725-731
[7]   Composite kernels for hyperspectral image classification [J].
Camps-Valls, G ;
Gomez-Chova, L ;
Muñoz-Marí, J ;
Vila-Francés, J ;
Calpe-Maravilla, J .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2006, 3 (01) :93-97
[8]   Deep Feature Extraction and Classification of Hyperspectral Images Based on Convolutional Neural Networks [J].
Chen, Yushi ;
Jiang, Hanlu ;
Li, Chunyang ;
Jia, Xiuping ;
Ghamisi, Pedram .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2016, 54 (10) :6232-6251
[9]   Spectral-Spatial Classification of Hyperspectral Data Based on Deep Belief Network [J].
Chen, Yushi ;
Zhao, Xing ;
Jia, Xiuping .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2015, 8 (06) :2381-2392
[10]   Deep Learning-Based Classification of Hyperspectral Data [J].
Chen, Yushi ;
Lin, Zhouhan ;
Zhao, Xing ;
Wang, Gang ;
Gu, Yanfeng .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2014, 7 (06) :2094-2107