arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2203.16952cs.CVcs.LGeess.IV

多模态融合Transformer用于遥感图像分类

Multimodal Fusion Transformer for Remote Sensing Image Classification

Jalpaiguri Government Engineering College(贾尔派古里政府工程学院);Technical University of Munich(慕尼黑工业大学);Aerospace Information Research Institute, Chinese Academy of Sciences(中国科学院空天信息创新研究院);Helmholtz-Zentrum Dresden-Rossendorf(亥姆霍兹德累斯顿罗森多夫研究中心);University of Extremadura(埃斯特雷马杜拉大学);Univ. Grenoble Alpes(格勒诺布尔阿尔卑斯大学);CNRS(法国国家科学研究中心);Grenoble INP(格勒诺布尔国立理工学院)

Swalpa Kumar Roy, Ankur Deria, Danfeng Hong, Behnood Rasti, Antonio Plaza, Jocelyn Chanussot

AI总结:

提出一种多模态融合Transformer网络,利用多头交叉补丁注意力结合LiDAR等互补信息,提升高光谱图像土地覆盖分类的泛化性能,在多个基准数据集上超越现有方法。

摘要:

视觉Transformer(ViTs)由于其在图像分类任务中相比卷积神经网络(CNNs)表现出的优越性能而成为趋势。因此,许多研究者尝试将ViTs应用于高光谱图像(HSI)分类任务。为了达到接近CNNs的令人满意的性能,Transformer需要更少的参数。ViTs和其他类似的Transformer使用一个随机初始化且往往难以很好泛化的外部分类(CLS)标记,而其他多模态数据源,如光探测和测距(LiDAR),则提供了通过CLS改进这些模型的潜力。在本文中,我们提出了一种新的多模态融合Transformer(MFT)网络,其包含用于HSI土地覆盖分类的多头交叉补丁注意力(mCrossPA)。我们的mCrossPA在Transformer编码器中除了利用HSI外,还利用其他互补信息源以实现更好的泛化。采用标记化概念生成CLS和HSI补丁标记,有助于在缩减的分层特征空间中学习独特的表示。我们在广泛使用的基准数据集上进行了大量实验,即休斯顿大学、特伦托、南密西西比大学Gulfpark(MUUFL)和奥格斯堡数据集。我们将所提出的MFT模型的结果与其他最先进的Transformer、经典CNNs和传统分类器模型进行了比较。所提出模型取得的优越性能归因于多头交叉补丁注意力的使用。源代码将在https://github.com/AnkurDeria/MFT公开提供。

英文摘要:

Vision transformers (ViTs) have been trending in image classification tasks due to their promising performance when compared to convolutional neural networks (CNNs). As a result, many researchers have tried to incorporate ViTs in hyperspectral image (HSI) classification tasks. To achieve satisfactory performance, close to that of CNNs, transformers need fewer parameters. ViTs and other similar transformers use an external classification (CLS) token which is randomly initialized and often fails to generalize well, whereas other sources of multimodal datasets, such as light detection and ranging (LiDAR) offer the potential to improve these models by means of a CLS. In this paper, we introduce a new multimodal fusion transformer (MFT) network which comprises a multihead cross patch attention (mCrossPA) for HSI land-cover classification. Our mCrossPA utilizes other sources of complementary information in addition to the HSI in the transformer encoder to achieve better generalization. The concept of tokenization is used to generate CLS and HSI patch tokens, helping to learn a {distinctive representation} in a reduced and hierarchical feature space. Extensive experiments are carried out on {widely used benchmark} datasets {i.e.,} the University of Houston, Trento, University of Southern Mississippi Gulfpark (MUUFL), and Augsburg. We compare the results of the proposed MFT model with other state-of-the-art transformers, classical CNNs, and conventional classifiers models. The superior performance achieved by the proposed model is due to the use of multihead cross patch attention. The source code will be made available publicly at \url{https://github.com/AnkurDeria/MFT}.}

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑