arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.08159eess.IVcs.CV

PFCM:用于低剂量CT图像去噪的泊松流一致性模型

PFCM: Poisson flow consistency models for low-dose CT image denoising

Stanford University(斯坦福大学);KTH Royal Institute of Technology(瑞典皇家理工学院);Karolinska University Hospital(卡罗林斯卡大学医院);GE HealthCare(GE医疗);Rensselaer Polytechnic Institute(伦斯勒理工学院)

Dennis Hein, Grant Stevens, Adam Wang, Ge Wang

AI总结:

本文提出泊松流一致性模型(PFCM),结合PFGM++稳健性与一致性模型单步采样,通过任务特定采样器实现低剂量CT去噪,在Mayo数据集上取得优异性能,并展示了临床泛化能力。

摘要:

X射线计算机断层扫描(CT)广泛用于医学诊断和治疗计划;然而,对电离辐射暴露的担忧推动了在较低剂量下优化图像质量的努力。本研究引入了泊松流一致性模型(PFCM),这是一个新颖的深度生成模型家族,结合了PFGM++的稳健性与一致性模型的高效单步采样。PFCM通过变量替换和更新的噪声分布将一致性蒸馏推广到PFGM++而推导出来。作为PFGM++的蒸馏版本,PFCM继承了通过超参数$D \in (0,\infty)$在稳健性和刚性之间进行权衡的能力。我们利用这一特性,通过一种“任务特定”采样器将该新颖生成模型适应于低剂量CT图像去噪任务,该采样器通过用低剂量CT图像替换中间状态来“劫持”生成过程。虽然这种“劫持”引入了严重的不匹配——低剂量CT图像的噪声特性与泊松流过程中中间状态的噪声特性不同——但我们表明,PFCM在小$D$下固有的稳健性有效缓解了这一问题。所得采样器在Mayo低剂量CT数据集上实现了优秀的LPIPS、SSIM和PSNR性能。相比之下,基于标准一致性模型的类似采样器在相同条件下被发现明显不那么稳健,突显了我们的新颖框架所提供的可调$D$的重要性。为了突出泛化能力,我们展示了使用更锐利的核并在一系列能量水平下重建的原型光子计数系统临床图像的有效去噪。

英文摘要:

X-ray computed tomography (CT) is widely used for medical diagnosis and treatment planning; however, concerns about ionizing radiation exposure drive efforts to optimize image quality at lower doses. This study introduces Poisson Flow Consistency Models (PFCM), a novel family of deep generative models that combines the robustness of PFGM++ with the efficient single-step sampling of consistency models. PFCM are derived by generalizing consistency distillation to PFGM++ through a change-of-variables and an updated noise distribution. As a distilled version of PFGM++, PFCM inherit the ability to trade off robustness for rigidity via the hyperparameter $D \in (0,\infty)$. A fact that we exploit to adapt this novel generative model for the task of low-dose CT image denoising, via a ``task-specific'' sampler that ``hijacks'' the generative process by replacing an intermediate state with the low-dose CT image. While this ``hijacking'' introduces a severe mismatch -- the noise characteristics of low-dose CT images are different from that of intermediate states in the Poisson flow process -- we show that the inherent robustness of PFCM at small $D$ effectively mitigates this issue. The resulting sampler achieves excellent performance in terms of LPIPS, SSIM, and PSNR on the Mayo low-dose CT dataset. By contrast, an analogous sampler based on standard consistency models is found to be significantly less robust under the same conditions, highlighting the importance of a tunable $D$ afforded by our novel framework. To highlight generalizability, we show effective denoising of clinical images from a prototype photon-counting system reconstructed using a sharper kernel and at a range of energy levels.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑