用元学习实现5次少量标注即可精准分割心脏左房壁
Few-Shot Left Atrial Wall Segmentation in 3D LGE MRI via Meta-Learning
- 基于3D残差U-Net和元学习框架,支持5/10/20样本小样本训练
- 5次标注下分割精度达Dice=0.54,比基线提升0.06
- 适用于标注稀缺的临床场景,尤其适合心脏重构评估
从晚期钆增强磁共振成像(LGE-MRI)中分割左心房(LA)壁极具挑战,因其结构纤细、对比度低且专家标注有限。本文提出一种模型无关的元学习(MAML)框架,采用3D残差U-Net作为主干网络,支持K-shot(K=5,10,20)学习。该框架在LA壁任务基础上,联合辅助的LA腔室与右心房(RA)腔室任务进行元训练,并引入边界感知复合损失以提升细结构分割效果。在独立测试集上,5次标注下MAML的Dice系数(DSC)达0.54,优于基线的0.48;Hausdorff距离(HD95)为4.60毫米,优于基线的6.40毫米。20次标注时,性能接近全监督模型(DSC=0.59 vs 0.61)。在未见的合成域偏移和本地队列数据上,性能有所下降但随K增大持续改善:5次标注下,在合成偏移中表现为DSC=0.52、HD95=5.02毫米,在本地队列中为DSC=0.50、HD95=5.43毫米。结果表明,元学习可有效提升小样本条件下的薄壁分割能力,有望降低心脏重塑评估中的标注负担。
原文摘要 · Abstract (English)
Segmenting the left atrial (LA) wall from late gadolinium enhancement magnetic resonance imaging (LGE-MRI) is challenging because of its thin geometry, low contrast, and limited expert annotations. We propose a model-agnostic meta-learning (MAML) framework with a 3D residual U-Net backbone for K-shot (K = 5, 10, 20) LA wall segmentation. The framework is meta-trained on LA wall tasks together with auxiliary LA and right atrial (RA) cavity tasks and uses a boundary-aware composite loss to improve thin-structure delineation. We evaluated MAML on a held-out clean test set and assessed its robustness under an unseen synthetic domain shift and on a local cohort. On the held-out clean test set, MAML outperformed the K-shot fine-tuning baseline at 5-shot, achieving Dice coefficient (DSC) = 0.54 versus 0.48 and Hausdorff distance (HD95) = 4.60 versus 6.40 mm. At 20-shot, MAML approached the fully supervised model trained from scratch, with DSC = 0.59 versus 0.61. Under unseen shifts, performance decreased relative to clean testing but improved consistently as K increased. At 5-shot, MAML achieved DSC = 0.52 and HD95 = 5.02 mm under the unseen synthetic shift, and DSC = 0.50 and HD95 = 5.43 mm on the local cohort. These results suggest that meta-learning can improve thin-wall delineation in low-shot adaptation and may reduce the annotation burden for atrial remodeling assessment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。