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

用于结构裂缝分割的拓扑感知 Mamba

Topology-aware Mamba for Crack Segmentation in Structures

School of Computer Science and Engineering, Jiangsu University of Science and Technology(江苏科技大学计算机科学与工程学院);School of Electronic and Informatics Engineering, Jiangsu University(江苏大学电子与信息工程学院);School of Civil and Environmental Engineering, Oklahoma State University(俄克拉荷马州立大学土木与环境工程学院)

Xin Zuo, Yu Sheng, Jifeng Shen, Yongwei Shan

AI总结:

本文提出拓扑感知 Mamba 模型 CrackMamba,通过 VMambaV2 编码器、Snake Scan 与三分支 SCVSS 模块提升复杂裂缝分割精度,并在多个裂缝及视网膜血管数据集上验证其 SOTA 表现与泛化能力。

摘要:

CrackMamba 是一种基于 Mamba 的模型,旨在为基础设施结构健康监测提供高效且准确的裂缝分割。传统卷积神经网络(CNN)模型受限于感受野有限,而 Vision Transformer(ViT)虽然提升了分割精度,却具有很高的计算开销。CrackMamba 通过采用在 ImageNet-1k 上预训练权重的 VMambaV2 作为编码器,并使用新设计的解码器来提升性能,从而应对这些挑战。为处理裂缝扩展随机且复杂的特性,本文提出 Snake Scan 模块以重塑裂缝特征序列,增强特征提取。此外,还提出三分支 Snake Conv VSS(SCVSS)块,以更有效地聚焦裂缝。实验表明,CrackMamba 在 CrackSeg9k 和 SewerCrack 数据集上达到最先进(SOTA)性能,并在视网膜血管分割数据集 CHASE_DB1 上展现出具有竞争力的性能,凸显其泛化能力。代码公开发布于:https://github.com/shengyu27/CrackMamba。

英文摘要:

CrackMamba, a Mamba-based model, is designed for efficient and accurate crack segmentation for monitoring the structural health of infrastructure. Traditional Convolutional Neural Network (CNN) models struggle with limited receptive fields, and while Vision Transformers (ViT) improve segmentation accuracy, they are computationally intensive. CrackMamba addresses these challenges by utilizing the VMambaV2 with pre-trained ImageNet-1k weights as the encoder and a newly designed decoder for better performance. To handle the random and complex nature of crack development, a Snake Scan module is proposed to reshape crack feature sequences, enhancing feature extraction. Additionally, the three-branch Snake Conv VSS (SCVSS) block is proposed to target cracks more effectively. Experiments show that CrackMamba achieves state-of-the-art (SOTA) performance on the CrackSeg9k and SewerCrack datasets, and demonstrates competitive performance on the retinal vessel segmentation dataset CHASE\underline{~}DB1, highlighting its generalization capability. The code is publicly available at: {https://github.com/shengyu27/CrackMamba.}

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

↑