返回
Hyperspectral Image Classification Based on Spectral-Spatial Attention Tensor Network
DOI:10.1109/LGRS.2023.3332600.png)
摘要
En 中文
As an essential task in remote sensing, hyperspectral image classification (HSIC) has been extensively studied. Although many methods based on deep learning have been inducted to improve the performance of HSIC, they still face two challenges: 1) Can the joint spectral-spatial information in the hyperspectral image (HSI) be effectively utilized? 2) Does each pixel in a certain sample contribute equally to classification? We exploit a spectral-spatial attention tensor network (SSATN) in this letter, where the coordinate attention (CA) mechanism is first introduced into a full tensor network. We present a learnable tensor squeezing projection layer (TSPL) instead of the classical pooling layer for the CA block, which enables the network to selectively focus on discriminative features in spectral and spatial. Besides, the SSATN accepts raw HSI data as the input without dimensionality reduction in advance. Compared with the state-of-the-art convolutional neural network (CNN) methods, SSATN can effectively capture certain spectral-spatial features via tensor transformation with fewer parameters. The experimental results on two widely used HSI datasets prove the advantage of the SSATN method.
Keyword:
Tensors
Feature extraction
Task analysis
Convolutional neural networks
Transformers
Image classification
Hyperspectral imaging
Hyperspectral image classification (HSIC)
spectral-spatial attention
tensor network
期刊
IF:
16.4
论文数:
1.0W
被引数:
5.1K
机构
引用论文
Investigation on the genotoxicity of different sizes of gold nanoparticles administered to the lungs of rats不同粒径纳米金对大鼠肺的遗传毒性研究
Caw’s Walking State Recognition Based on Accelerometers and Gyroscopes Installed on Ear-Tags and Collar-Tags基于安装在耳标和项圈上的加速度计和陀螺仪的Caw步行状态识别
Spectral Super-Resolution of Multispectral Images Using Spatial-Spectral Residual Attention Network基于空间光谱残差注意网络的多光谱图像光谱超分辨

