Fast Image Super-Resolution via Consistency Rectified Flow
Jiaqi Xu, Wenbo Li, Haoze Sun, Fan Li, Zhixin Wang + 6 more
TLDR
FlowSR achieves fast, high-quality single-step image super-resolution by reformulating the problem as a rectified flow with enhanced consistency.
Key contributions
- FlowSR reformulates image super-resolution as a rectified flow from low-resolution to high-resolution.
- Uses improved consistency learning with HR regularization to ensure precise convergence to ground-truth.
- Implements a fast-slow scheduling strategy for efficiency and capturing fine-grained texture details.
Why it matters
Diffusion models for SR are slow, and current single-step methods compromise quality. FlowSR overcomes this by offering a fast, high-quality single-step solution. Its novel rectified flow approach improves both efficiency and fidelity, making high-resolution image generation more practical.
Original Abstract
Diffusion models (DMs) have demonstrated remarkable success in real-world image super-resolution (SR), yet their reliance on time-consuming multi-step sampling largely hinders their practical applications. While recent efforts have introduced few- or single-step solutions, existing methods either inefficiently model the process from noisy input or fail to fully exploit iterative generative priors, compromising the fidelity and quality of the reconstructed images. To address this issue, we propose FlowSR, a novel approach that reformulates the SR problem as a rectified flow from low-resolution (LR) to high-resolution (HR) images. Our method leverages an improved consistency learning strategy to enable high-quality SR in a single step. Specifically, we refine the original consistency distillation process by incorporating HR regularization, ensuring that the learned SR flow not only enforces self-consistency but also converges precisely to the ground-truth HR target. Furthermore, we introduce a fast-slow scheduling strategy, where adjacent timesteps for consistency learning are sampled from two distinct schedulers: a fast scheduler with fewer timesteps to improve efficiency, and a slow scheduler with more timesteps to capture fine-grained texture details. Extensive experiments demonstrate that FlowSR achieves outstanding performance in both efficiency and image quality.
📬 Weekly AI Paper Digest
Get the top 10 AI/ML arXiv papers from the week — summarized, scored, and delivered to your inbox every Monday.