Joint U-Nets with hierarchical graph structure and sparse Transformer for hyperspectral image classification

被引:0
|
作者
Zhu, Pengfei [1 ]
Liu, Jinglei [1 ]
机构
[1] Yantai Univ, Sch Comp & Control Engn, Yantai 264005, Shandong, Peoples R China
基金
中国国家自然科学基金;
关键词
Hyperspectral image (HSI) classification; Hierarchical graph merging block; Sparse transformer block; Sparse self-attention mechanism; U-net; CONVOLUTIONAL NETWORKS;
D O I
10.1016/j.eswa.2025.127046
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Hyperspectral image (HSI) classification is a critical task in remote sensing, but existing U-Net and transformer- based models encounter significant challenges. Traditional U-Net architectures struggle with multi-scale feature extraction due to their fixed convolutional kernels, limiting their effectiveness in capturing complex spatial distributions. Transformer models, while adept at capturing global context, suffer from high computational complexity and inadequate sensitivity to local features in HSI. To address these limitations, we propose a novel joint U-Nets with hierarchical graph structure and sparse transformer (HGSTNet). HGSTNet introduces hierarchical graph merging blocks and incremental merging methods to dynamically extract and fuse multi-scale features, leveraging superpixel segmentation and hierarchical graph topology to enhance spatial correlation. Furthermore, to enhance the model's global context perception, we integrate a sparse transformer block in the first four encoder-decoder. Unlike traditional transformers, the sparse transformer reduces computational complexity and enhances feature capture by incorporating the sparse self-attention (SPA) module, which utilizes the sparse self-attention mechanism to suppress low-relevance or redundant information, thereby improving the capture of both local and global features. Experiments conducted on multiple HSI datasets, along with comparisons to other deep learning methods, demonstrate that HGSTNet exhibits strong competitiveness.
引用
收藏
页数:19
相关论文
共 32 条
  • [31] Dual-Branch Grouping Multiscale Residual Embedding U-Net and Cross-Attention Fusion Networks for Hyperspectral Image Classification
    Ouyang, Ning
    Huang, Chenyu
    Lin, Leping
    INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2024, 15 (01) : 614 - 621
  • [32] GLGAT-CFSL: Global-Local Graph Attention Network-Based Cross-Domain Few-Shot Learning for Hyperspectral Image Classification
    Ding, Chen
    Deng, Zhicong
    Xu, Yaoyang
    Zheng, Mengmeng
    Zhang, Lei
    Cao, Yu
    Wei, Wei
    Zhang, Yanning
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62 : 1 - 19