同时实现医学影像分割的多样性和个性化,提升诊断可靠性。
Probabilistic Modeling of Multi-rater Medical Image Segmentation for Diversity and Personalization
- 引入双潜变量建模专家偏好与边界模糊性
- 在NPC和LIDC-IDRI数据集上达新最优性能
- 适合需要多专家视角融合的临床辅助诊断
病灶分割受成像不确定性影响,源于病灶边界不明确及诊断者间差异。以往方法虽处理多标注任务,但要么生成缺乏专家特异性的多样化结果,要么仅复制单一标注者。本文提出ProSeg模型,通过两个潜变量分别建模专家标注偏好与病灶边界模糊性,利用变分推断获得条件概率分布,采样生成兼具多样性和个性化的分割结果。在鼻咽癌(NPC)和肺结节(LIDC-IDRI)数据集上的实验表明,ProSeg达到当前最优性能,有效平衡了多样性与个性化。
原文摘要 · Abstract (English)
Lesion segmentation is inherently influenced by imaging uncertainty, arising from ill-defined lesion boundaries and inter-observer variability in diagnosis. To address this challenge, previous works formulated the multi-rater medical image segmentation task, where multiple experts provide separate annotations for each image. However, existing models are typically constrained to either generate diverse segmentation that lacks expert specificity or to produce personalized outputs that merely replicate individual annotators. We propose \textbf{Pro}babilistic modeling of multi-rater lesion \textbf{Seg}mentation (\textbf{ProSeg}) that simultaneously enables both diversification and personalization. Specifically, we introduce two latent variables to model expert annotation preferences and lesion boundary ambiguity. Their conditional probabilistic distributions are then obtained through variational inference, allowing segmentation outputs to be generated by sampling from these distributions. Extensive experiments on both the nasopharyngeal carcinoma dataset (NPC) and the lung nodule dataset (LIDC-IDRI) demonstrate that our ProSeg achieves a new state-of-the-art performance, providing segmentation results that are both diverse and expert-personalized.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。