Latent Refinement via Flow Matching for Training-free Linear Inverse Problem Solving
By: Hossein Askari , Yadan Luo , Hongfu Sun and more
Potential Business Impact:
Makes blurry pictures clear using smart math.
Recent advances in inverse problem solving have increasingly adopted flow priors over diffusion models due to their ability to construct straight probability paths from noise to data, thereby enhancing efficiency in both training and inference. However, current flow-based inverse solvers face two primary limitations: (i) they operate directly in pixel space, which demands heavy computational resources for training and restricts scalability to high-resolution images, and (ii) they employ guidance strategies with prior-agnostic posterior covariances, which can weaken alignment with the generative trajectory and degrade posterior coverage. In this paper, we propose LFlow (Latent Refinement via Flows), a training-free framework for solving linear inverse problems via pretrained latent flow priors. LFlow leverages the efficiency of flow matching to perform ODE sampling in latent space along an optimal path. This latent formulation further allows us to introduce a theoretically grounded posterior covariance, derived from the optimal vector field, enabling effective flow guidance. Experimental results demonstrate that our proposed method outperforms state-of-the-art latent diffusion solvers in reconstruction quality across most tasks. The code will be publicly available at https://github.com/hosseinaskari-cs/LFlow .
Similar Papers
Physics-Constrained Fine-Tuning of Flow-Matching Models for Generation and Inverse Problems
Machine Learning (CS)
Teaches computers to predict science by learning rules.
FlowDPS: Flow-Driven Posterior Sampling for Inverse Problems
CV and Pattern Recognition
Fixes blurry pictures using smart math.
Solving Inverse Problems with FLAIR
CV and Pattern Recognition
Makes blurry pictures sharp and clear.