arXiv:2606.16188cs.CV2026-06中稿 · ECCV

无需额外模型,一统多步采样,高效实现真实图像超分

TEASR: Training-Efficient Any-Step Diffusion Transformer for Real-World Image Super-Resolution

论文配图:TEASR: Training-Efficient Any-Step Diffusion Transformer for Real-World Image Super-Resolution
图 1 · 摘自论文原文
  • 自对抗蒸馏+时间步感知修正,单模型实现任意步采样
  • 200亿参数模型单卡训练,推理速度提升10倍以上
  • 适合追求高效率与灵活性的图像重建研究者

扩散模型在真实世界图像超分辨率(Real-ISR)中表现优异,因其强大的生成先验,但存在采样速度慢的问题。现有一步蒸馏方法虽加速推理,但通常需辅助教师模型,增加训练内存开销,且难以扩展至大规模架构。此外,这些固定步数模型缺乏速度与质量之间的灵活权衡。本文提出TEASR,一种训练高效的任意步扩散框架,可在统一模型中实现一步与多步恢复。核心思想是在单个扩散模型内进行自对抗蒸馏,无需额外教师或判别器。具体提出时间步感知修正策略,稳定不同噪声水平下的一步生成。两项设计使200亿参数扩散模型可在单张GPU上训练,显著提升训练效率。同时引入双分支扩散变压器,解耦时间步条件以分离当前噪声状态与去噪目标,提升采样质量。大量实验表明,TEASR支持无缝任意步采样,在多个数据集上持续优于当前最优方法。

原文摘要 · Abstract (English)

Diffusion models excel in Real-World Image Super-Resolution (Real-ISR) due to their powerful generative priors but suffer from slow iterative sampling. Although existing one-step distillation methods accelerate inference, they typically require auxiliary teacher models that inflate training memory and restrict scalability to large-scale architectures. Furthermore, these fixed-step models lack the flexibility to trade off speed for quality. In this paper, we propose TEASR, a training-efficient any-step diffusion framework for Real-ISR that enables both one-step and multi-step restoration within a unified model. Our key idea is to perform self-adversarial distillation within a single diffusion model, eliminating the need for auxiliary teachers or discriminators. Specifically, we propose a timestep-aware rectification strategy that stabilizes one-step generation across noise levels. These two designs further enables the distillation of 20B-parameter diffusion models on a single GPU, significantly improving training efficiency. Moreover, we introduce a dual-branch diffusion transformer with decoupled timestep condition to separate the current noise state and the denoising target to enhance sampling quality. Extensive experiments demonstrate that TEASR supports seamless any-step sampling and consistently outperforms state-of-the-art methods across multiple datasets.

图像超分扩散模型高效训练任意步采样

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