等变增强的耦合流​SE(3) Equivariant Augmented Coupling Flows 2308

文章预览

SE(3) Equivariant Augmented Coupling Flows SE(3)等变增广耦合流 https://arxiv.org/pdf/2308.10364 摘要 耦合归一化流（Coupling Normalizing Flows）允许快速采样和密度评估，使其成为物理系统概率建模的首选工具。然而，标准的耦合架构无法赋予在原子笛卡尔坐标上操作的流以物理系统的SE(3)和排列不变性。本文提出了一种通过沿附加的增强维度进行坐标分割来保持SE(3)和排列等变性的耦合流。在每一层中，流将原子的位置映射到学习的SE(3)不变基中，在其中应用标准流变换（如单调有理二次样条），然后再返回到原始基。至关重要的是，我们的流保持了快速采样和密度评估，并且可以通过重要性采样生成目标分布期望的无偏估计。在DW4、LJ13和QM9-positional数据集上训练时，我们的流与等变连续归一化流和扩散模型相比具有竞争力，同时允许采样速度快一个数量级以上。此外 ………………………………