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

用于条件扩散的球面高斯约束引导

Guidance with Spherical Gaussian Constraint for Conditional Diffusion

ShanghaiTech University(上海科技大学)

Lingxiao Yang, Shutong Ding, Yifan Cai, Jingyi Yu, Jingya Wang, Ye Shi

AI总结:

本文针对条件扩散模型中损失引导导致的流形偏差问题,提出球面高斯约束扩散(DSG),通过约束引导步骤在数据流形内实现更大步长和更高样本质量,并可作为即插即用模块提升现有免训练方法性能。

摘要:

扩散模型的最新进展尝试通过利用可微损失函数进行引导来处理条件生成任务,而无需额外训练。尽管这些方法取得了一定的成功,但它们往往牺牲样本质量,并且需要较小的引导步长,导致采样过程更长。本文揭示了根本问题在于采用损失引导时采样过程中出现的流形偏差。我们通过为损失引导的估计误差建立某个下界,从理论上证明了流形偏差的存在。为了缓解这一问题,受高维高斯分布中的集中现象启发,我们提出了球面高斯约束扩散(DSG)。DSG通过优化有效地将引导步骤约束在中间数据流形内,并允许使用更大的引导步长。此外,我们给出了带球面高斯约束的DSG去噪的闭式解。值得注意的是,DSG可以作为插件模块无缝集成到现有的免训练条件扩散方法中。实现DSG只需添加几行额外代码,几乎不增加计算开销,却能带来显著的性能提升。在各种条件生成任务上的综合实验结果验证了DSG在样本质量和时间效率方面的优越性与适应性。

英文摘要:

Recent advances in diffusion models attempt to handle conditional generative tasks by utilizing a differentiable loss function for guidance without the need for additional training. While these methods achieved certain success, they often compromise on sample quality and require small guidance step sizes, leading to longer sampling processes. This paper reveals that the fundamental issue lies in the manifold deviation during the sampling process when loss guidance is employed. We theoretically show the existence of manifold deviation by establishing a certain lower bound for the estimation error of the loss guidance. To mitigate this problem, we propose Diffusion with Spherical Gaussian constraint (DSG), drawing inspiration from the concentration phenomenon in high-dimensional Gaussian distributions. DSG effectively constrains the guidance step within the intermediate data manifold through optimization and enables the use of larger guidance steps. Furthermore, we present a closed-form solution for DSG denoising with the Spherical Gaussian constraint. Notably, DSG can seamlessly integrate as a plugin module within existing training-free conditional diffusion methods. Implementing DSG merely involves a few lines of additional code with almost no extra computational overhead, yet it leads to significant performance improvements. Comprehensive experimental results in various conditional generation tasks validate the superiority and adaptability of DSG in terms of both sample quality and time efficiency.

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

↑