LSSMA: Lightweight Spectral–Spatial Neural Architecture With Multiattention Feature Extraction for Hyperspectral Image Classification

被引:2
|
作者
Ding, Shujie [1 ]
Ruan, Xiaoli [1 ]
Yang, Jing [1 ]
Sun, Jie [1 ]
Li, Shaobo [1 ]
Hu, Jianjun [2 ]
机构
[1] Guizhou Univ, State Key Lab Publ Big Data, Guiyang 550025, Peoples R China
[2] Univ South Carolina, Dept Comp Sci & Engn, Columbia, SC 29208 USA
关键词
Feature extraction; Hyperspectral imaging; Data mining; Image classification; Computational modeling; Data models; Context modeling; Hyperspectral image (HSI) classification; lightweight; multiattention feature extraction; spectral-spatial neural network; ATTENTION NETWORK; RESIDUAL NETWORK; GENERATION;
D O I
10.1109/JSTARS.2024.3371536
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Deep learning has been utilized for hyperspectral image (HSI) classification in recent years, with notable performance improvements. In particular, convolutional neural networks (CNNs) methods have achieved major advancements in this area. However, there are some drawbacks to the existing CNN-based HSI classification approaches: 1) the lack of effective and simple feature representations in CNNs, which overlook the effects of spectral differences and spatial contextual information; 2) the classification model has an enormous network complexity as a result of its numerous training parameters and high computational requirements; and 3) the category samples in HSI data exhibit a significant long tail distribution issue, which affects the classification performance of HSI. To address these issues, we propose a lightweight spectral-spatial neural architecture with multiattention feature extraction (LSSMA) for HSI classification. The main work consists of three areas: 1) A spectral feature extraction and fusing module is created to facilitate the fusing of spectral-spatial features while reducing the number of trainable parameters and computational complexity of the model. This module uses convolutions of various kernel sizes for residual connection and feature fusion, and introduces group convolution to achieve efficient feature representation. 2) To utilize the spectral-spatial correlation of HSI data to its fullest, a multiscale convolutional activation guided attention mechanism is designed and the position attention module is referenced, which can capture spectral differences and spatial contextual relationships between ground objects. and 3) Focal loss is applied in computer vision tasks to the LSSMA model to enhance its capacity to handle category imbalance. Experimental results on four publicly available hyperspectral datasets show that the method obtains better classification performance at a lower computational cost.
引用
收藏
页码:6394 / 6413
页数:20
相关论文
共 50 条
  • [31] HYPERSPECTRAL IMAGE CLASSIFICATION BASED ON MULTISCALE SPATIAL AND SPECTRAL FEATURE NETWORK
    Tang, Xu
    Meng, Fanbo
    Ma, Jingjing
    Zhang, Xiangrong
    Liu, Fang
    Peng, Qunnie
    Jiao, Licheng
    IGARSS 2020 - 2020 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM, 2020, : 838 - 841
  • [32] Multiscale spectral-spatial feature learning for hyperspectral image classification
    Sohail, Muhammad
    Chen, Zhao
    Yang, Bin
    Liu, Guohua
    DISPLAYS, 2022, 74
  • [33] Adaptive Spatial-Spectral Feature Learning for Hyperspectral Image Classification
    Li, Simin
    Zhu, Xueyu
    Liu, Yang
    Bao, Jie
    IEEE ACCESS, 2019, 7 : 61534 - 61547
  • [34] Spectral-Spatial Feature Extraction for Hyperspectral Image Classification: A Dimension Reduction and Deep Learning Approach
    Zhao, Wenzhi
    Du, Shihong
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2016, 54 (08): : 4544 - 4554
  • [35] Hyperspectral Image Classification Based on Spectral Spatial Feature Extraction and Deep Rotation Forest Ensemble with AdaBoost
    Deng, Lindiao
    Cao, Guo
    Xu, Ling
    Xu, Hao
    Pan, Qikun
    Ding, Lanwei
    Shang, Yanfeng
    FOURTEENTH INTERNATIONAL CONFERENCE ON GRAPHICS AND IMAGE PROCESSING, ICGIP 2022, 2022, 12705
  • [36] Three-Dimensional Spatial-Spectral Filtering Based Feature Extraction for Hyperspectral Image Classification
    Akyurek, Hasan Ali
    Kocer, Baris
    ADVANCES IN ELECTRICAL AND COMPUTER ENGINEERING, 2017, 17 (02) : 95 - 102
  • [37] PASSNet: A Spatial-Spectral Feature Extraction Network With Patch Attention Module for Hyperspectral Image Classification
    Ji, Renjie
    Tan, Kun
    Wang, Xue
    Pan, Chen
    Xin, Liang
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2023, 20
  • [38] Spectral–Spatial Hyperspectral Image Classification via Non-local Means Filtering Feature Extraction
    Tu B.
    Zhang X.
    Wang J.
    Zhang G.
    Ou X.
    Tu, Bing (tubing@hnist.edu.cn), 2018, Springer Science and Business Media, LLC (19):
  • [39] Spectral-Spatial Attention Feature Extraction for Hyperspectral Image Classification Based on Generative Adversarial Network
    Liang, Hongbo
    Bao, Wenxing
    Shen, Xiangfei
    Zhang, Xiaowu
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2021, 14 : 10017 - 10032
  • [40] Discriminative Spectral-Spatial Feature Extraction-Based Band Selection for Hyperspectral Image Classification
    Baisantry, Munmun
    Sao, Anil Kumar
    Shukla, Dericks Praise
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60