Hyperspectral Image Classification with the Orthogonal Self-Attention ResNet and Two-Step Support Vector Machine

被引:3
作者
Sun, Heting [1 ]
Wang, Liguo [1 ]
Liu, Haitao [1 ]
Sun, Yinbang [1 ]
机构
[1] Dalian Minzu Univ, Coll Informat & Commun Engn, Dalian 116600, Peoples R China
基金
中国国家自然科学基金;
关键词
hyperspectral image classification; orthogonal self-attention module; channel attention module; two-step support vector machine;
D O I
10.3390/rs16061010
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
Hyperspectral image classification plays a crucial role in remote sensing image analysis by classifying pixels. However, the existing methods require more spatial-global information interaction and feature extraction capabilities. To overcome these challenges, this paper proposes a novel model for hyperspectral image classification using an orthogonal self-attention ResNet and a two-step support vector machine (OSANet-TSSVM). The OSANet-TSSVM model comprises two essential components: a deep feature extraction network and an improved support vector machine (SVM) classification module. The deep feature extraction network incorporates an orthogonal self-attention module (OSM) and a channel attention module (CAM) to enhance the spatial-spectral feature extraction. The OSM focuses on computing 2D self-attention weights for the orthogonal dimensions of an image, resulting in a reduced number of parameters while capturing comprehensive global contextual information. In contrast, the CAM independently learns attention weights along the channel dimension. The CAM autonomously learns attention weights along the channel dimension, enabling the deep network to emphasise crucial channel information and enhance the spectral feature extraction capability. In addition to the feature extraction network, the OSANet-TSSVM model leverages an improved SVM classification module known as the two-step support vector machine (TSSVM) model. This module preserves the discriminative outcomes of the first-level SVM subclassifier and remaps them as new features for the TSSVM training. By integrating the results of the two classifiers, the deficiencies of the individual classifiers were effectively compensated, resulting in significantly enhanced classification accuracy. The performance of the proposed OSANet-TSSVM model was thoroughly evaluated using public datasets. The experimental results demonstrated that the model performed well in both subjective and objective evaluation metrics. The superiority of this model highlights its potential for advancing hyperspectral image classification in remote sensing applications.
引用
收藏
页数:20
相关论文
共 26 条
[1]   Feature Line Embedding Based on Support Vector Machine for Hyperspectral Image Classification [J].
Chen, Ying-Nong ;
Thaipisutikul, Tipajin ;
Han, Chin-Chuan ;
Liu, Tzu-Jui ;
Fan, Kuo-Chin .
REMOTE SENSING, 2021, 13 (01) :1-29
[2]   SUPPORT-VECTOR NETWORKS [J].
CORTES, C ;
VAPNIK, V .
MACHINE LEARNING, 1995, 20 (03) :273-297
[3]   Classification of visible and infrared hyperspectral images based on image segmentation and edge-preserving filtering [J].
Cui, Binge ;
Ma, Xiudan ;
Xie, Xiaoyun ;
Ren, Guangbo ;
Ma, Yi .
INFRARED PHYSICS & TECHNOLOGY, 2017, 81 :79-88
[4]  
Dosovitskiy A, 2021, Arxiv, DOI arXiv:2010.11929
[5]   Multi-Complementary Generative Adversarial Networks With Contrastive Learning for Hyperspectral Image Classification [J].
Feng, Jie ;
Gao, Zizhuo ;
Shang, Ronghua ;
Zhang, Xiangrong ;
Jiao, Licheng .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
[6]   Hyperspectral Image Classification Using a Hybrid 3D-2D Convolutional Neural Networks [J].
Ghaderizadeh, Saeed ;
Abbasi-Moghadam, Dariush ;
Sharifi, Alireza ;
Zhao, Na ;
Tariq, Aqil .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2021, 14 (14) :7570-7588
[7]   Generative Adversarial Networks [J].
Goodfellow, Ian ;
Pouget-Abadie, Jean ;
Mirza, Mehdi ;
Xu, Bing ;
Warde-Farley, David ;
Ozair, Sherjil ;
Courville, Aaron ;
Bengio, Yoshua .
COMMUNICATIONS OF THE ACM, 2020, 63 (11) :139-144
[8]   HyperViTGAN: Semisupervised Generative Adversarial Network With Transformer for Hyperspectral Image Classification [J].
He, Ziping ;
Xia, Kewen ;
Ghamisi, Pedram ;
Hu, Yuhen ;
Fan, Shurui ;
Zu, Baokai .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2022, 15 :6053-6068
[9]   Semisupervised Learning of Hyperspectral Data With Unknown Land-Cover Classes [J].
Jun, Goo ;
Ghosh, Joydeep .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2013, 51 (01) :273-282
[10]   Deep proximal support vector machine classifiers for hyperspectral images classification [J].
Kalaiarasi, Ganesan ;
Maheswari, Sureshbabu .
NEURAL COMPUTING & APPLICATIONS, 2021, 33 (20) :13391-13415