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

TBConvL-Net:一种用于鲁棒医学图像分割的混合深度学习架构

TBConvL-Net: A Hybrid Deep Learning Architecture for Robust Medical Image Segmentation

COMSATS University Islamabad (CUI)(COMSATS大学伊斯兰堡校区);University of New South Wales(新南威尔士大学);University of Engineering and Technology (UET) Lahore(拉合尔工程技术大学)

Shahzaib Iqbal, Tariq M. Khan, Syed S. Naqvi, Asim Naveed, Erik Meijering

AI总结:

本文提出混合架构 TBConvL-Net,融合 CNN、双卷积 LSTM 与 ViT 建模局部、长程及时序上下文,并采用复合损失提升边界与鲁棒性,在十种公开数据集上优于现有方法。

摘要:

深度学习在自动化医学图像分割方面展现出巨大潜力,可提高疾病诊断的精度和速度。然而,由于病变在尺度、形状、纹理和对比度方面存在差异,该任务面临显著困难。传统卷积神经网络(CNN)模型在有效建模多尺度上下文信息以及促进跨层级跳跃连接之间的信息交互方面存在一定局限。为克服这些局限,本文提出一种新颖的医学图像分割深度学习架构,充分利用 CNN 和 vision transformer 的优势。我们提出的模型名为 TBConvL-Net,是一种混合网络,将 CNN 编码器-解码器架构的局部特征与使用双卷积长短期记忆(LSTM)网络和 vision transformer(ViT)获得的长程依赖和时间依赖相结合。这使模型能够捕获数据中的上下文通道关系,并考虑随时间变化的分割不确定性。此外,我们提出一种新颖的复合损失函数,同时考虑分割鲁棒性以及预测输出与金标准之间的边界一致性。在涵盖七种不同医学成像模态的十个公开数据集上,所提模型相较最先进方法表现出稳定提升。

英文摘要:

Deep learning has shown great potential for automated medical image segmentation to improve the precision and speed of disease diagnostics. However, the task presents significant difficulties due to variations in the scale, shape, texture, and contrast of the pathologies. Traditional convolutional neural network (CNN) models have certain limitations when it comes to effectively modelling multiscale context information and facilitating information interaction between skip connections across levels. To overcome these limitations, a novel deep learning architecture is introduced for medical image segmentation, taking advantage of CNNs and vision transformers. Our proposed model, named TBConvL-Net, involves a hybrid network that combines the local features of a CNN encoder-decoder architecture with long-range and temporal dependencies using biconvolutional long-short-term memory (LSTM) networks and vision transformers (ViT). This enables the model to capture contextual channel relationships in the data and account for the uncertainty of segmentation over time. Additionally, we introduce a novel composite loss function that considers both the segmentation robustness and the boundary agreement of the predicted output with the gold standard. Our proposed model shows consistent improvement over the state of the art on ten publicly available datasets of seven different medical imaging modalities.

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

↑