arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2404.01717cs.CVeess.IV

AddSR:利用对抗扩散蒸馏加速基于扩散的盲超分辨率

AddSR: Accelerating Diffusion-based Blind Super-Resolution with Adversarial Diffusion Distillation

State Key Laboratory for Novel Software Technology, Nanjing University(南京大学计算机软件新技术国家重点实验室);Southwest University(西南大学)

Rui Xie, Chen Zhao, Kai Zhang, Zhenyu Zhang, Jun Zhou, Jian Yang, Ying Tai

AI总结:

本文提出AddSR,通过结合蒸馏与ControlNet、引入预测自精炼策略和时间步自适应ADD,在提升盲超分辨率恢复质量的同时实现比SeeSR快7倍的推理速度。

摘要:

基于稳定扩散的盲超分辨率方法在从低分辨率输入重建具有复杂细节的清晰高分辨率图像方面展现出强大的生成能力。然而,其实际应用往往受到效率低下的制约,这源于需要数千或数百个采样步骤。受高效对抗扩散蒸馏(ADD)的启发,我们设计了AddSR,通过结合蒸馏和ControlNet的思想来解决这一问题。具体而言,我们首先提出了一种基于预测的自精炼策略,以极小的额外时间成本在学生模型输出中提供高频信息。此外,我们改进了训练过程,采用高分辨率图像而非低分辨率图像来约束教师模型,为蒸馏提供更稳健的约束。其次,我们引入了一种时间步自适应ADD,以解决原始ADD引入的感知-失真不平衡问题。大量实验表明,我们的AddSR能够生成更好的恢复结果,同时比先前基于SD的最先进模型速度更快(例如,比SeeSR快7倍)。

英文摘要:

Blind super-resolution methods based on stable diffusion showcase formidable generative capabilities in reconstructing clear high-resolution images with intricate details from low-resolution inputs. However, their practical applicability is often hampered by poor efficiency, stemming from the requirement of thousands or hundreds of sampling steps. Inspired by the efficient adversarial diffusion distillation (ADD), we design~\name~to address this issue by incorporating the ideas of both distillation and ControlNet. Specifically, we first propose a prediction-based self-refinement strategy to provide high-frequency information in the student model output with marginal additional time cost. Furthermore, we refine the training process by employing HR images, rather than LR images, to regulate the teacher model, providing a more robust constraint for distillation. Second, we introduce a timestep-adaptive ADD to address the perception-distortion imbalance problem introduced by original ADD. Extensive experiments demonstrate our~\name~generates better restoration results, while achieving faster speed than previous SD-based state-of-the-art models (e.g., $7$$\times$ faster than SeeSR).

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑