IR-Flow: Bridging Discriminative and Generative Image Restoration via Rectified Flow

• URL: http://arxiv.org/abs/2604.19680v1
• Date: Tue, 21 Apr 2026 16:58:14 GMT
• Title: IR-Flow: Bridging Discriminative and Generative Image Restoration via Rectified Flow
• Abstract summary: We propose IR-Flow, a novel image restoration method based on Rectified Flow.<n>We first construct multilevel data distribution flows, which expand the ability of models to learn from and adapt to various levels of degradation.<n>We show that directly establishing a linear transport flow between degraded and clean image domains not only enables fast inference but also improves adaptability to out-of-distribution degradations.
• Score: 46.532530550015736
• License: http://creativecommons.org/licenses/by/4.0/
• Abstract: In image restoration, single-step discriminative mappings often lack fine details via expectation learning, whereas generative paradigms suffer from inefficient multi-step sampling and noise-residual coupling. To address this dilemma, we propose IR-Flow, a novel image restoration method based on Rectified Flow that serves as a unified framework bridging the gap between discriminative and generative paradigms. Specifically, we first construct multilevel data distribution flows, which expand the ability of models to learn from and adapt to various levels of degradation. Subsequently, cumulative velocity fields are proposed to learn transport trajectories across varying degradation levels, guiding intermediate states toward the clean target, while a multi-step consistency constraint is presented to enforce trajectory coherence and boost few-step restoration performance. We show that directly establishing a linear transport flow between degraded and clean image domains not only enables fast inference but also improves adaptability to out-of-distribution degradations. Extensive evaluations on deraining, denoising and raindrop removal tasks demonstrate that IR-Flow achieves competitive quantitative results with only a few sampling steps, offering an efficient and flexible framework that maintains an excellent distortion-perception balance. Our code is available at https://github.com/fanzh03/IR-Flow.

Related papers

• Flow Straight to Reality: Perceptually Consistent Flow Matching for Efficient Image Restoration [3.901664292361698]
  PCFlow is a unified framework that parameterizes a continuous transport from degraded observations to clean targets.<n>It achieves competitive performance across diverse restoration tasks at a fraction of traditional computational costs.
  arXiv  Detail & Related papers  (2026-08-11T06:37:10Z)
• ScaleResfusion: Residual Rectified Flow based on Residual Vector Field [24.544037828084154]
  Real-world Image Restoration (Real-IR) aims to recover high-quality (HQ) images from complex and unknown degradations.<n>Recent diffusion-based methods have substantially improved perceptual quality, but their current designs leave two key challenges unresolved.<n>We present ScaleResfusion, a scalable diffusion framework for real-world image restoration built on pre-trained text-to-image rectified-flow models.
  arXiv  Detail & Related papers  (2026-07-28T04:26:05Z)
• Perceptual Flow Matching for Few-Step Generative Modeling [64.26537634881244]
  Perceptual Flow Matching is a framework for few-step generation in flow-matching models.<n>We show that PFM produces high-quality few-step generators when supervised in an appropriate representation space.
  arXiv  Detail & Related papers  (2026-07-03T17:55:01Z)
• One-Step Face Restoration via Shortcut-Enhanced Coupling Flow [2.8265172105754104]
  We propose Shortcut-enhanced Coupling flow for Face Restoration (SCFlowFR)<n>It explicitly models the LQ--HQ dependency, minimizing path crossovers and promoting near-linear transport.<n> Experiments demonstrate that SCFlowFR achieves state-of-the-art one-step face restoration quality with inference speed comparable to traditional non-diffusion methods.
  arXiv  Detail & Related papers  (2026-03-04T02:11:15Z)
• Trajectory Stitching for Solving Inverse Problems with Flow-Based Models [68.36374645801901]
  Flow-based generative models have emerged as powerful priors for solving inverse problems.<n>We propose MS-Flow, which represents the trajectory as a sequence of intermediate latent states rather than a single initial code.<n>We demonstrate the effectiveness of MS-Flow over existing methods on image recovery and inverse problems, including inpainting, super-resolution, and computed tomography.
  arXiv  Detail & Related papers  (2026-02-09T11:36:41Z)
• Restora-Flow: Mask-Guided Image Restoration with Flow Matching [0.39141750421215127]
  Flow matching has emerged as a promising generative approach that addresses the lengthy sampling times associated with state-of-the-art diffusion models.<n>We introduce Restora-Flow, a training-free method that guides flow matching sampling by a degradation mask.<n>We show superior perceptual quality and processing time compared to diffusion and flow matching-based reference methods.
  arXiv  Detail & Related papers  (2025-11-25T10:22:26Z)
• OSCAR: Orthogonal Stochastic Control for Alignment-Respecting Diversity in Flow Matching [14.664226708184676]
  Flow-based text-to-image models follow deterministic trajectories, forcing users to repeatedly sample to discover diverse modes.<n>We present a training-free, inference-time control mechanism that makes the flow itself diversity-aware.
  arXiv  Detail & Related papers  (2025-10-10T07:07:19Z)
• FlowIE: Efficient Image Enhancement via Rectified Flow [71.6345505427213]
  FlowIE is a flow-based framework that estimates straight-line paths from an elementary distribution to high-quality images. Our contributions are rigorously validated through comprehensive experiments on synthetic and real-world datasets.
  arXiv  Detail & Related papers  (2024-06-01T17:29:29Z)
• Deep Equilibrium Diffusion Restoration with Parallel Sampling [120.15039525209106]
  Diffusion model-based image restoration (IR) aims to use diffusion models to recover high-quality (HQ) images from degraded images, achieving promising performance. Most existing methods need long serial sampling chains to restore HQ images step-by-step, resulting in expensive sampling time and high computation costs. In this work, we aim to rethink the diffusion model-based IR models through a different perspective, i.e., a deep equilibrium (DEQ) fixed point system, called DeqIR.
  arXiv  Detail & Related papers  (2023-11-20T08:27:56Z)
• DeFlow: Learning Complex Image Degradations from Unpaired Data with Conditional Flows [145.83812019515818]
  We propose DeFlow, a method for learning image degradations from unpaired data. We model the degradation process in the latent space of a shared flow-decoder network. We validate our DeFlow formulation on the task of joint image restoration and super-resolution.
  arXiv  Detail & Related papers  (2021-01-14T18:58:01Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.