arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2606.08751cs.CV
收藏

少即是多:通过全局-局部轨迹缩减实现低计数PET去噪的3D扩散模型免训练加速框架

Less Is More: Training-Free Acceleration Framework of 3D Diffusion Models for Low-Count PET Denoising via Global-Local Trajectory Reduction

Northwestern University(西北大学);Hefei University of Technology(合肥工业大学)

Yuhan Liu, Scott M. Leonard, Marlee Crews, Muhannad Fadhel, Jinkui Hao, Tianqi Chen, Ryan J. Avery, Bo Zhou

AI总结:

提出一种免训练的全局-局部跳跃策略,通过噪声一致变换初始化中间步骤和重用U-Net特征,在加速3D扩散模型去噪的同时提升重建质量。

摘要:

PET中的准确定量和摄取测量对于评估疾病进展和支持临床决策至关重要。虽然高计数PET提供了可靠的图像质量,但相关的辐射剂量和长时间采集仍然是重要的临床问题,促使采用低计数协议。基于扩散模型的方法在将低计数PET恢复至接近高计数质量方面显示出巨大潜力,但其迭代采样过程在应用于高分辨率3D PET体积时变得极其昂贵,导致显著的推理延迟,限制了实际临床部署。为了解决这些挑战,我们提出了一种免训练的全局-局部跳跃策略,该策略加速了基于扩散模型的3D PET去噪,同时提高了重建质量。所提出的方法即插即用,可直接应用于预训练扩散模型,无需重新训练或修改架构。具体而言,我们引入了:(i) 全局去噪步骤跳跃策略,通过使用低计数输入的噪声一致变换从中间去噪步骤初始化反向扩散过程,大幅减少所需的去噪步骤数;(ii) 局部特征重用捷径,在相邻去噪步骤间重用缓慢变化的高级U-Net特征,进一步减少每步计算量同时保持图像保真度。我们在来自内部和公共数据集的多种PET示踪剂上评估了所提出的方法,包括18F-FDG PET、68Ga-DOTATATE PET和18F-PSMA PET,结果显示相对于全步骤基线,实现了超过一个数量级的一致加速以及改进或相当的重建性能。盲法读者研究进一步证实了增强的临床信心和感知诊断质量。

英文摘要:

Accurate quantification and uptake measurement in PET are critical for assessing disease progression and supporting clinical decision-making. While high-count PET provides reliable image quality, the associated radiation dose and prolonged acquisition remain significant clinical concerns, motivating the adoption of low-count protocols. Diffusion-model-based methods have demonstrated strong potential for restoring low-count PET to near high-count quality, but their iterative sampling procedure becomes prohibitively expensive when applied to high-resolution 3D PET volumes, introducing substantial inference latency that limits practical clinical deployment. To address these challenges, we propose a training-free Global-Local Skipping Strategy that accelerates diffusion model-based 3D PET denoising while simultaneously improving reconstruction quality. The proposed method is plug-and-play and directly applicable to pre-trained diffusion models without retraining or architectural modification. Specifically, we introduce: (i) a global denoising step skipping strategy that initializes the reverse diffusion process from an intermediate denoising step using a noise-consistent transformation of the low-count input, substantially reducing the number of required denoising steps; and (ii) a local feature reuse shortcut that reuses slowly-varying high-level U-Net features across neighboring denoising steps, further reducing per-step computation while preserving image fidelity. We evaluate the proposed approach on multiple PET tracers from in-house and public datasets, including 18F-FDG PET, 68Ga-DOTATATE PET, and 18F-PSMA PET, demonstrating consistent acceleration of over an order of magnitude alongside improved or comparable reconstruction performance relative to the full-step baseline. Blinded reader studies further confirm enhanced clinical confidence and perceived diagnostic quality.

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

↑