利用心腔先验知识提升3D心脏核磁薄壁分割精度
C2W-Tune: Cavity-to -Wall Transfer Learning for Thin Atrial Wall Segmentation in 3D LGE-MRI
- 先分心腔再转分心壁,借已学结构信息指导薄壁识别
- 心壁Dice提升至0.814,表面距离指标改善显著
- 适合数据少或标注难的医学图像分割任务
在3D延迟钆增强磁共振成像(LGE-MRI)中精确分割左心房(LA)壁对壁厚映射和纤维化量化至关重要,但受壁薄、解剖复杂及对比度低影响,分割困难。本文提出C2W-Tune,一种两阶段的心腔到心壁迁移学习框架,利用高精度心腔模型作为解剖先验,提升薄壁分割效果。第一阶段使用带ResNeXt编码器和实例归一化的3D U-Net预训练心腔分割,学习鲁棒的心房表征;第二阶段通过渐进式层解冻策略迁移权重,保留心腔特征的同时实现心壁特异性优化。在2018年左心房分割挑战赛数据集上,与从头训练的同架构基线相比,心壁Dice分数从0.623升至0.814,1mm处表面Dice从0.553增至0.731,95%分位豪斯多夫距离(HD95)由2.95mm降至2.55mm,平均对称表面距离(ASSD)由0.71mm降至0.63mm。在仅用70个样本的弱监督设置下,仍达Dice 0.78,优于多数近期多类双心房基准(通常为0.6–0.7)。结果表明,基于解剖先验的可控微调可有效提升3D LGE-MRI中薄心壁分割的准确性。
原文摘要 · Abstract (English)
Accurate segmentation of the left atrial (LA) wall in 3D late gadolinium-enhanced MRI (LGE-MRI) is essential for wall thickness mapping and fibrosis quantification, yet it remains challenging due to the wall's thin geometry, complex anatomy, and low contrast. We propose C2W-Tune, a two-stage cavity-to-wall transfer framework that leverages a high-accuracy LA cavity model as an anatomical prior to improve thin-wall delineation. Using a 3D U-Net with a ResNeXt encoder and instance normalization, Stage 1 pre-trains the network to segment the LA cavity, learning robust atrial representations. Stage 2 transfers these weights and adapts the network to LA wall segmentation using a progressive layer-unfreezing schedule to preserve cavity features while enabling wall-specific refinement. On the 2018 LA Segmentation Challenge dataset, C2W-Tune outperformed an architecture-matched baseline trained from scratch. The wall Dice score increased from 0.623 to 0.814, surface Dice at 1 mm increased from 0.553 to 0.731, 95th-percentile Hausdorff distance (HD95) decreased from 2.95 mm to 2.55 mm, and average symmetric surface distance (ASSD) decreased from 0.71 mm to 0.63 mm. Under reduced supervision using 70 training volumes sampled from the same training set, C2W-Tune achieved a Dice of 0.78, remaining competitive with recent multi-class bi-atrial benchmarks, typically 0.6-0.7. These results show that anatomically grounded task transfer with controlled fine-tuning improves accuracy for thin LA wall segmentation in 3D LGE-MRI.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。