Multiple vision architectures-based hybrid network for hyperspectral image classification

被引:25
|
作者
Zhao, Feng [1 ]
Zhang, Junjie [1 ]
Meng, Zhe [1 ]
Liu, Hanqiang [2 ]
Chang, Zhenhui [3 ]
Fan, Jiulun [1 ]
机构
[1] Xian Univ Posts & Telecommun, Sch Commun & Informat Engn, Xian 710121, Peoples R China
[2] Shaanxi Normal Univ, Sch Comp Sci, Xian 710119, Peoples R China
[3] Xian Univ Posts & Telecommun, Sch Cyberspace Secur, Xian 710121, Peoples R China
基金
中国国家自然科学基金;
关键词
Hyperspectral image classification; Convolutional neural network; Vision transformer; Graph convolutional network; GRAPH CONVOLUTIONAL NETWORKS;
D O I
10.1016/j.eswa.2023.121032
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
More recently, vision transformer (ViT) has shown competitive performance with convolutional neural network (CNN) on computer vision tasks, which provided more possibilities for accurate classification of hyperspectral image (HSI). However, whether CNN or ViT, they generally only focus on single type of feature, resulting in insufficient information utilization. For instance, CNN has powerful local feature extraction ability, while ViT pays more attention to long-range dependencies and global features. To consider multiple types of feature information, we propose a multiple vision architectures-based hybrid network (MVAHN) for HSI classification, which consists of joint CNN and transformer (JCT) structure and graph convolutional module (GCM). Firstly, JCT successfully embeds convolution operations into ViT to capture local and global features, which mainly include: 1) A spectral spatial convolution block (SSCB) is proposed to unearth local spectral spatial features. 2) A convolution embedding is aggregated into self-attention to design a local-global attention (LGA) mechanism, which can realize the seamless integration of CNN and ViT, thereby capturing local-global combined features. Secondly, a plug-and-play GCM is developed in parallel with transformer encoders to further improve the model classification ability by mining the similarity relationship between pixels in HSI. Overall, an elegant integration of these seemingly distinct paradigms is realized by MVAHN to capture multiple types of feature information. The overall accuracies (OAs) of MVAHN on Pavia University, Houston 2013, Salinas Valley, Kennedy Space Center, Indian Pines and Botswana datasets are 96.37%, 88.33%, 97.57%, 98.96%, 96.25% and 99.26%, respectively. Compared with the state-of-the-art hybrid models, MVAHN achieves competitive classification results. The source code will be available at https://github.com/ZJier/MVAHN.
引用
收藏
页数:16
相关论文
共 50 条
  • [1] HYBRID VISION TRANSFORMER MODEL FOR HYPERSPECTRAL IMAGE CLASSIFICATION
    Yang, Jiaqi
    Du, Bo
    Wu, Chen
    2022 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2022), 2022, : 1388 - 1391
  • [2] A hybrid neural network for hyperspectral image classification
    Li, Jiangyun
    Liang, Binxiu
    Wang, Yuhao
    REMOTE SENSING LETTERS, 2020, 11 (01) : 96 - 105
  • [3] A Hybrid Capsule Network for Hyperspectral Image Classification
    Khodadadzadeh, Massoud
    Ding, Xuemei
    Chaurasia, Priyanka
    Coyle, Damien
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2021, 14 : 11824 - 11839
  • [4] ParHybNet: Parallel Hybrid Network for Hyperspectral Image Classification
    Sarkar, Anish
    Nandi, Utpal
    Changdar, Chiranjit
    Paul, Bachchu
    Si, Tapas
    JOURNAL OF THE INDIAN SOCIETY OF REMOTE SENSING, 2024, : 1507 - 1524
  • [5] Adaptive hybrid attention network for hyperspectral image classification *
    Pande, Shivam
    Banerjee, Biplab
    PATTERN RECOGNITION LETTERS, 2021, 144 : 6 - 12
  • [6] Hyperspectral image classification based on multi-scale hybrid convolutional network
    Yang, Yun
    Zhou, Yao
    Chen, Jia-ning
    CHINESE JOURNAL OF LIQUID CRYSTALS AND DISPLAYS, 2023, 38 (03) : 368 - 377
  • [7] High-Resolution Hyperspectral Image Classification Based on Hybrid Convolutional Network
    Shen Bingzhi
    Nie Ruomei
    Jiang Haipeng
    Yang Zhishuai
    Song Mingrui
    Chen Siqi
    Li Xinwei
    LASER & OPTOELECTRONICS PROGRESS, 2022, 59 (24)
  • [8] Hyperspectral Image Classification Based on Capsule Network
    Ma Qiaoyu
    Zhang Xin
    Zhang Chunlei
    Zhou Heng
    CHINESE JOURNAL OF ELECTRONICS, 2022, 31 (01) : 146 - 154
  • [9] HYPERSPECTRAL IMAGE CLASSIFICATION BASED ON PCA NETWORK
    Wang, Fan
    Zhang, Rong
    Wu, Qian
    2016 8TH WORKSHOP ON HYPERSPECTRAL IMAGE AND SIGNAL PROCESSING: EVOLUTION IN REMOTE SENSING (WHISPERS), 2016,
  • [10] CLASSIFICATION BASED ON CAPSULE NETWORK WITH HYPERSPECTRAL IMAGE
    Ma, Yi
    Zheng, Zezhong
    Guo, Zhengqiang
    Mou, Fan
    Zhou, Fangrong
    Kong, Rui
    Hou, Ankai
    Zhu, Mingcang
    He, Yong
    Ren, Juan
    Chen, Huaixin
    Liu, Zhigang
    Zhou, Guoqing
    Li, Jiang
    2019 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2019), 2019, : 2750 - 2753