摘要
使用监督深度学习方法检测脑部 MRI 扫描中的异常存在挑战,因为解剖结构多样性以及像素级标注的劳动密集型要求。生成模型,如去噪扩散概率模型 (DDPM) 及其变体,如 pDDPM、mDDPM、cDDPM,最近已成为在脑部 MRI 扫描中执行无监督异常检测的强大替代方案。这些方法利用健康大脑的帧级标签来生成脑部 MRI 扫描中的健康组织。在推理过程中,当异常(或不健康)的扫描图像作为输入呈现时,这些模型会生成与输入异常扫描相对应的健康扫描图像,并且生成健康扫描图像与原始异常扫描图像之间的差异图提供了异常组织的必要像素级识别。然而,来自 DDPM、pDDPM 和 mDDPM 模型的生成健康图像存在保真度问题,并包含没有医学意义的伪影。虽然 cDDPM 实现了略高的保真度和伪影抑制,但它需要巨大的内存占用,并且在计算上比其他基于 DDPM 的模型更昂贵。在这项工作中,我们提出了一种改进版的 DDPM,称为多通道条件去噪扩散概率模型 (MCDDPM),用于脑部 MRI 扫描中的无监督异常检测。我们提出的模型通过在训练过程中利用来自健康图像的附加信息来实现高保真度,从而增强了 DDPM 模型的表示能力,并且其计算成本和内存需求与 DDPM、pDDPM 和 mDDPM 模型相当。在多个数据集(例如 BraTS20、BraTS21)上的实验结果证明了该方法的良好性能。代码可在 https://github.com/vivekkumartri/MCDDPM 获取。