基本信息

近年来，去噪扩散模型在为图像生成建模生成语义上有价值的像素级表示方面取得了显着的成功。在这项研究中，我们提出了一种新颖的端到端框架，称为 Diff-UNet，用于医学体积分割。我们的方法将扩散模型集成到标准 U 形架构中，以有效地从输入体积中提取语义信息，从而为医学体积分割提供出色的像素级表示。为了增强扩散模型预测结果的鲁棒性，我们还在推理过程中引入了基于步骤不确定性的融合（SUF）模块，以组合每个步骤的扩散模型的输出。我们在三个数据集上评估我们的方法，包括 MRI 中的多模态脑肿瘤、肝脏肿瘤和多器官 CT 体积，并证明 Diff-UNet 显着优于其他最先进的方法。我们的实验结果也表明了所提出模型的普遍性和有效性。所提出的框架有可能通过更精确地分割解剖结构来促进医疗状况的准确诊断和治疗。 Diff-UNet 的代码可在此 https URL 获取 ...