arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.04495cond-mat.stat-mechcs.LGstat.ML
收藏

扩散模型的速度-精度关系:来自非平衡热力学与最优输运的启示

Speed-accuracy relations for diffusion models: Wisdom from nonequilibrium thermodynamics and optimal transport

Department of Mathematical Engineering and Information Physics, School of Engineering, The University of Tokyo(数学工程与信息物理系,工学院,东京大学);Department of Earth and Planetary Physics, School of Science, The University of Tokyo(地球与行星物理系,理学院,东京大学);Preferred Networks Inc.(Preferred Networks公司);Universal Biology Institute, Graduate School of Science, The University of Tokyo(生物科学研究所,研究生院,东京大学)

Kotaro Ikeda, Tomoya Uda, Daisuke Okanohara, Sosuke Ito

AI总结:

本研究借助随机热力学技术推导扩散模型的速度-精度关系,结合最优输运理论的2-瓦瑟斯坦测地线提出最优学习方案,并通过数值实验验证了关系的有效性与方案的实际应用价值。

摘要:

我们探讨了名为扩散模型的生成模型与针对福克-普朗克方程的非平衡热力学(即随机热力学)之间的关联。借助随机热力学的技术,我们推导了扩散模型的速度-精度关系,这是一组将数据生成精度与熵产生率关联起来的不等式。该关系可被解释为无耗散力时扩散动力学的速度。从随机热力学的视角来看,我们的结果为如何在扩散模型中最优地生成数据提供了定量洞见。最优学习方案由最优输运理论中2-瓦瑟斯坦距离空间的测地线定义。我们通过数值实验验证了速度-精度关系在不同噪声调度和不同数据下的有效性,还数值分析了最优与次优学习方案的结果,并证明了该结果可应用于真实图像数据集的数据生成。

英文摘要:

We discuss a connection between a generative model, called the diffusion model, and nonequilibrium thermodynamics for the Fokker-Planck equation, called stochastic thermodynamics. Using techniques from stochastic thermodynamics, we derive the speed-accuracy relations for diffusion models, which are inequalities that relate the accuracy of data generation to the entropy production rate. This relation can be interpreted as the speed of the diffusion dynamics in the absence of the non-conservative force. From a stochastic thermodynamic perspective, our results provide quantitative insight into how best to generate data in diffusion models. The optimal learning protocol is introduced by the geodesic of space of the 2-Wasserstein distance in optimal transport theory. We numerically illustrate the validity of the speed-accuracy relations for diffusion models with different noise schedules and different data. We numerically discuss our results for optimal and suboptimal learning protocols. We also demonstrate the applicability of our results to data generation from the real-world image datasets.

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

↑