Multi-Structure KELM With Attention Fusion Strategy for Hyperspectral Image Classification

被引:43
作者
Sun, Le [1 ,2 ]
Fang, Yu [1 ]
Chen, Yuwen [3 ]
Huang, Wei [4 ]
Wu, Zebin [5 ]
Jeon, Byeungwoo [6 ]
机构
[1] Nanjing Univ Informat Sci & Technol, Sch Comp Sci, Nanjing 210044, Peoples R China
[2] Nanjing Univ Informat Sci & Technol NUIST, Engn Res Ctr Digital Forens, Minist Educ, Nanjing 210044, Peoples R China
[3] Chinese Acad Sci, Chongqing Inst Green & Intelligent Technol, Chongqing 400714, Peoples R China
[4] Zhengzhou Univ Light Ind, Sch Comp & Commun Engn, Zhengzhou 450000, Peoples R China
[5] Nanjing Univ Sci & Technol, Sch Comp Engn, Nanjing 210094, Peoples R China
[6] Sungkyunkwan Univ, Sch Elect Engn, Suwon 440746, South Korea
来源
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING | 2022年 / 60卷
基金
中国国家自然科学基金;
关键词
Feature extraction; Training; Deep learning; Kernel; Hyperspectral imaging; Electronic mail; Extreme learning machines; Attention mechanisms; hyperspectral image (HSI) classification; kernel extreme learning machine (KELM); multifeature; multiscale (MS); RANDOM FOREST; FRAMEWORK; MACHINE;
D O I
10.1109/TGRS.2022.3208165
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
Hyperspectral image (HSI) classification refers to accurately corresponding each pixel in an HSI to a land-cover label. Recently, the successful application of multiscale and multifeature methods has greatly improved the performance of HSI classification due to their enhanced utilization of the available spectral-spatial information. However, as the number of scales and the number of features increases, it becomes more difficult to achieve an optimal degree of fusion for multiple classifiers [e.g., kernel extreme learning machine (KELM)]. On the other hand, a limited sample size of the HSI may cause overfitting problems, which seriously affects the classification accuracy. Therefore, in this article, a novel multi-structure KELM with attention fusion strategy (MSAF-KELM) is proposed to achieve accurate fusion of multiple classifiers for effective HSI classification with ultrasmall sample rates. First, a multi-structure network is built, which combines multiple scales and multiple features to extract abundant spectral-spatial information. Second, a fast and efficient KELM is employed to enable rapid classification. Finally, a weighted self-attention fusion strategy (WSAFS) is introduced, which combines the output weights of each KELM subbranch and the self-attention mechanism to achieve an efficient fusion result on multi-structure networks. We conducted experiments on four types of HSI datasets with different evaluation methods and compared them with several classical and state-of-the-art methods, which demonstrate the excellent performance of our method on ultrasmall sample rates. The code is available at https://github.com/Fang666666/MSAF-KELM for reproducibility.
引用
收藏
页数:17
相关论文
共 57 条
[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]   Classification of hyperspectral data from urban areas based on extended morphological profiles [J].
Benediktsson, JA ;
Palmason, JA ;
Sveinsson, JR .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2005, 43 (03) :480-491
[3]   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
[4]   Local Block Multilayer Sparse Extreme Learning Machine for Effective Feature Extraction and Classification of Hyperspectral Images [J].
Cao, Faxian ;
Yang, Zhijing ;
Ren, Jinchang ;
Chen, Weizhao ;
Han, Guojun ;
Shen, Yuzhen .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2019, 57 (08) :5580-5594
[5]   A Hyperspectral Image Classification Method Using Multifeature Vectors and Optimized KELM [J].
Chen, Huayue ;
Miao, Fang ;
Chen, Yijia ;
Xiong, Yijun ;
Chen, Tao .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2021, 14 :2781-2795
[6]  
Chen Yujia, 2020, IEEE Trans. Geosci. Remote Sens., V18, P162
[7]   Classification of Hyperspectral Images by Using Extended Morphological Attribute Profiles and Independent Component Analysis [J].
Dalla Mura, Mauro ;
Villa, Alberto ;
Benediktsson, Jon Atli ;
Chanussot, Jocelyn ;
Bruzzone, Lorenzo .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2011, 8 (03) :542-546
[8]   Noise-Robust Hyperspectral Image Classification via Multi-Scale Total Variation [J].
Duan, Puhong ;
Kang, Xudong ;
Li, Shutao ;
Ghamisi, Pedram .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2019, 12 (06) :1948-1962
[9]   Spectral and Spatial Classification of Hyperspectral Data Using SVMs and Morphological Profiles [J].
Fauvel, Mathieu ;
Benediktsson, Jon Atli ;
Chanussot, Jocelyn ;
Sveinsson, Johannes R. .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2008, 46 (11) :3804-3814
[10]  
Fong A., 2020, C LAS EL 10 15 MAY 2, DOI [https://doi.org/10.1364/CLEO_AT.2020.AW3K.2, DOI 10.1364/CLEO_AT.2020.AW3K.2]