arXiv:2502.09655v1 类型: cross 摘要: 扩散桥梁已经在配对的图像到图像(I2I)转换任务中显示出了潜力。然而，现有的方法受限于它们的单向性质，需要分别的模型来进行前向和反向转换。这不仅增加了计算成本，还限制了它们的实用性。在本文中，我们介绍了双向扩散桥梁模型(BDBM)，这是一种可扩展的方法，通过使用单个网络在两个联接分布之间实现双向转换。BDBM利用Chapman-Kolmogorov 方程来构建桥梁，能够通过在框架内利用起始和目标时间步的可互换性，建模跨时间步的数据分布变化，从而在前向和后向方向上进行双向转换。值得注意的是，当给定端点的边际分布是高斯分布时，BDBM的双向过渡核具有解析形式，允许使用单个网络进行高效的训练。我们展示了BDBM与现有的桥梁方法（如Doob的h-变换和变分方法）之间的联系，并突出了其优势。在高清I2I转换任务的广泛实验中，证明了BDBM不仅能够实现双向转换而几乎不增加额外成本，而且在现有最佳桥梁模型中表现更优。我们的源代码可在[https://github.com/kvmduc/BDBM||https://github.com/kvmduc/BDBM] 获取。