DMAF-NET: Deep Multi-Scale Attention Fusion Network for Hyperspectral Image Classification with Limited Samples

被引:3
作者
Guo, Hufeng [1 ,2 ]
Liu, Wenyi [1 ]
机构
[1] North Univ China, Sch Instrument & Elect, State Key Lab Dynam Measurement Technol, Taiyuan 030051, Peoples R China
[2] Henan Coll Transportat, Dept Transportat Informat Engn, Zhengzhou 451460, Peoples R China
基金
美国国家科学基金会;
关键词
convolutional neural network (CNN); hyperspectral image (HSI) classification; limited samples; multi-scale feature extraction; multi-scale spatial-spectral attention; pyramidal multi-scale channel attention; multi-attention feature fusion;
D O I
10.3390/s24103153
中图分类号
O65 [分析化学];
学科分类号
070302 ; 081704 ;
摘要
In recent years, deep learning methods have achieved remarkable success in hyperspectral image classification (HSIC), and the utilization of convolutional neural networks (CNNs) has proven to be highly effective. However, there are still several critical issues that need to be addressed in the HSIC task, such as the lack of labeled training samples, which constrains the classification accuracy and generalization ability of CNNs. To address this problem, a deep multi-scale attention fusion network (DMAF-NET) is proposed in this paper. This network is based on multi-scale features and fully exploits the deep features of samples from multiple levels and different perspectives with an aim to enhance HSIC results using limited samples. The innovation of this article is mainly reflected in three aspects: Firstly, a novel baseline network for multi-scale feature extraction is designed with a pyramid structure and densely connected 3D octave convolutional network enabling the extraction of deep-level information from features at different granularities. Secondly, a multi-scale spatial-spectral attention module and a pyramidal multi-scale channel attention module are designed, respectively. This allows modeling of the comprehensive dependencies of coordinates and directions, local and global, in four dimensions. Finally, a multi-attention fusion module is designed to effectively combine feature mappings extracted from multiple branches. Extensive experiments on four popular datasets demonstrate that the proposed method can achieve high classification accuracy even with fewer labeled samples.
引用
收藏
页数:26
相关论文
共 65 条
[11]   Drop an Octave: Reducing Spatial Redundancy in Convolutional Neural Networks with Octave Convolution [J].
Chen, Yunpeng ;
Fan, Haoqi ;
Xu, Bing ;
Yan, Zhicheng ;
Kalantidis, Yannis ;
Rohrbach, Marcus ;
Yan, Shuicheng ;
Feng, Jiashi .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :3434-3443
[12]   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
[13]   Integrating Hybrid Pyramid Feature Fusion and Coordinate Attention for Effective Small Sample Hyperspectral Image Classification [J].
Ding, Chen ;
Chen, Youfa ;
Li, Runze ;
Wen, Dushi ;
Xie, Xiaoyan ;
Zhang, Lei ;
Wei, Wei ;
Zhang, Yanning .
REMOTE SENSING, 2022, 14 (10)
[14]   Context-Aware Guided Attention Based Cross-Feedback Dense Network for Hyperspectral Image Super-Resolution [J].
Dong, Wenqian ;
Qu, Jiahui ;
Zhang, Tongzhen ;
Li, Yunsong ;
Du, Qian .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[15]   Hyperspectral Image Classification With Squeeze Multibias Network [J].
Fang, Leyuan ;
Liu, Guangyun ;
Li, Shutao ;
Ghamisi, Pedram ;
Benediktsson, Jon Atli .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2019, 57 (03) :1291-1301
[16]   A New Spatial-Spectral Feature Extraction Method for Hyperspectral Images Using Local Covariance Matrix Representation [J].
Fang, Leyuan ;
He, Nanjun ;
Li, Shutao ;
Plaza, Antonio J. ;
Plaza, Javier .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2018, 56 (06) :3534-3546
[17]   Crop Type Classification by DESIS Hyperspectral Imagery and Machine Learning Algorithms [J].
Farmonov, Nizom ;
Amankulova, Khilola ;
Szatmari, Jozsef ;
Sharifi, Alireza ;
Abbasi-Moghadam, Dariush ;
Nejad, Seyed Mahdi Mirhoseini ;
Mucsi, Laszlo .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2023, 16 :1576-1588
[18]  
Fauvel M, 2007, INT GEOSCI REMOTE SE, P4834
[19]   3D Octave and 2D Vanilla Mixed Convolutional Neural Network for Hyperspectral Image Classification with Limited Samples [J].
Feng, Yuchao ;
Zheng, Jianwei ;
Qin, Mengjie ;
Bai, Cong ;
Zhang, Jinglin .
REMOTE SENSING, 2021, 13 (21)
[20]   Semi-supervised enhanced discriminative local constraint preserving projection for dimensionality reduction of medical hyperspectral images [J].
Gao, Hongmin ;
Yang, Mengran ;
Cao, Xueying ;
Liu, Qin ;
Xu, Peipei .
COMPUTERS IN BIOLOGY AND MEDICINE, 2023, 167