arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2306.03373eess.IVcs.CV

CiT-Net:卷积神经网络与视觉Transformer携手用于医学图像分割

CiT-Net: Convolutional Neural Networks Hand in Hand with Vision Transformers for Medical Image Segmentation

Shaanxi University of Science and Technology(陕西科技大学);Xi’an Jiaotong University(西安交通大学);Northwestern Polytechnical University(西北工业大学);Brunel University London(伦敦布鲁内尔大学)

Tao Lei, Rui Sun, Xuan Wang, Yingbo Wang, Xi He, Asoke Nandi

AI总结:

针对CNN与Transformer混合架构在医学图像分割中自适应特征学习不足和忽略跨维度自注意力的问题,提出CiT-Net,通过动态可变形卷积和移位窗口自适应互补注意力模块,实现更优分割性能且低参数低成本。

摘要:

卷积神经网络(CNN)与Transformer的混合架构在医学图像分割中非常流行。然而,它面临两个挑战。首先,尽管CNN分支可以使用普通卷积捕获局部图像特征,但它无法实现自适应特征学习。其次,尽管Transformer分支可以捕获全局特征,但它忽略了通道和跨维度自注意力,导致在复杂内容图像上的分割精度较低。为了解决这些挑战,我们提出了一种新颖的卷积神经网络与视觉Transformer携手(CiT-Net)的混合架构,用于医学图像分割。我们的网络有两个优点。首先,我们设计了一种动态可变形卷积,并将其应用于CNN分支,这克服了固定大小卷积核和在不同输入之间共享核参数的僵硬设计导致的弱特征提取能力。其次,我们设计了一个移位窗口自适应互补注意力模块和一个紧凑的卷积投影。我们将它们应用于Transformer分支,以学习医学图像的跨维度长期依赖。实验结果表明,我们的CiT-Net比流行的SOTA方法提供了更好的医学图像分割结果。此外,我们的CiT-Net需要更少的参数和更少的计算成本,并且不依赖预训练。代码可在https://github.com/SR0920/CiT-Net公开获取。

英文摘要:

The hybrid architecture of convolutional neural networks (CNNs) and Transformer are very popular for medical image segmentation. However, it suffers from two challenges. First, although a CNNs branch can capture the local image features using vanilla convolution, it cannot achieve adaptive feature learning. Second, although a Transformer branch can capture the global features, it ignores the channel and cross-dimensional self-attention, resulting in a low segmentation accuracy on complex-content images. To address these challenges, we propose a novel hybrid architecture of convolutional neural networks hand in hand with vision Transformers (CiT-Net) for medical image segmentation. Our network has two advantages. First, we design a dynamic deformable convolution and apply it to the CNNs branch, which overcomes the weak feature extraction ability due to fixed-size convolution kernels and the stiff design of sharing kernel parameters among different inputs. Second, we design a shifted-window adaptive complementary attention module and a compact convolutional projection. We apply them to the Transformer branch to learn the cross-dimensional long-term dependency for medical images. Experimental results show that our CiT-Net provides better medical image segmentation results than popular SOTA methods. Besides, our CiT-Net requires lower parameters and less computational costs and does not rely on pre-training. The code is publicly available at https://github.com/SR0920/CiT-Net.

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

↑