arXiv:2602.15022cs.LGcs.AI2026-02被引 4

通过标准化重构扩散模型,提升分子3D生成效率与质量。

Rethinking Diffusion Models with Symmetries through Canonicalization with Applications to Molecular Graph Generation

  • 先将分子映射到标准构型,再在标准空间训练非等变扩散模型。
  • 在GEOM-DRUG数据集上,少步生成效果超越现有基线,计算量相当。
  • 适合需要高效高精度分子结构生成的研究者使用。

化学与科学中的许多生成任务涉及群对称性不变分布(如置换和旋转)。传统方法通过架构约束(如等变去噪器和不变先验)强制对称性。本文提出新视角:先将每个样本映射到轨道代表(标准构型或顺序),在标准切片上训练无约束(非等变)的扩散或流模型,并在生成时随机施加对称变换以恢复不变分布。基于商空间理论,证明了标准化生成模型在正确性、通用性和表达能力上优于不变目标;且标准化可加速训练,降低由群混合引起的扩散得分复杂度和流匹配中的条件方差。我们进一步发现对齐先验与最优传输可与标准化互补,提升训练效率。该框架应用于$S_n \times SE(3)$对称性的分子图生成。通过基于几何谱的标准化和轻量位置编码,标准化扩散在3D分子生成任务中显著优于等变基线,计算量相当甚至更低。新型架构Canon和CanonFlow在挑战性数据集GEOM-DRUG上达到最先进性能,少步生成优势仍显著。

原文摘要 · Abstract (English)

Many generative tasks in chemistry and science involve distributions invariant to group symmetries (e.g., permutation and rotation). A common strategy enforces invariance and equivariance through architectural constraints such as equivariant denoisers and invariant priors. In this paper, we challenge this tradition through the alternative canonicalization perspective: first map each sample to an orbit representative with a canonical pose or order, train an unconstrained (non-equivariant) diffusion or flow model on the canonical slice, and finally recover the invariant distribution by sampling a random symmetry transform at generation time. Building on a formal quotient-space perspective, our work provides a comprehensive theory of canonical diffusion by proving: (i) the correctness, universality and superior expressivity of canonical generative models over invariant targets; (ii) canonicalization accelerates training by removing diffusion score complexity induced by group mixtures and reducing conditional variance in flow matching. We then show that aligned priors and optimal transport act complementarily with canonicalization and further improves training efficiency. We instantiate the framework for molecular graph generation under $S_n \times SE(3)$ symmetries. By leveraging geometric spectra-based canonicalization and mild positional encodings, canonical diffusion significantly outperforms equivariant baselines in 3D molecule generation tasks, with similar or even less computation. Moreover, with a novel architecture Canon, CanonFlow achieves state-of-the-art performance on the challenging GEOM-DRUG dataset, and the advantage remains large in few-step generation.

分子生成扩散模型对称性标准化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。