arXiv:2405.16381v2 通告类型: replace-cross 摘要：在流形上生成数据是一个重要的任务，通常需要在平坦空间中扩散模型进行非平凡的适应。本文展示了如何通过一种称为`平凡化`的技术，将扩散模型在欧几里得空间中的有效性转移至李群中。特别是，算法上引入了一个辅助动量变量，以帮助在数据分布与固定、易于采样的分布之间运输位置变量。通常，这会因为动量存在于随位置变化的空间中而增加额外的难度。然而，我们的平凡化技术创建了一个新的动量变量，该变量保持在一个简单的固定向量空间中。这种设计，结合一个保流形的积分器，简化了实施过程并避免了投影到切空间和流形等近似方法所创建的不准确，从而在保真度和效率方面促进了生成。该方法在蛋白质和RNA扭转角生成以及复杂的环面上数据集上取得了当前最先进的性能。我们还首次尝试生成高维特殊正交群和单性群上的数据，后者对于量子问题至关重要。相关代码可在https://github.com/yuchen-zhu-zyc/TDM 获取。