Search papers, labs, and topics across Lattice.
This paper introduces Tracing the Oracle (TrO), a framework that enhances timestep scheduling in diffusion models for 3D CT reconstruction by addressing the inefficiencies of uniform schedules. By treating densely sampled numerical integration trajectories as a reference oracle, the method optimally allocates sampling steps to critical evolution stages, thereby minimizing truncation errors. Experimental results on the AAPM dataset reveal that TrO, when integrated with the DDS reconstruction method, significantly boosts both fidelity and efficiency, particularly when constrained to a maximum of 10 sampling steps.
Optimized timestep scheduling can dramatically enhance 3D CT reconstruction fidelity while slashing computational costs, even under strict sampling constraints.
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.
