Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Efficient One-Step Diffusion Restoration Model with Compact Token Compression and Linear Attention

About

Real-world image super-resolution aims to recover high-quality images from complex and unknown real-world degradations. However, existing generative Real-ISR methods largely inherit the dense latent representations and quadratic-cost global modeling paradigm developed for high-resolution image synthesis, causing computation, memory usage, and inference latency to scale unfavorably with resolution and thus limiting practical deployment. We argue that the key bottleneck lies not in insufficient restoration priors, but in excessive token redundancy and costly token interactions during high-resolution restoration. Motivated by this observation, we revisit Real-ISR from the perspectives of compact latent representation and linear-complexity modeling, and propose SANA-SR, an efficient one-step restoration framework. Specifically, SANA-SR employs a deep compression autoencoder with a 32x compression ratio to drastically reduce latent tokens while preserving restoration-relevant structures and textures. On top of this compact latent space, we introduce a linear-attention DiT with LoRA fine-tuning, enabling efficient high-resolution restoration with linear-complexity token mixing. Extensive experiments on all benchmark datasets demonstrate that SANA-SR achieves highly competitive and often superior quantitative performance against existing methods, while restoring clearer and more realistic textures. Moreover, after pruning, the deployed model runs in 0.019s with 407.95G MACs and 344M parameters, highlighting its strong potential for practical mobile deployment.

Bingtian Qiao, Yue Shi, Yingjie Zhou, Yong Guo, Guangtao Zhai, Jiezhang Cao• 2026

Related benchmarks

TaskDatasetResultRank
Image Super-resolutionRealSR
PSNR25.96
190
Image Super-resolutionDIV2K (val)
LPIPS0.3075
189
Super-ResolutionRealSR (test)
PSNR25.96
92
Real-world Single Image Super-ResolutionDRealSR (test)
PSNR27.99
23
Super-ResolutionDRealSR
Parameters (B)0.344
13
Showing 5 of 5 rows

Other info

Follow for update