用扩散模型捕捉医学图像分割的不确定性,性能领先。
Diffusion Based Ambiguous Image Segmentation
- 设计扩散模型生成分割图,优化噪声调度与预测方式。
- 在LIDC-IDRI数据集上达当前最优,随机裁剪版也表现优异。
- 适合关注医学图像不确定性的研究者与临床应用开发。
医学图像分割常因专家标注差异而存在固有不确定性。捕捉这种不确定性是重要目标,以往工作采用多种生成模型来表征合理专家真值的完整分布。本文探索扩散模型在生成分割中的设计空间,研究噪声调度、预测类型和损失权重的影响。显著发现:结合输入缩放使噪声调度更难可显著提升性能;x-和v-预测优于epsilon-预测,可能因扩散过程位于离散分割域;只要在扩散末期给予足够权重,多种损失权重均能取得相似性能。实验基于LIDC-IDRI肺部病灶数据集,实现当前最优(SOTA)表现。此外,引入一个随机裁剪版本的LIDC-IDRI数据集,更适合评估分割不确定性,模型在此更难设定下仍达到SOTA。
原文摘要 · Abstract (English)
Medical image segmentation often involves inherent uncertainty due to variations in expert annotations. Capturing this uncertainty is an important goal and previous works have used various generative image models for the purpose of representing the full distribution of plausible expert ground truths. In this work, we explore the design space of diffusion models for generative segmentation, investigating the impact of noise schedules, prediction types, and loss weightings. Notably, we find that making the noise schedule harder with input scaling significantly improves performance. We conclude that x- and v-prediction outperform epsilon-prediction, likely because the diffusion process is in the discrete segmentation domain. Many loss weightings achieve similar performance as long as they give enough weight to the end of the diffusion process. We base our experiments on the LIDC-IDRI lung lesion dataset and obtain state-of-the-art (SOTA) performance. Additionally, we introduce a randomly cropped variant of the LIDC-IDRI dataset that is better suited for uncertainty in image segmentation. Our model also achieves SOTA in this harder setting.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。