arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.10491cs.CV
收藏

实现廉价缩放:一种用于更高分辨率适配的自级联扩散模型

Make a Cheap Scaling: A Self-Cascade Diffusion Model for Higher-Resolution Adaptation

Nanyang Technological University(南洋理工大学);Tencent AI Lab(腾讯AI实验室);The Hong Kong University of Science and Technology(香港科技大学);Clemson University(克莱姆森大学)

Lanqing Guo, Yingqing He, Haoxin Chen, Menghan Xia, Xiaodong Cun, Yufei Wang, Siyu Huang, Yong Zhang, Xintao Wang, Qifeng Chen, Ying Shan, Bihan Wen

AI总结:

本文提出自级联扩散模型,利用低分辨率模型知识快速适配高分辨率生成,实现5倍训练加速和仅0.002M调优参数,微调1万步即可高效生成高分辨率图像和视频。

摘要:

扩散模型已被证明在图像和视频生成方面非常有效;然而,由于单尺度训练数据,它们在生成不同尺寸图像时面临物体正确组合的挑战。将大型预训练扩散模型适配到更高分辨率需要大量的计算和优化资源,但实现与低分辨率模型相当的生成能力仍然具有挑战性。本文提出了一种新颖的自级联扩散模型,利用训练良好的低分辨率图像/视频生成模型所获得的知识,能够快速适配到更高分辨率的生成。在此基础上,我们采用枢轴替换策略,通过逐步利用从低分辨率模型获得的可靠语义指导,实现免调优版本。我们进一步提出集成一系列可学习的多尺度上采样器模块,用于调优版本,能够从少量新获取的高分辨率训练数据中高效学习新尺度的结构细节。与完全微调相比,我们的方法实现了5倍的训练加速,且仅需0.002M个调优参数。大量实验表明,我们的方法仅需微调1万步即可快速适配到更高分辨率的图像和视频合成,且几乎不增加推理时间。

英文摘要:

Diffusion models have proven to be highly effective in image and video generation; however, they encounter challenges in the correct composition of objects when generating images of varying sizes due to single-scale training data. Adapting large pre-trained diffusion models to higher resolution demands substantial computational and optimization resources, yet achieving generation capabilities comparable to low-resolution models remains challenging. This paper proposes a novel self-cascade diffusion model that leverages the knowledge gained from a well-trained low-resolution image/video generation model, enabling rapid adaptation to higher-resolution generation. Building on this, we employ the pivot replacement strategy to facilitate a tuning-free version by progressively leveraging reliable semantic guidance derived from the low-resolution model. We further propose to integrate a sequence of learnable multi-scale upsampler modules for a tuning version capable of efficiently learning structural details at a new scale from a small amount of newly acquired high-resolution training data. Compared to full fine-tuning, our approach achieves a $5\times$ training speed-up and requires only 0.002M tuning parameters. Extensive experiments demonstrate that our approach can quickly adapt to higher-resolution image and video synthesis by fine-tuning for just $10k$ steps, with virtually no additional inference time.

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

↑