论文标题
使用扩散schrödinger桥的有条件模拟
Conditional Simulation Using Diffusion Schrödinger Bridges
论文作者
论文摘要
去核扩散模型最近已成为强大的生成模型类别。它们提供最新的结果,不仅用于无条件的模拟,而且还提供了解决在各种反问题中产生的条件模拟问题时。这些模型的一个局限性在于它们在生成时间是计算密集的,因为它们需要长期模拟扩散过程。进行无条件的模拟时,生成建模的Schrödinger桥式公式会导致理论上接地的算法缩短生成时间,这与其他提出的加速技术互补。我们将Schrödinger桥框架扩展到条件模拟。我们在各种应用程序上演示了这种新颖的方法,包括图像超分辨率,状态空间模型的最佳过滤以及预先训练的网络的完善。我们的代码可以在https://github.com/vdeborto/cdsb上找到。
Denoising diffusion models have recently emerged as a powerful class of generative models. They provide state-of-the-art results, not only for unconditional simulation, but also when used to solve conditional simulation problems arising in a wide range of inverse problems. A limitation of these models is that they are computationally intensive at generation time as they require simulating a diffusion process over a long time horizon. When performing unconditional simulation, a Schrödinger bridge formulation of generative modeling leads to a theoretically grounded algorithm shortening generation time which is complementary to other proposed acceleration techniques. We extend the Schrödinger bridge framework to conditional simulation. We demonstrate this novel methodology on various applications including image super-resolution, optimal filtering for state-space models and the refinement of pre-trained networks. Our code can be found at https://github.com/vdeborto/cdsb.