arXiv:2605.25577cs.LGcs.AI2026-05被引 1

基于流形分解的分子构象生成方法,提升结构精度与采样效率。

Geometric Flow Matching for Molecular Conformation Generation via Manifold Decomposition

论文配图:Geometric Flow Matching for Molecular Conformation Generation via Manifold Decomposition
图 1 · 摘自论文原文
  • 将分子生成分解为平移、旋转和构象三类几何空间,引入物理先验
  • 在GEOM-Drugs和GEOM-QM9上达到当前最优生成质量,50步即可高保真采样
  • 适合需要高效生成合理分子3D结构的研究者,如药物设计与材料模拟

准确生成分子三维构象是计算化学与药物发现中的关键挑战。尽管扩散与流匹配模型已取得显著进展,但其数学形式与分子物理现实存在根本性错配。现有方法多将分子视为笛卡尔空间中的无结构点云,忽略了键长、键角刚性而扭转角主导柔性这一内在层次力学特性。这种缺乏流形意识的设计迫使模型从零重新学习基本几何约束,常产生物理解释性差的中间结构。为此,我们提出GO-Flow,通过流形分解实现生成建模与分子几何的一致性。不强制在欧氏空间中运动,而是将生成过程分解为:平移空间(线性最优传输)、旋转空间(SO(3)上的测地线流)和构象空间(熵正则最优传输)。该分解注入几何归纳偏置,使生成路径更契合分子自由度。结合等变神经架构后,可实现旋转一致性生成并提升几何有效性。在GEOM-Drugs和GEOM-QM9上的大量实验表明,GO-Flow达到当前最优生成质量。尤其通过在正确流形上学习更直的概率路径,本方法仅需50步即可实现高保真采样,有效弥合了结构精度与计算效率之间的差距。

原文摘要 · Abstract (English)

The generation of accurate 3D molecular conformations is a pivotal challenge in computational chemistry and drug discovery. Recently, diffusion and flow matching models have achieved remarkable success. However, there is a critical misalignment between their mathematical formulation and the physical reality of molecules. Existing approaches predominantly treat molecules as unstructured point clouds in Cartesian space, overlooking the intrinsic hierarchical mechanics where bond lengths and bond angles are relatively stiff, whereas torsion angles constitute the dominant flexible degrees of freedom. This lack of manifold awareness forces models to relearn fundamental geometric constraints from scratch, often leading to physically implausible intermediate structures. To address this, we propose GO-Flow that aligns generative modeling with molecular geometry via manifold decomposition. Instead of forcing motion through Euclidean space, GO-Flow decomposes the generation process into three physically motivated subspaces: translation space with linear optimal transport, rotation space with geodesic flows on $SO(3)$, and conformation space with entropic optimal transport. This decomposition injects geometric inductive biases and makes the generative paths better aligned with molecular degrees of freedom. When combined with equivariant neural architectures, it encourages rotation-consistent generation and improves geometric validity. Extensive experiments on GEOM-Drugs and GEOM-QM9 demonstrate that GO-Flow achieves state-of-the-art generation quality. Notably, by learning straighter probability paths on the correct manifolds naturally, our method enables high-fidelity sampling with as few as 50 steps, effectively bridging the gap between structural precision and computational efficiency.

分子生成流形学习几何先验高效采样

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