arrow
返回

A Lightweight Transformer Network for Hyperspectral Image Classification

delete2023-01-01
delete22
PRE
AI
X
Xuming Zhang
苏远超 封面图
苏远超 (Yuanchao Su)
高
高连如 (Lianru Gao)
L
Lorenzo Bruzzone
顾行发 封面图
顾行发 (Xingfa Gu)
Q
Qingjiu Tian *
DOI:10.1109/TGRS.2023.3297858delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Transformer是捕获远程依赖性的强大工具,并且在高光谱图像 (HSI) 分类中显示出令人印象深刻的性能。然而,这种能力伴随着沉重的存储器占用和巨大的计算负担。在本文中,我们提出了两种类型的轻量级自注意模块 (通道轻量级多头自注意 (CLMSA) 模块和位置轻量级多头自注意 (PLMSA) 模块),以减少内存和计算量,同时将每个像素或通道与全局信息相关联。此外,我们发现,由于固定的输入大小,变压器在显式提取局部和多尺度特征方面是无效的,并且在处理少量训练样本时往往会过度拟合。因此,提出了一种使用建议的轻量级自我注意模块构建的轻量级变压器 (LiT) 网络。LiT采用卷积块来显式提取早期层中的局部信息,并采用变压器来捕获深层中的远程依赖关系。此外,我们设计了一种受控多类分层 (CMS) 采样策略,以生成适当大小的输入数据,确保平衡采样,并减少训练样本和测试样本之间特征提取区域的重叠。通过适当的训练数据,卷积标记化和LiTs,LiT可以缓解过度拟合,并具有较高的计算效率和良好的性能。在多个HSI数据集上的实验结果验证了我们设计的有效性。
Keyword:
Deep learning (DL)
hyperspectral image (HSI) classification
transformer

期刊

IEEE Transactions on Geoscience and Remote Sensing 封面图
IEEE Transactions on Geoscience and Remote Sensing
IF:
8.6
论文数:
2.1W
被引数:
10.7W

机构

A
aerospace information research institute, cas
学者数:
1.5K
论文数: 1.3K
被引数: 0
X
xi'an university of science & technology
学者数:
6.9K
论文数: 4.8K
被引数: 5
N
nanjing university
学者数:
7.8W
论文数: 5.6W
被引数: 87
C
chinese academy of sciences
学者数:
56.7W
论文数: 45.0W
被引数: 704
学者 查看更多机构
引用论文

引用论文

Interactive Rendering of Acquired Materials on Dynamic Geometry Using Frequency Analysis
err2013-05-01
err0
errOAAI
errMahdi Mohammad Bagher; Cyril Soler; Kartic Subr; Laurent Belcour; Nicolas Holzschuch
err分享
err收藏
Efficient Deep Learning of Nonlocal Features for Hyperspectral Image Classification
err2021-07-01
err81
errOAAI
errShen, Yu; Zhu, Sijie; Chen, Chen; Du, Qian; Xiao, Liang; Chen, Jianyu; Pan, Delu
err分享
err收藏
Spectral and Spatial Classification of Hyperspectral Data Using SVMs and Morphological Profiles
err2008-11-01
err1.0K
PREAI
errFauvel, Mathieu; Benediktsson, Jon Atli; Chanussot, Jocelyn; Sveinsson, Johannes R.
err分享
err收藏
学者 查看更多内容