arXiv:2604.19680cs.CV2026-04

用修正流统一生成与判别模型,实现快速高质图像修复。

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

论文配图:IR-Flow: Bridging Discriminative and Generative Image Restoration via Rectified Flow
图 1 · 摘自论文原文
  • 基于修正流构建多级退化分布,提升对不同退化程度的适应能力。
  • 仅需几步采样即达优秀效果,定量指标优于主流方法。
  • 适合追求高效且兼顾细节与感知质量的图像修复场景。

在图像修复任务中,单步判别映射常因期望学习而丢失细节,生成模型则存在多步采样效率低和噪声残差耦合问题。为此,本文提出IR-Flow,一种基于修正流的新型修复方法,统一判别与生成范式。首先构建多级数据分布流,增强模型对各类退化水平的学习与适应能力;其次引入累积速度场,学习跨退化等级的传输轨迹,引导中间状态向干净目标收敛;同时设计多步一致性约束,强化轨迹连贯性,提升少步修复性能。实验表明,直接建立退化与干净图像域间的线性传输流,不仅实现快速推理,还增强对分布外退化的适应能力。在去雨、去噪、雨滴去除任务上,IR-Flow以极少采样步骤取得竞争力强的定量结果,提供高效灵活的修复框架,并保持优异的失真-感知平衡。代码已开源。

原文摘要 · Abstract (English)

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.

图像修复修正流生成模型快速推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。