Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces flow map denoisers that navigate the distortion-perception (DP) frontier for image restoration, addressing the tradeoff between minimizing error and maximizing perceptual quality. By employing a lookahead parameter, the method allows for continuous adjustment between mean squared error (MMSE) and perceptual regimes, effectively spanning the DP frontier without the need for paired data or extensive hyperparameter tuning. Experimental results demonstrate that this approach matches or surpasses specialized baselines across various inverse problems, including tasks with both Gaussian targets and natural images.
Flow map denoisers can seamlessly traverse the distortion-perception frontier, achieving optimal image restoration without the need for complex tuning or paired data.
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.