Residual Spatial Attention Kernel Generation Network for Hyperspectral Image Classification with Small Sample Size

Yanbing Xu, Yanmei Zhang*, Chengcheng Yu, Chao Ji, Tingxuan Yue, Huan Li

*此作品的通讯作者

科研成果: 期刊稿件文章同行评审

10 引用 (Scopus)

摘要

With the rapid development of deep learning, the convolutional neural networks (CNNs) have been widely used in hyperspectral image classification (HSIC) and achieved excellent performance. However, CNNs reuse the same kernel weights over different locations, thus resulting in the insufficient capability of capturing diversity spatial interactions. Moreover, CNNs usually require a large amount of training samples to optimize the learnable parameters. When training samples are limited, the classification performance of CNN tends to drop off a cliff. To tackle the aforementioned issues, a novel residual spatial attention kernel generation network (RSAKGN) is proposed for HSIC. First, a spatial attention kernel generation module (SAKGM) is built to extract discriminative semantic features, which can dynamically calculate the attention weights to generate specific spatial attention kernels over different locations. Then, we combine the SAKGM with residual learning framework by embedding the SAKGM into a bottleneck residual block to obtain the residual spatial attention block (RSAB). The RSAKGN is constructed by stacking several RSABs. Experimental results on three public HSI datasets demonstrate that the proposed RSAKGN method outperforms several state-of-the-arts with small sample size.

源语言英语
文章编号5529714
期刊IEEE Transactions on Geoscience and Remote Sensing
60
DOI
出版状态已出版 - 2022

指纹

探究 'Residual Spatial Attention Kernel Generation Network for Hyperspectral Image Classification with Small Sample Size' 的科研主题。它们共同构成独一无二的指纹。

引用此