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

RS3Mamba:用于遥感图像语义分割的视觉状态空间模型

RS3Mamba: Visual State Space Model for Remote Sensing Images Semantic Segmentation

the Future Network of Intelligence Institute (FNii), the Chinese University of Hong Kong, Shenzhen(香港中文大学(深圳)未来网络智能研究院);School of Information Science and Engineering, Wuhan University of Science and Technology(武汉科技大学信息科学与工程学院)

Xianping Ma, Xiaokang Zhang, Man-On Pun

AI总结:

本文提出双分支网络 RS3Mamba,利用 VSS 辅助分支与 CCM 融合全局和卷积特征,提升遥感图像语义分割性能。

摘要:

遥感图像的语义分割是地球科学研究中的一项基础任务。然而,被广泛使用的卷积神经网络(CNNs)和 Transformers 都存在一些显著缺陷。前者受限于不足的长程建模能力,而后者则受到计算复杂性的拖累。最近,以 Mamba 为代表的新型视觉状态空间(VSS)模型出现了,能够以线性可计算性建模长程关系。在本工作中,我们提出了一种名为遥感图像语义分割 Mamba(RS3Mamba)的新型双分支网络,以将这项创新技术引入遥感任务。具体而言,RS3Mamba 利用 VSS 块构建辅助分支,为基于卷积的主分支提供额外的全局信息。此外,考虑到两个分支的不同特征,我们引入了协同补全模块(CCM),以增强并融合来自双编码器的特征。在两个广泛使用的数据集 ISPRS Vaihingen 和 LoveDA Urban 上的实验结果证明了所提出的 RS3Mamba 的有效性和潜力。据我们所知,这是首个专为遥感图像语义分割设计的视觉 Mamba。源代码将在 https://github.com/sstary/SSRS 提供。

英文摘要:

Semantic segmentation of remote sensing images is a fundamental task in geoscience research. However, there are some significant shortcomings for the widely used convolutional neural networks (CNNs) and Transformers. The former is limited by its insufficient long-range modeling capabilities, while the latter is hampered by its computational complexity. Recently, a novel visual state space (VSS) model represented by Mamba has emerged, capable of modeling long-range relationships with linear computability. In this work, we propose a novel dual-branch network named remote sensing images semantic segmentation Mamba (RS3Mamba) to incorporate this innovative technology into remote sensing tasks. Specifically, RS3Mamba utilizes VSS blocks to construct an auxiliary branch, providing additional global information to convolution-based main branch. Moreover, considering the distinct characteristics of the two branches, we introduce a collaborative completion module (CCM) to enhance and fuse features from the dual-encoder. Experimental results on two widely used datasets, ISPRS Vaihingen and LoveDA Urban, demonstrate the effectiveness and potential of the proposed RS3Mamba. To the best of our knowledge, this is the first vision Mamba specifically designed for remote sensing images semantic segmentation. The source code will be made available at https://github.com/sstary/SSRS.

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

↑