Hyperspectral Image Classification Based on Multibranch Adaptive Feature Fusion Network

被引:1
作者
Li, Chen [1 ]
Wang, Yi [1 ,2 ]
Fang, Zhice [1 ]
Li, Penglei [1 ]
机构
[1] China Univ Geosci, Sch Geophys & Geomat, Wuhan 430074, Peoples R China
[2] State Key Lab Resources & Environm Informat Syst, Beijing 100101, Peoples R China
来源
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING | 2024年 / 62卷
基金
中国国家自然科学基金;
关键词
Feature extraction; Hyperspectral imaging; Data mining; Three-dimensional displays; Convolution; Interference; Convolutional neural networks; Convolutional neural networks (CNNs); feature fusion; hyperspectral image classification (HSIC); multibranch; SPECTRAL-SPATIAL CLASSIFICATION; ZERO-SHOT; ATTENTION;
D O I
10.1109/TGRS.2024.3449878
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
Convolutional neural networks (CNNs) are widely used in hyperspectral image classification (HSIC) due to their exceptional performance. However, current methods for multiscale feature extraction typically rely on single-branch CNNs, potentially causing interference among features of varying scales. To mitigate this issue, we present a multibranch adaptive feature fusion network (MBAFFN) classification method. MBAFFN enhances feature uniqueness and improves the accuracy and reliability of classification results by extracting information at multiple scales through three parallel branches. Furthermore, to address the challenge of capturing global features within CNNs, we introduce a global detail attention (GDA) mechanism aimed at bolstering the network's capability to capture comprehensive information. In addition, we mitigate the issue of neglecting center-pixel importance in convolution operations through a distance suppression attention (DSA) design. To effectively integrate outcomes from multiple branches, we propose a pixel-based adaptive feature fusion strategy, thereby increasing the proportion of features conducive to improved classification results. Lastly, auxiliary loss functions are employed to train the multibranch network. Experimental results on four benchmark datasets demonstrate the superiority of our approach over several state-of-the-art methods, particularly in managing imbalanced small samples. Furthermore, ablation studies validate the effectiveness of the proposed modules.
引用
收藏
页数:18
相关论文
共 47 条
[1]   Tri-CNN: A Three Branch Model for Hyperspectral Image Classification [J].
Alkhatib, Mohammed Q. Q. ;
Al-Saad, Mina ;
Aburaed, Nour ;
Almansoori, Saeed ;
Zabalza, Jaime ;
Marshall, Stephen ;
Al-Ahmad, Hussain .
REMOTE SENSING, 2023, 15 (02)
[2]   A Novel Hyperspectral Image Classification Model Using Bole Convolution With Three-Direction Attention Mechanism: Small Sample and Unbalanced Learning [J].
Cai, Weiwei ;
Ning, Xin ;
Zhou, Guoxiong ;
Bai, Xiao ;
Jiang, Yizhang ;
Li, Wei ;
Qian, Pengjiang .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
[3]   Cascaded dual-scale crossover network for hyperspectral image classification [J].
Cao, Feilong ;
Guo, Wenhui .
KNOWLEDGE-BASED SYSTEMS, 2020, 189
[4]   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
[5]   Feature Fusion Network Model Based on Dual Attention Mechanism for Hyperspectral Image Classification [J].
Cui, Ying ;
Li, WenShan ;
Chen, Liwei ;
Wang, Liguo ;
Jiang, Jing ;
Gao, Shan .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
[6]   Lightweight Spectral-Spatial Attention Network for Hyperspectral Image Classification [J].
Cui, Ying ;
Xia, Jinbiao ;
Wang, Zhiteng ;
Gao, Shan ;
Wang, Liguo .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[7]  
Gao L., 2021, IEEE Transactions on Geoscience and Remote Sensing, DOI [DOI 10.1109/TGRS.2021.3130716, 10.1109/TGRS.2021.3130716]
[8]   A Survey on Spectral-Spatial Classification Techniques Based on Attribute Profiles [J].
Ghamisi, Pedram ;
Dalla Mura, Mauro ;
Benediktsson, Jon Atli .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2015, 53 (05) :2335-2353
[9]   Dual-View Spectral and Global Spatial Feature Fusion Network for Hyperspectral Image Classification [J].
Guo, Tan ;
Wang, Ruizhi ;
Luo, Fulin ;
Gong, Xiuwen ;
Zhang, Lei ;
Gao, Xinbo .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
[10]   Investigation of the random forest framework for classification of hyperspectral data [J].
Ham, J ;
Chen, YC ;
Crawford, MM ;
Ghosh, J .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2005, 43 (03) :492-501