提出新方法加速3D PET图像重建,减少调参并解决切片不一致问题。
Likelihood-Scheduled Score-Based Generative Modeling for Fully 3D PET Image Reconstruction
- 通过匹配似然函数优化生成模型反向扩散过程,简化超参数。
- 在低计数仿真数据上实现更优的重建质量,且速度更快。
- 首次成功应用于真实3D PET数据,适合医学影像重建研究者。
基于预训练得分生成模型(SGMs)的医学图像重建相比现有先进深度学习方法具有优势,包括对不同扫描仪设置的鲁棒性及更优的图像分布建模能力。尽管已有研究将SGM应用于模拟正电子发射断层扫描(PET)数据,在分布外病灶的对比度恢复方面表现优异,但现有方法存在重建慢、超参数调优繁琐及3D中切片不一致等问题。本文提出一种实用的全3D重建方法,通过将SGM逆向扩散过程的似然与最大似然期望最大化算法的当前迭代匹配,加速重建并减少关键超参数数量。以低计数[$^{18}$F]DPA-714仿真数据为例,本方法在保持或超越现有最先进SGM-PET重建的NRMSE和SSIM指标的同时,显著降低重建时间并减少调参需求。我们还对比了最先进的监督与传统重建算法,并首次实现基于SGM的真正3D PET数据重建,采用垂直预训练的SGMs有效消除切片不一致性问题。
原文摘要 · Abstract (English)
Medical image reconstruction with pre-trained score-based generative models (SGMs) has advantages over other existing state-of-the-art deep-learned reconstruction methods, including improved resilience to different scanner setups and advanced image distribution modeling. SGM-based reconstruction has recently been applied to simulated positron emission tomography (PET) datasets, showing improved contrast recovery for out-of-distribution lesions relative to the state-of-the-art. However, existing methods for SGM-based reconstruction from PET data suffer from slow reconstruction, burdensome hyperparameter tuning and slice inconsistency effects (in 3D). In this work, we propose a practical methodology for fully 3D reconstruction that accelerates reconstruction and reduces the number of critical hyperparameters by matching the likelihood of an SGM's reverse diffusion process to a current iterate of the maximum-likelihood expectation maximization algorithm. Using the example of low-count reconstruction from simulated [$^{18}$F]DPA-714 datasets, we show our methodology can match or improve on the NRMSE and SSIM of existing state-of-the-art SGM-based PET reconstruction while reducing reconstruction time and the need for hyperparameter tuning. We evaluate our methodology against state-of-the-art supervised and conventional reconstruction algorithms. Finally, we demonstrate a first-ever implementation of SGM-based reconstruction for real 3D PET data, specifically [$^{18}$F]DPA-714 data, where we integrate perpendicular pre-trained SGMs to eliminate slice inconsistency issues.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。