arXiv:2411.06308eess.IVcs.CV2024-11

用扩散模型检测稀疏视角CT重建中的异常数据,提升可靠性。

Exploring Out-of-distribution Detection for Sparse-view Computed Tomography with Diffusion Models

  • 以扩散模型构建正常重建分布先验,通过反向重建误差判断异常。
  • 在MNIST上验证有效,但噪声FBP输入需依赖测量条件才能准确检测。
  • 引入加权机制增强对高信息量异常的鲁棒性,牺牲部分性能。

近期研究证明扩散模型可作为无监督求解逆成像问题的有效方法。稀疏视角计算机断层扫描(CT)因此受益,实现了不依赖测量参数的更好泛化能力。然而,这带来了潜在的幻觉风险,尤其在处理分布外(OOD)数据时。为确保可靠性,必须研究在临床与工业应用中对CT重建的OOD检测。该需求进一步延伸至使检测器能作为异常检查工具使用。本文探索利用训练有素的扩散模型,捕捉目标分布以进行CT重建,作为分布内先验。基于最新研究,我们使用模型对部分扩散的输入图像进行重建,并通过多种重建误差评估分布外程度。针对稀疏视角CT,需重新定义‘输入’与‘重建误差’概念。本文采用滤波反投影(FBP)重建作为输入,研究不同重建误差定义。在MNIST数据集上的概念验证实验展示了成功与失败案例,揭示将此类OOD检测器集成到CT重建系统中的潜力与局限。研究发现,通过比较测量值与前向投影重建结果,可实现有效的OOD检测,前提是来自噪声FBP输入的重建需依赖测量值。然而,这种依赖有时会使检测器意外地良好重建分布外图像。为此,我们提出一种加权方法,提升对高度信息性分布外测量的鲁棒性,尽管在某些情况下会牺牲性能。

原文摘要 · Abstract (English)

Recent works demonstrate the effectiveness of diffusion models as unsupervised solvers for inverse imaging problems. Sparse-view computed tomography (CT) has greatly benefited from these advancements, achieving improved generalization without reliance on measurement parameters. However, this comes at the cost of potential hallucinations, especially when handling out-of-distribution (OOD) data. To ensure reliability, it is essential to study OOD detection for CT reconstruction across both clinical and industrial applications. This need further extends to enabling the OOD detector to function effectively as an anomaly inspection tool. In this paper, we explore the use of a diffusion model, trained to capture the target distribution for CT reconstruction, as an in-distribution prior. Building on recent research, we employ the model to reconstruct partially diffused input images and assess OOD-ness through multiple reconstruction errors. Adapting this approach for sparse-view CT requires redefining the notions of ``input'' and ``reconstruction error''. Here, we use filtered backprojection (FBP) reconstructions as input and investigate various definitions of reconstruction error. Our proof-of-concept experiments on the MNIST dataset highlight both successes and failures, demonstrating the potential and limitations of integrating such an OOD detector into a CT reconstruction system. Our findings suggest that effective OOD detection can be achieved by comparing measurements with forward-projected reconstructions, provided that reconstructions from noisy FBP inputs are conditioned on the measurements. However, conditioning can sometimes lead the OOD detector to inadvertently reconstruct OOD images well. To counter this, we introduce a weighting approach that improves robustness against highly informative OOD measurements, albeit with a trade-off in performance in certain cases.

CT重建扩散模型异常检测稀疏视角

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