arXiv:2412.07904v2 通知类型: replace-cross 摘要: 我们推导了一般变量替换公式，对于一个光滑可逆的变换 $\mathbf{y} = \phi(\mathbf{x})$，我们展示了变换后的分数函数 $\nabla_{\mathbf{y}} \log q(\mathbf{y})$ 可以直接用 $\nabla_{\mathbf{x}} \log p(\mathbf{x})$ 表达。利用这一结果，我们开发了两个应用：首先，我们建立了基于分数扩散模型的反时间伊托引理，这允许在变换空间中逆向一个 SDE，而不直接学习 $\nabla_{\mathbf{y}} \log q_t(\mathbf{y})$。这种方法使得可以在一个空间训练扩散模型但在另一个空间进行采样，有效地解耦了正向和反向过程。其次，我们引入了一般化切片分数匹配，将传统的切片分数匹配从线性投影扩展到任意光滑的变换。这在高维密度估计中提供了更大的灵活性。我们通过扩散在概率单纯形上的应用展示这些理论进步，并且实证地将我们的一般化分数匹配方法与传统切片分数匹配方法进行了比较。