Feb, 2024

结构保持扩散模型

TL;DR最近几年扩散模型已成为主要的分布学习方法。本文介绍了一种称为结构保持扩散过程的扩散模型系列,用于学习具有额外结构特征(例如群对称性)的分布。该研究通过开发理论条件,使扩散转移步骤保持所述对称性,从而使得数据采样轨迹具有等变性。我们以开发一系列不同对称等变的扩散模型来验证这些结果,并能够学习具有固有对称性的分布。通过在合成和真实数据集上进行实证研究,我们验证了所提出的模型符合提出的理论,并在样本均等性方面能够实现比现有方法更好的性能。我们还展示了如何使用所提出的模型,在不事先了解图像方向的情况下实现理论上有保证的等变图像降噪。