arXiv:cs.LG· Nicolas Zilberstein, Morteza Mardani, Santiago Segarra·· 5 小时前AI 评分40
Flow Map Denoisers:用单参数遍历失真-感知平面求解逆问题
Flow Map Denoisers: Traversing the Distortion-Perception Plane for Inverse Problems
AI 导读
Flow Map Denoisers 提出利用 flow map 模型隐式定义的单参数去噪器族,通过 lookahead 参数 t 在 MMSE 与感知质量之间连续调节,单个训练模型即可覆盖失真-感知(DP)前沿。
正文
Abstract:Image restoration faces a fundamental tradeoff: methods that minimize error produce blurry reconstructions, while those that maximize perceptual quality yield sharp but less faithful images. Existing approaches either commit to a single operating point on this distortion perception (DP) frontier or require paired-data supervision, auxiliary models, or hyperparameter tuning of the sampler to access different points. We show that flow map models, a recent extension of flow matching for few-step sampling that learns an average field, implicitly define a one-parameter family of denoisers that continuously spans the DP frontier. The lookahead parameter t acts as a control knob between the MMSE and perceptual regimes. For Gaussian targets, we prove that varying t exactly recovers the optimal DP frontier; for natural images, we observe similar behavior empirically. Within a Plug-and-Play solver, the same mechanism extends to general inverse problems, where it controls a tradeoff between perceptual alignment and data consistency. Despite the lack of exact optimality guarantees in this setting, a single trained flow map spans the DP tradeoff, matching or exceeding specialized baselines at both extremes. Extensive experiments on CelebA ($128\times 128$) and AFHQ ($256\times 256$) across several linear and nonlinear inverse tasks validate our findings.
| Subjects: | Machine Learning (cs.LG); Computer Vision and Pattern Recognition (cs.CV) |
| Cite as: | arXiv:2606.19802 [cs.LG] |
| (or arXiv:2606.19802v2 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2606.19802 arXiv-issued DOI via DataCite |
Submission history
From: Nicolas Zilberstein [view email]
[v1]
Thu, 18 Jun 2026 05:15:43 UTC (108,460 KB)
[v2]
Thu, 1 Oct 2026 18:02:56 UTC (108,610 KB)
来源:arXiv:cs.LG · arxiv.org