arXiv Machine Learning By Wenhan Guo, Jinglun Yu, Yaning Wang, Jin U. Kang, Yu Sun

PSI3D: Plug-and-Play 3D Stochastic Inference with Slice-wise Latent Diffusion Prior

Read the original on arXiv Machine Learning →

arXiv:2512. 18367v2 Announce Type: replace-cross Abstract: Diffusion models are highly expressive image priors for Bayesian inverse problems.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Machine Learning.

arXiv AI
Sep 4

A Posterior-Dynamics Framework for Imaging Inverse Problems with Pretrained Diffusion Priors

The paper introduces a Posterior‑Dynamics Framework that leverages pretrained diffusion models as multiscale priors for linear imaging inverse problems such as deblurring, super‑resolution, and inpainting. By constructing a surrogate likelihood centered on the clean image and incorporating diffusion uncertainty, the authors derive continuous posterior dynamics and a tunable Langevin component for adaptive exploration. They prove theoretical guarantees (endpoint consistency, finite‑horizon tracking, weak accuracy) and present the PD‑IMEX sampler, which achieves high‑quality reconstructions with only 100 score evaluations and controllable fidelity‑diversity trade‑offs.

By Zhaoqiang Liu, Tongyao Pang, Ruibing Wang, Yang Zheng
arXiv Machine Learning
Aug 19

Picard Proximal Monte Carlo for Parallel Bayesian Imaging with Score-Based Generative Priors

The paper introduces PiX-MC, a time‑parallel posterior sampling framework that combines proximal Langevin dynamics with Picard iteration for Bayesian imaging inverse problems. By leveraging efficient proximal operators for many imaging likelihoods and exploiting parallelism across discretization nodes, PiX-MC supports multi‑GPU implementation and includes multi‑block and annealed variants to enhance scalability. Experiments on various imaging tasks, including a large‑scale sparse‑view CT problem, show that PiX‑MC can reduce runtime by up to 50× while maintaining reconstruction quality.

By Deliang Wei, Evan Bell, Wenhan Guo, Yifan Chen, Yu Sun
arXiv Computer Vision
Sep 18

FlowSGS: Improving Flow Matching Priors for Inverse Imaging with Stochastic Interpolants

FlowSGS introduces a flow-based posterior sampling method that combines Split Gibbs Sampling (SGS) with Langevin dynamics for the likelihood step and Stochastic Interpolants (SI) for the prior step. By integrating a pretrained flow model into the prior step via SI's reverse-time SDE and a novel timestep correction, FlowSGS reduces the number of network evaluations compared to plug‑and‑play diffusion samplers. Experiments demonstrate state‑of‑the‑art performance on various inverse problems, including the first flow‑based solution to a nonlinear inverse problem (Fourier phase retrieval).

By Tianao Li, Xinhui Qian, Emma Alexander
arXiv Machine Learning
Sep 18

Score-based diffusion models for severely ill-posed problems in diffuse optical tomography

Score-based diffusion models, a recent framework for posterior sampling in Bayesian inverse problems, are applied to diffuse optical tomography (DOT), a highly ill‑posed boundary value problem for recovering tissue absorption and scattering. The authors introduce a mixed score that combines a learned component with a model‑based component, providing a theoretical justification for its local approximation to the true score in the small diffusion‑time regime. Four difference‑imaging approaches are compared—classical model‑based, approximate diffusion, exact posterior sampling (UCoS), and a regularized UCoS—showing that UCoS yields more accurate reconstructions, especially under limited‑view geometry and real experimental data.

By Fabian Schneider, Meghdoot Mozumder, Konstantin Tamarov, Leila Taghizadeh, Tanja Tarvainen, Tapio Helin, Duc-Lam Duong