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 条
[1]   Measurement of Total Dissolved Solids and Total Suspended Solids in Water Systems: A Review of the Issues, Conventional, and Remote Sensing Techniques [J].
Adjovu, Godson Ebenezer ;
Stephen, Haroon ;
James, David ;
Ahmad, Sajjad .
REMOTE SENSING, 2023, 15 (14)
[2]   Opportunities and Challenges of Spaceborne Sensors in Delineating Land Surface Temperature Trends: A Review [J].
Ahmed, M. Razu ;
Ghaderpour, Ebrahim ;
Gupta, Anil ;
Dewan, Ashraf ;
Hassan, Quazi K. .
IEEE SENSORS JOURNAL, 2023, 23 (07) :6460-6472
[3]   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)
[4]   New Generation Hyperspectral Data From DESIS Compared to High Spatial Resolution PlanetScope Data for Crop Type Classification [J].
Aneece, Itiya ;
Foley, Daniel ;
Thenkabail, Prasad ;
Oliphant, Adam ;
Teluguntla, Pardhasaradhi .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2022, 15 :7846-7858
[5]   Deep Learning for Classification of Hyperspectral Data [J].
Audebert, Nicolas ;
Le Saux, Bertrand ;
Lefevre, Sebastien .
IEEE GEOSCIENCE AND REMOTE SENSING MAGAZINE, 2019, 7 (02) :159-173
[6]   3-D Deep Learning Approach for Remote Sensing Image Classification [J].
Ben Hamida, Amina ;
Benoit, Alexandre ;
Lambert, Patrick ;
Ben Amar, Chokri .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2018, 56 (08) :4420-4434
[7]   Transformer-Based Masked Autoencoder With Contrastive Loss for Hyperspectral Image Classification [J].
Cao, Xianghai ;
Lin, Haifeng ;
Guo, Shuaixu ;
Xiong, Tao ;
Jiao, Licheng .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
[8]   Machine Learning Methods in Weather and Climate Applications: A Survey [J].
Chen, Liuyi ;
Han, Bocheng ;
Wang, Xuesong ;
Zhao, Jiazhen ;
Yang, Wenke ;
Yang, Zhengyi .
APPLIED SCIENCES-BASEL, 2023, 13 (21)
[9]   Multiscale Feature Learning by Transformer for Building Extraction From Satellite Images [J].
Chen, Xin ;
Qiu, Chunping ;
Guo, Wenyue ;
Yu, Anzhu ;
Tong, Xiaochong ;
Schmitt, Michael .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
[10]   Hyperspectral Image Classification via Kernel Sparse Representation [J].
Chen, Yi ;
Nasrabadi, Nasser M. ;
Tran, Trac D. .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2013, 51 (01) :217-231