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

使用边界聚焦的 nnU-Net 从 LGE-MRI 中自动分割左心房和瘢痕

Automatically Segment the Left Atrium and Scars from LGE-MRIs Using a Boundary-focused nnU-Net

Peking University(北京大学);University of Liverpool(利物浦大学);Liverpool Heart & Chest Hospital(利物浦心胸医院)

Yuchen Zhang, Yanda Meng, Yalin Zheng

AI总结:

提出一种边界聚焦 nnU-Net,结合 TopK 损失、边界距离图约束和 UAM,从 LGE-MRI 中自动分割左心房腔并量化瘢痕。

摘要:

心房颤动(AF)是最常见的心律失常。准确分割左心房(LA)和 LA 瘢痕可为预测 AF 的治疗结果提供有价值的信息。本文提出利用晚期钆增强磁共振成像(LGE-MRI)自动分割 LA 腔并量化 LA 瘢痕。我们采用 nnU-Net 作为基线模型,并利用 TopK 损失作为损失函数来挖掘 LA 边界特征的重要性。具体而言,在训练过程中实现了对 LA 边界像素的聚焦,从而提供更准确的边界预测。另一方面,将预测的 LA 边界的距离图变换视为 LA 瘢痕预测的额外输入,为瘢痕位置提供边缘约束。我们进一步设计了一种新颖的不确定性感知模块(UAM),以为高不确定性预测生成更好的结果。在 LAScarQS 2022 数据集上的实验证明了我们的模型在 LA 腔和 LA 瘢痕分割上的优越性能。具体而言,我们在 LA 腔和瘢痕分割上分别取得了 88.98% 和 64.08% 的 Dice 系数。我们将在 https://github.com/level6626/Boundary-focused-nnU-Net 上公开实现代码。

英文摘要:

Atrial fibrillation (AF) is the most common cardiac arrhythmia. Accurate segmentation of the left atrial (LA) and LA scars can provide valuable information to predict treatment outcomes in AF. In this paper, we proposed to automatically segment LA cavity and quantify LA scars with late gadolinium enhancement Magnetic Resonance Imagings (LGE-MRIs). We adopted nnU-Net as the baseline model and exploited the importance of LA boundary characteristics with the TopK loss as the loss function. Specifically, a focus on LA boundary pixels is achieved during training, which provides a more accurate boundary prediction. On the other hand, a distance map transformation of the predicted LA boundary is regarded as an additional input for the LA scar prediction, which provides marginal constraint on scar locations. We further designed a novel uncertainty-aware module (UAM) to produce better results for predictions with high uncertainty. Experiments on the LAScarQS 2022 dataset demonstrated our model's superior performance on the LA cavity and LA scar segmentation. Specifically, we achieved 88.98\% and 64.08\% Dice coefficient for LA cavity and scar segmentation, respectively. We will make our implementation code public available at https://github.com/level6626/Boundary-focused-nnU-Net.

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

↑