优化3D CT重建扩散模型的采样步骤分配,提升精度与效率
Tracing the Oracle: Improving Diffusion Timestep Scheduling for 3D CT Reconstruction

- 基于动态规划从参考轨迹中提取最优采样点
- 10步以内采样时重建质量显著优于传统方法
- 可插拔部署,适配主流3D CT重建模型
预训练扩散模型在解决高度病态的3D计算机断层扫描(CT)逆问题上展现出巨大潜力,但推理过程存在显著计算开销。现有均匀时间步调度无法捕捉反向条件扩散随机微分方程的非均匀演化,导致较大截断误差。为此,我们提出Tracing the Oracle(TrO),一种可插拔的改进时间步调度框架。具体地,将少数样本上密集采样的数值积分轨迹视为参考真值(oracle),利用动态规划全局最小化多步近似与真值之间的累积误差,精确将有限采样步骤分配至易受截断误差影响的关键演化阶段。在AAPM数据集上的大量实验表明,结合当前最先进的3D CT重建方法DDS,所提优化时间步在不超过10步的严格预算下,显著提升了重建保真度与计算效率。
原文摘要 · Abstract (English)
Pretrained diffusion models demonstrate impressive potential in solving highly ill-posed 3D computed tomography (CT) inverse problems, while the inference process suffers from significant computational overhead. Furthermore, existing uniform timestep schedules fail to capture the non-uniform evolution of the reverse conditional diffusion stochastic differential equation, thereby introducing substantial truncation errors. To overcome this limitation, we propose Tracing the Oracle (TrO), a plug-and-play framework for improved timestep scheduling. Specifically, we treat densely sampled numerical integration trajectories on a few samples as the reference oracle. The optimized schedule is extracted by leveraging dynamic programming to globally minimize the cumulative error between the few-step approximation and the oracle. This mechanism precisely allocates the limited sampling steps to critical evolution stages that are highly susceptible to truncation errors. Our extensive experiments on the AAPM dataset across multiple 3D CT reconstruction tasks demonstrate that, when combined with the state-of-the-art 3D CT reconstruction method DDS, our optimized timesteps significantly improve reconstruction fidelity and computational efficiency compared to existing heuristic schedules, especially under a strict budget of no more than 10 sampling steps.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。