One Diffusion Step to Real-World Super-Resolution via Flow Trajectory Distillation
Jianze Li, Jiezhang Cao, Yong Guo, Wenbo Li, Yulun Zhang
Abstract
Diffusion models (DMs) have significantly advanced the development of real-world image super-resolution (Real-ISR), but the computational cost of multi-step diffusion models limits their application. One-step diffusion models generate high-quality images in a one sampling step, greatly reducing computational overhead and inference latency. However, most existing one-step diffusion methods are constrained by the performance of the teacher model, where poor teacher performance results in image artifacts. To address this limitation, we propose FluxSR, a novel one-step diffusion Real-ISR technique based on flow matching models. We use the state-of-the-art diffusion model FLUX.1-dev as both the teacher model and the base model. First, we introduce Flow Trajectory Distillation (FTD) to distill a multi-step flow matching model into a one-step Real-ISR. Second, to improve image realism and address high-frequency artifact issues in generated images, we propose TV-LPIPS as a perceptual loss and introduce Attention Diversification Loss (ADL) as a regularization term to reduce token similarity in transformer, thereby eliminating high-frequency artifacts. Comprehensive experiments demonstrate that our method outperforms existing one-step diffusion-based Real-ISR methods. The code and model will be released at https://github.com/JianzeLi-114/FluxSR.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers16
- SeedVR2: One-Step Video Restoration via Diffusion Adversarial Post-TrainingJianyi Wang, Shanchuan Lin, Zhijie Lin, Yuxi Ren et al.ICLR 2026 · 51 citations
- DiT4SR: Taming Diffusion Transformer for Real-World Image Super-ResolutionZheng-Peng Duan, Jiawei Zhang, Xin Jin, Ziheng Zhang et al.ICCV 2025 · 18 citations
- Efficient Rectified Flow for Image FusionZirui Wang, Jiayi Zhang, Tianwei Guan, Yuhan Zhou et al.NeurIPS 2025 · 16 citations
- One-Step Diffusion Transformer for Controllable Real-World Image Super-ResolutionYushun Fang, Yuxiang Chen, Shibo Yin, Qiang Hu et al.CVPR 2026 · 9 citations
- PocketSR: The Super-Resolution Expert in Your Pocket MobilesHaoze Sun, Linfeng Jiang, Fan Li, Renjing Pei et al.NeurIPS 2025 · 8 citations
Builds on24
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- High-Resolution Image Synthesis with Latent Diffusion ModelsRobin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser et al.CVPR 2022 · 13,123 citations
- Adding Conditional Control to Text-to-Image Diffusion ModelsLvmin Zhang, Anyi Rao, Maneesh AgrawalaICCV 2023 · 6,759 citations
- SDXL: Improving Latent Diffusion Models for High-Resolution Image SynthesisDustin Podell, Zion English, Kyle Lacey, Andreas Blattmann et al.ICLR 2024 · 4,569 citations
- Scaling Rectified Flow Transformers for High-Resolution Image SynthesisPatrick Esser, Sumith Kulal, Andreas Blattmann, Rahim Entezari et al.ICML 2024 · 3,620 citations
Related papers
- One-Step Flow for Image Super-Resolution with Tunable Fidelity-Realism Trade-offsYuanzhi Zhu, Ruiqing Wang, Shilin Lu, Hanshu Yan et al.ICLR 2026 · 22 citations
- Unleashing the Power of One-Step Diffusion based Image Super-Resolution via a Large-Scale Diffusion DiscriminatorJianze Li, Jiezhang Cao, Zichen Zou, Xiongfei Su et al.NeurIPS 2025 · 18 citations
- Consistency Trajectory Matching for One-Step Generative Super-ResolutionWeiyi You, Mingyang Zhang, Leheng Zhang, Xingyu Zhou et al.ICCV 2025 · 5 citations
- GenDR: Lighten Generative Detail RestorationYan Wang, Shijie Zhao, Kexin Zhang, Junlin Li et al.ICLR 2026 · 5 citations
- One-Step Residual Shifting Diffusion for Image Super-Resolution via DistillationDaniil Selikhanovych, David Li, Aleksei Leonov, Nikita Gushchin et al.ICML 2026 · 3 citations
