arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.17701eess.IVcs.CVcs.LG

旋转扫描:用于医学图像分割的带三重SSM模块的类UNet Mamba

Rotate to Scan: UNet-like Mamba with Triplet SSM Module for Medical Image Segmentation

Hao Tang, Lianglun Cheng, Guoheng Huang, Zhengguang Tan, Junhao Lu, Kaihong Wu

AI总结:

本文提出Triplet Mamba-UNet,通过残差VSS模块和Triplet SSM在空间与通道维度融合特征,在多个医学图像分割数据集上取得优越性能,并相比VM-UNet减少三分之一参数。

摘要:

图像分割在医学领域的诊断和治疗中占据着至关重要的地位。传统的卷积神经网络(CNNs)和Transformer模型在这一领域已经取得了显著进展,但由于感受野有限或计算复杂度较高,它们仍然面临挑战。近年来,状态空间模型(SSMs),尤其是Mamba及其变体,在视觉领域表现出了卓越的性能。然而,它们的特征提取方法可能不够有效,并且保留了一些冗余结构,从而留下了参数缩减的空间。受先前空间注意力和通道注意力方法的启发,我们提出了Triplet Mamba-UNet。该方法利用残差VSS模块提取密集的上下文特征,同时采用Triplet SSM在空间维度和通道维度上融合特征。我们在ISIC17、ISIC18、CVC-300、CVC-ClinicDB、Kvasir-SEG、CVC-ColonDB和Kvasir-Instrument数据集上进行了实验,证明了我们提出的TM-UNet具有优越的分割性能。此外,与之前的VM-UNet相比,我们的模型实现了参数数量减少三分之一。

英文摘要:

Image segmentation holds a vital position in the realms of diagnosis and treatment within the medical domain. Traditional convolutional neural networks (CNNs) and Transformer models have made significant advancements in this realm, but they still encounter challenges because of limited receptive field or high computing complexity. Recently, State Space Models (SSMs), particularly Mamba and its variants, have demonstrated notable performance in the field of vision. However, their feature extraction methods may not be sufficiently effective and retain some redundant structures, leaving room for parameter reduction. Motivated by previous spatial and channel attention methods, we propose Triplet Mamba-UNet. The method leverages residual VSS Blocks to extract intensive contextual features, while Triplet SSM is employed to fuse features across spatial and channel dimensions. We conducted experiments on ISIC17, ISIC18, CVC-300, CVC-ClinicDB, Kvasir-SEG, CVC-ColonDB, and Kvasir-Instrument datasets, demonstrating the superior segmentation performance of our proposed TM-UNet. Additionally, compared to the previous VM-UNet, our model achieves a one-third reduction in parameters.

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

↑