arXiv:2602.10630cs.CV2026-02中稿 · ICLR被引 1

去掉VAE实现快速高分辨率图像修复,1秒处理4K图

Eliminating VAE for Fast and High-Resolution Generative Detail Restoration

论文配图:Eliminating VAE for Fast and High-Resolution Generative Detail Restoration
图 1 · 摘自论文原文
  • 用像素重排替代VAE,直接在像素空间生成图像
  • 相比原方法提速2.8倍,内存减少60%,4K图1秒完成
  • 适合追求高速高分辨率图像修复的开发者和应用者

扩散模型在真实世界超分辨率任务中取得显著进展,但推理速度慢且对设备要求高。现有方法如GenDR通过步骤蒸馏将推理步数压缩至一步,但仍受内存限制,需分块处理高分辨率图像。分析发现,变分自编码器(VAE)是延迟与内存瓶颈。为此,我们采用像素重排操作消除VAE,将基于隐空间的GenDR转化为像素空间的GenDR-Pix。然而x8像素重排可能引发重复模式伪影。为此提出多阶段对抗蒸馏,利用前阶段生成特征指导判别器;引入随机填充增强特征,防止判别器崩溃;并设计掩码傅里叶空间损失,惩罚幅度异常点。为提升性能,采用基于填充的自集成与无分类器引导。实验表明,GenDR-Pix相比GenDR实现2.8倍加速与60%内存节省,视觉损失可忽略,优于其他单步扩散超分方法,可在1秒内完成4K图像恢复,仅需6GB显存。

原文摘要 · Abstract (English)

Diffusion models have attained remarkable breakthroughs in the real-world super-resolution (SR) task, albeit at slow inference and high demand on devices. To accelerate inference, recent works like GenDR adopt step distillation to minimize the step number to one. However, the memory boundary still restricts the maximum processing size, necessitating tile-by-tile restoration of high-resolution images. Through profiling the pipeline, we pinpoint that the variational auto-encoder (VAE) is the bottleneck of latency and memory. To completely solve the problem, we leverage pixel-(un)shuffle operations to eliminate the VAE, reversing the latent-based GenDR to pixel-space GenDR-Pix. However, upscale with x8 pixelshuffle may induce artifacts of repeated patterns. To alleviate the distortion, we propose a multi-stage adversarial distillation to progressively remove the encoder and decoder. Specifically, we utilize generative features from the previous stage models to guide adversarial discrimination. Moreover, we propose random padding to augment generative features and avoid discriminator collapse. We also introduce a masked Fourier space loss to penalize the outliers of amplitude. To improve inference performance, we empirically integrate a padding-based self-ensemble with classifier-free guidance to improve inference scaling. Experimental results show that GenDR-Pix performs 2.8x acceleration and 60% memory-saving compared to GenDR with negligible visual degradation, surpassing other one-step diffusion SR. Against all odds, GenDR-Pix can restore 4K image in only 1 second and 6GB.

图像修复扩散模型超分辨率高效生成

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