arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.08461cs.CV
收藏

VistaFormer:用于卫星图像时间序列分割的可扩展视觉Transformer

VistaFormer: Scalable Vision Transformers for Satellite Image Time Series Segmentation

University of Victoria(维多利亚大学)

Ezra MacDonald, Derek Jacoby, Yvonne Coady

AI总结:

本研究提出轻量级遥感图像语义分割模型VistaFormer,采用无位置自注意力与多尺度编解码结构,通过邻域注意力替换多头自注意力提升可扩展性,在两个作物分割基准上以更低运算量和参数实现了更优性能。

摘要:

我们提出了VistaFormer,这是一种基于Transformer的轻量级模型架构,用于遥感图像的语义分割。该模型采用基于Transformer的多尺度编码器,搭配轻量级解码器,对编码器块中捕获的全局和局部注意力进行聚合。VistaFormer使用无位置自注意力层,简化了模型架构,且无需对时间和空间编码进行插值——当训练和测试图像分辨率存在差异时,插值操作会降低模型性能。我们研究了用于过滤云层等噪声输入信号的简单技术,并证明用邻域注意力(Neighbourhood Attention, NA)替代多头自注意力(Multi-Head Self-Attention, MHSA)可提升模型可扩展性。在PASTIS和MTLCC作物类型分割基准上的实验表明,VistaFormer性能优于同类模型:使用MHSA时仅需8%的浮点运算量,使用NA时仅需11%,同时可训练参数也更少。搭载MHSA的VistaFormer在PASTIS基准上的mIoU指标较现有最优水平提升0.1%,在MTLCC基准上提升3%;搭载NA的VistaFormer在MTLCC基准上提升3.7%。

英文摘要:

We introduce VistaFormer, a lightweight Transformer-based model architecture for the semantic segmentation of remote-sensing images. This model uses a multi-scale Transformer-based encoder with a lightweight decoder that aggregates global and local attention captured in the encoder blocks. VistaFormer uses position-free self-attention layers which simplifies the model architecture and removes the need to interpolate temporal and spatial codes, which can reduce model performance when training and testing image resolutions differ. We investigate simple techniques for filtering noisy input signals like clouds and demonstrate that improved model scalability can be achieved by substituting Multi-Head Self-Attention (MHSA) with Neighbourhood Attention (NA). Experiments on the PASTIS and MTLCC crop-type segmentation benchmarks show that VistaFormer achieves better performance than comparable models and requires only 8% of the floating point operations using MHSA and 11% using NA while also using fewer trainable parameters. VistaFormer with MHSA improves on state-of-the-art mIoU scores by 0.1% on the PASTIS benchmark and 3% on the MTLCC benchmark while VistaFormer with NA improves on the MTLCC benchmark by 3.7%.

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

↑