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

MDNet:用于腹部 CT 器官分割的多解码器网络

MDNet: Multi-Decoder Network for Abdominal CT Organs Segmentation

Northwestern University(西北大学);The University of Texas at Austin(德克萨斯大学奥斯汀分校)
Debesh Jha, Nikhil Kumar Tomar, Koushik Biswas, Gorkem Durak, Matthew Antalek, Zheyuan Zhang, Bin Wang, Md Mostafijur Rahman, Hongyi Pan, Alpay Medetalibeyoglu,… 展开作者

Debesh Jha, Nikhil Kumar Tomar, Koushik Biswas, Gorkem Durak, Matthew Antalek, Zheyuan Zhang, Bin Wang, Md Mostafijur Rahman, Hongyi Pan, Alpay Medetalibeyoglu, Yury Velichko, Daniela Ladner, Amir Borhani, Ulas Bagci

AI总结:

本文提出多解码器网络 MDNet,以 MiT-B2 为编码器,结合多尺度空洞增强、跨解码器特征融合与掩码空间注意力,提升腹部 CT 器官分割精度。

摘要:

从腹部 CT 扫描中准确分割器官,对于诊断、治疗计划和患者监测等临床应用至关重要。为应对器官形状、大小的异质性以及复杂解剖关系所带来的挑战,我们提出了 MDNet,这是一种编码器-解码器网络,采用预训练的 MiT-B2 作为编码器,并使用多个不同的解码器网络。每个解码器网络都通过一个多尺度特征增强空洞模块连接到编码器的不同部分。借助每个解码器,我们迭代地增加网络深度并细化分割掩码,同时通过整合先前解码器的特征图来丰富特征图。为进一步细化特征图,我们还将先前解码器预测的掩码用于当前解码器,以在前景和背景区域之间提供空间注意力。MDNet 有效细化了分割掩码,在 Liver Tumor segmentation(LiTS)和 MSD Spleen 数据集上分别取得了 0.9013 和 0.9169 的高 Dice 相似系数(DSC)。此外,它将 LiTS 数据集的 Hausdorff 距离(HD)降至 3.79,将脾脏分割数据集的 HD 降至 2.26,突显了 MDNet 在捕捉复杂轮廓方面的精确性。此外,与其他基线模型相比,MDNet 具有更强的可解释性和鲁棒性。

英文摘要:

Accurate segmentation of organs from abdominal CT scans is essential for clinical applications such as diagnosis, treatment planning, and patient monitoring. To handle challenges of heterogeneity in organ shapes, sizes, and complex anatomical relationships, we propose a \textbf{\textit{\ac{MDNet}}}, an encoder-decoder network that uses the pre-trained \textit{MiT-B2} as the encoder and multiple different decoder networks. Each decoder network is connected to a different part of the encoder via a multi-scale feature enhancement dilated block. With each decoder, we increase the depth of the network iteratively and refine segmentation masks, enriching feature maps by integrating previous decoders' feature maps. To refine the feature map further, we also utilize the predicted masks from the previous decoder to the current decoder to provide spatial attention across foreground and background regions. MDNet effectively refines the segmentation mask with a high dice similarity coefficient (DSC) of 0.9013 and 0.9169 on the Liver Tumor segmentation (LiTS) and MSD Spleen datasets. Additionally, it reduces Hausdorff distance (HD) to 3.79 for the LiTS dataset and 2.26 for the spleen segmentation dataset, underscoring the precision of MDNet in capturing the complex contours. Moreover, \textit{\ac{MDNet}} is more interpretable and robust compared to the other baseline models.

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

↑