English

Saving Foundation Flow-Matching Priors for Inverse Problems

Machine Learning 2026-05-13 v3 Computer Vision and Pattern Recognition Image and Video Processing Signal Processing

Abstract

Foundation flow-matching (FM) models promise universal priors for solving inverse problems (IPs); yet today, they trail behind domain-specific and even untrained priors. \emph{How can we unlock their potential?} We introduce FMPlug, a plug-in framework that redefines how foundation FMs are used in IPs. FMPlug combines an instance-guided, time-dependent warm-start strategy with sharp Gaussianity regularization, adding problem-specific guidance while preserving the Gaussian structures. For evaluation, we consider both simple image restoration tasks and scientific IPs with a few similar samples -- where the prohibitive cost of data collection and model training hinders the development of domain-specific generative models. Our superior experimental results confirm the effectiveness of FMPlug. Overall, FMPlug paves the way for making foundation FM models practical, reusable priors for IPs, especially scientific ones with few similar samples. More details are available at https://sun-umn.github.io/xm-plug/ .

Keywords

Cite

@article{arxiv.2511.16520,
  title  = {Saving Foundation Flow-Matching Priors for Inverse Problems},
  author = {Yuxiang Wan and Ryan Devera and Wenjie Zhang and Ju Sun},
  journal= {arXiv preprint arXiv:2511.16520},
  year   = {2026}
}

Comments

Accepted by ICML 2026

R2 v1 2026-07-01T07:47:35.281Z