用条件扩散模型修复脑影像,精准识别疾病异常。
CADD: Context aware disease deviations via restoration of brain images using normative conditional diffusion models
- 基于临床信息引导的条件扩散模型,实现健康区域精准修复
- 在三组真实临床数据上达到当前最优异常检测效果
- 适合处理低对比、厚层、运动伪影等复杂脑影像数据
将机器学习应用于医院档案中的真实医疗数据,有望革新脑影像疾病检测。然而,在异质队列中检测病理仍具挑战。范式建模(一种无监督异常检测方法)通过建模“正常”行为,可在个体层面检测与疾病相关的偏离。扩散模型因其能捕捉复杂数据分布并生成高质量图像,成为异常检测的强大工具,其性能依赖于图像修复能力——原始图像与重建图像的差异可揭示潜在异常。但现有方法未融入临床信息,缺乏上下文指导;且常对健康区域修复不佳,导致重建质量差、检测性能受限。本文提出CADD,首个用于3D脑影像的条件扩散范式建模方法。为指导健康区域修复,我们设计一种新型推理补全策略,在移除异常的同时保留个体特异性特征。在三个具有挑战性的数据集(含临床扫描,可能包含低对比度、厚层、运动伪影)上评估,CADD在异质队列中检测神经异常方面达到当前最优性能。
原文摘要 · Abstract (English)
Applying machine learning to real-world medical data, e.g. from hospital archives, has the potential to revolutionize disease detection in brain images. However, detecting pathology in such heterogeneous cohorts is a difficult challenge. Normative modeling, a form of unsupervised anomaly detection, offers a promising approach to studying such cohorts where the ``normal'' behavior is modeled and can be used at subject level to detect deviations relating to disease pathology. Diffusion models have emerged as powerful tools for anomaly detection due to their ability to capture complex data distributions and generate high-quality images. Their performance relies on image restoration; differences between the original and restored images highlight potential abnormalities. However, unlike normative models, these diffusion model approaches do not incorporate clinical information which provides important context to guide the disease detection process. Furthermore, standard approaches often poorly restore healthy regions, resulting in poor reconstructions and suboptimal detection performance. We present CADD, the first conditional diffusion model for normative modeling in 3D images. To guide the healthy restoration process, we propose a novel inference inpainting strategy which balances anomaly removal with retention of subject-specific features. Evaluated on three challenging datasets, including clinical scans, which may have lower contrast, thicker slices, and motion artifacts, CADD achieves state-of-the-art performance in detecting neurological abnormalities in heterogeneous cohorts.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。