Search papers, labs, and topics across Lattice.
This paper introduces PiX-MC, a time-parallel posterior sampling framework for Bayesian imaging that leverages proximal Langevin dynamics and Picard iteration to enhance computational efficiency. By utilizing efficient proximal operators and enabling parallelism across discretization nodes, the method significantly accelerates sampling processes in large-scale imaging applications. Experimental results show that PiX-MC achieves up to a 50x speedup in runtime for sparse-view computed tomography while maintaining high reconstruction quality, demonstrating its practical viability for complex imaging tasks.
Achieving a 50x speedup in Bayesian imaging sampling without sacrificing quality could revolutionize real-time imaging applications.
Bayesian imaging inverse problems often require sampling from high-dimensional posterior distributions. While recent score-based and diffusion models provide expressive Bayesian priors, their sampling procedures remain inherently sequential and computationally expensive for large-scale imaging applications. We propose PiX-MC, a time-parallel posterior sampling framework based on proximal Langevin dynamics and Picard iteration. The proximal-likelihood formulation exploits the fact that many imaging likelihoods admit efficient, problem-specific proximal operators, while Picard refinement exposes parallelism across discretization nodes and naturally supports multi-GPU implementation. To further improve practical scalability and sampling performance, we develop multi-block and annealed variants of the proposed framework. We establish convergence guarantees under transparent assumptions, accommodating non-log-concave posteriors, imperfect learned score models, multi-block implementations, and annealing schedules. Experiments on a diverse collection of imaging inverse problems demonstrate that PiX-MC substantially reduces wall-clock time while preserving reconstruction quality. On a $512\times512\times80$ sparse-view computed tomography (CT) problem, annealed multi-block PiX-MC achieves up to a $50\times$ runtime speedup over the standard Langevin sampler using eight GPUs.
