arXiv:2502.16943cs.CVeess.IV2025-02被引 11

用扩散模型加掩码,无监督定位脑部异常影像。

MAD-AD: Masked Diffusion for Unsupervised Brain Anomaly Detection

  • 在潜空间对随机脑区块加噪,训练时学会识别并还原正常结构。
  • 在测试时通过反向扩散生成正常对应区域,精准定位异常位置。
  • 无需标注数据,适合临床脑部影像异常筛查场景。

无监督脑部影像异常检测对识别损伤和病理至关重要,但因脑结构复杂且异常标注数据稀缺,精确定位仍具挑战。本文提出MAD-AD方法,在扩散模型中引入掩码机制,仅使用正常脑部MRI扫描进行训练。模型在潜空间对随机选取的图像块添加噪声,并通过双重目标学习识别噪声块并恢复原始特征,从而捕捉正常脑结构的细微模式,同时将潜在异常表现为潜空间中的噪声。推理阶段,模型识别出噪声块并利用反向扩散过程生成其正常对应部分。实验表明,该方法在生成准确正常重构和异常定位方面优于现有无监督技术。代码已开源。

原文摘要 · Abstract (English)

Unsupervised anomaly detection in brain images is crucial for identifying injuries and pathologies without access to labels. However, the accurate localization of anomalies in medical images remains challenging due to the inherent complexity and variability of brain structures and the scarcity of annotated abnormal data. To address this challenge, we propose a novel approach that incorporates masking within diffusion models, leveraging their generative capabilities to learn robust representations of normal brain anatomy. During training, our model processes only normal brain MRI scans and performs a forward diffusion process in the latent space that adds noise to the features of randomly-selected patches. Following a dual objective, the model learns to identify which patches are noisy and recover their original features. This strategy ensures that the model captures intricate patterns of normal brain structures while isolating potential anomalies as noise in the latent space. At inference, the model identifies noisy patches corresponding to anomalies and generates a normal counterpart for these patches by applying a reverse diffusion process. Our method surpasses existing unsupervised anomaly detection techniques, demonstrating superior performance in generating accurate normal counterparts and localizing anomalies. The code is available at hhttps://github.com/farzad-bz/MAD-AD.

异常检测扩散模型脑影像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。