Categorical Schrödinger Bridge Matching

ML Generative Models DM
薛定谔桥(SB)是一个强大的框架,用于解决生成建模任务,例如非配对域转换。大多数与SB相关的研究集中在连续数据空间 \(\mathbb{R}^{D}\) 上,而关于将SB方法应用于离散数据(例如有限空间 \(\mathbb{S}^{D}\) 上的理论和算法问题仍然悬而未决。这类集合 \(\mathbb{S}\) 的显著例子包括现代自编码器中向量量化(VQ)表示的码本、文本中的标记、分子中原子的类别等。在本文中,我们为使用最近引入的迭代马尔可夫拟合(IMF)过程在离散空间中求解SB提供了理论和算法基础。具体而言,我们从理论上证明了离散时间IMF(D-IMF)在离散空间中收敛于SB。这使我们能够开发出一个实用的计算算法来求解SB,我们称之为分类薛定谔桥匹配(CSBM)。我们通过一系列合成数据和图像的VQ表示的实验展示了CSBM的性能。
The Schr\"odinger Bridge (SB) is a powerful framework for solving generative modeling tasks such as unpaired domain translation. Most SB-related research focuses on continuous data space $\mathbb{R}^{D}$ and leaves open theoretical and algorithmic questions about applying SB methods to discrete data, e.g, on finite spaces $\mathbb{S}^{D}$. Notable examples of such sets $\mathbb{S}$ are codebooks of vector-quantized (VQ) representations of modern autoencoders, tokens in texts, categories of atoms in molecules, etc. In this paper, we provide a theoretical and algorithmic foundation for solving SB in discrete spaces using the recently introduced Iterative Markovian Fitting (IMF) procedure. Specifically, we theoretically justify the convergence of discrete-time IMF (D-IMF) to SB in discrete spaces. This enables us to develop a practical computational algorithm for SB which we call Categorical Schr\"odinger Bridge Matching (CSBM). We show the performance of CSBM via a series of experiments with synthetic data and VQ representations of images.
许愿