用自适应生成流修复单步超分,让图像更真实
Allo{SR}$^2$: Rectifying One-Step Super-Resolution to Stay Real via Allomorphic Generative Flows

- 通过统计对齐的中间状态初始化,减少分布偏移
- 单步推理下保持高保真度,峰值信噪比超40.2
- 适合追求极致效率与真实感的图像重建场景
真实世界图像超分辨率(Real-SR)已因扩散模型(DMs)和流匹配(FM)的强大生成先验而革新。然而,现有单步方法通常在初始化时用退化低分辨率(LR)潜变量替代高斯噪声,引入显著分布偏移,导致轨迹偏离与先验崩溃。为此,我们提出Allo{SR}$^2$,一种基于流匹配的新框架,通过异构生成流修复单步超分流程以维持高保真生成真实性。具体而言,采用信噪比引导的轨迹初始化,识别预训练路径上的统计对齐中间状态,实现LR表示融入生成流;提出流锚定轨迹一致性(FATC),显式正则化概率流的速度场,确保稳定低曲率路径;设计异构轨迹匹配(ATM),一种自对抗蒸馏策略,统一建模超分流与生成流,使单步推理兼具生成先验。大量合成与真实世界基准实验表明,Allo{SR}$^2$ 在单步Real-SR上达到领先性能,同时在保真度与真实性间取得优异平衡,并保持极高效率。
原文摘要 · Abstract (English)
Real-world image super-resolution (Real-SR) has been revolutionized by leveraging the powerful generative priors from Diffusion Models (DMs) and Flow Matching (FM). However, existing one-step methods typically replace Gaussian noise with degraded low-resolution (LR) latents at initialization, introducing a substantial distribution shift that further leads to trajectory deviation and prior collapse under extreme acceleration. To overcome these limitations, we propose Allo{SR}$^2$, a novel FM-based framework that rectifies one-step SR flows via allomorphic generative flows to maintain high-fidelity generative realism. Specifically, we utilize SNR-Guided Trajectory Initialization to identify a statistically aligned intermediate state along the pre-trained path to integrate LR representations into the generative flow. To ensure a stable, low-curvature path for one-step inference, we propose Flow-Anchored Trajectory Consistency (FATC), which explicitly regularizes the velocity field of the underlying probability flow. Furthermore, we develop Allomorphic Trajectory Matching (ATM), a self-adversarial distillation strategy that jointly models the SR flow and the generative flow within a unified velocity field, enabling one-step Real-SR while preserving the generative prior. Extensive experiments on both synthetic and real-world benchmarks demonstrate that Allo{SR}$^2$ achieves state-of-the-art performance in one-step Real-SR, offering a superior balance between fidelity and realism while maintaining extreme efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。