用合成病灶训练扩散模型,让健康脑图重建更准、病灶检测更灵敏。
Conditional diffusion models for guided anomaly detection in brain images using fluid-driven anomaly randomization
- 用流体驱动生成逼真假病灶,辅助模型学习健康结构
- 在真实与合成数据上均超越自编码器和扩散模型基线
- 适合罕见病等缺乏病患数据的医学图像异常检测场景
监督学习在脑部MRI病理检测中表现优异,但需大量病患数据,而罕见病场景常难获取。基于重构的无监督异常检测,尤其是扩散模型,因仅需健康图像即可训练,受到关注。然而,现有方法假设模型无法准确重构异常区域,实际中常出现健康组织重建失败或异常区未被有效消除的问题。本文提出一种新型条件扩散模型框架,通过引入合成伪病灶图像进行弱监督训练,提升健康图像重建质量与异常检测能力。合成病灶通过在辅助数据集的真实病灶分割图上应用流体驱动异常随机化生成,确保其真实且解剖合理。我们在合成异常数据集及真实病灶数据(ATLAS数据集)上评估模型性能。实验表明:(i) 模型持续优于变分自编码器及条件/无条件潜在扩散模型;(ii) 在多数数据集上,其表现超过使用配对病/健图像的有监督修复方法。
原文摘要 · Abstract (English)
Supervised machine learning has enabled accurate pathology detection in brain MRI, but requires training data from diseased subjects that may not be readily available in some scenarios, for example, in the case of rare diseases. Reconstruction-based unsupervised anomaly detection, in particular using diffusion models, has gained popularity in the medical field as it allows for training on healthy images alone, eliminating the need for large disease-specific cohorts. These methods assume that a model trained on normal data cannot accurately represent or reconstruct anomalies. However, this assumption often fails with models failing to reconstruct healthy tissue or accurately reconstruct abnormal regions i.e., failing to remove anomalies. In this work, we introduce a novel conditional diffusion model framework for anomaly detection and healthy image reconstruction in brain MRI. Our weakly supervised approach integrates synthetically generated pseudo-pathology images into the modeling process to better guide the reconstruction of healthy images. To generate these pseudo-pathologies, we apply fluid-driven anomaly randomization to augment real pathology segmentation maps from an auxiliary dataset, ensuring that the synthetic anomalies are both realistic and anatomically coherent. We evaluate our model's ability to detect pathology, using both synthetic anomaly datasets and real pathology from the ATLAS dataset. In our extensive experiments, our model: (i) consistently outperforms variational autoencoders, and conditional and unconditional latent diffusion; and (ii) surpasses on most datasets, the performance of supervised inpainting methods with access to paired diseased/healthy images.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。