用能量模型改进3D分子结构生成,提升精度与稳定性。
Energy-Based Flow Matching for Generating 3D Molecular Structure
- 基于能量视角设计迭代映射网络,直接学习从随机构型到目标结构的变换。
- 在蛋白质对接和骨架生成任务上优于主流流匹配与扩散模型,计算成本相当。
- 方法具备理论保障,可对接AlphaFold等结构优化技术,适合生物分子设计者。
分子结构生成是确定分子原子三维位置的核心问题,广泛应用于分子对接、蛋白质折叠和分子设计等生物领域。近年来,扩散模型和流匹配等生成模型通过将分子构象建模为分布,在该任务上取得显著进展。本文聚焦流匹配,提出一种基于能量模型的视角,以改进训练与推理过程。该方法学习一个由深度网络表示的映射函数,可迭代地将来自源分布的随机构型映射至数据流形上的目标结构。这一框架概念简洁、实证有效,具有理论依据,并与幂等性、稳定性等基本性质以及AlphaFold中的结构精修等实用技术存在有趣关联。在蛋白质对接和蛋白质主链生成任务上的实验表明,本方法在相似计算预算下持续优于近期相关流匹配与扩散模型基线。
原文摘要 · Abstract (English)
Molecular structure generation is a fundamental problem that involves determining the 3D positions of molecules' constituents. It has crucial biological applications, such as molecular docking, protein folding, and molecular design. Recent advances in generative modeling, such as diffusion models and flow matching, have made great progress on these tasks by modeling molecular conformations as a distribution. In this work, we focus on flow matching and adopt an energy-based perspective to improve training and inference of structure generation models. Our view results in a mapping function, represented by a deep network, that is directly learned to \textit{iteratively} map random configurations, i.e. samples from the source distribution, to target structures, i.e. points in the data manifold. This yields a conceptually simple and empirically effective flow matching setup that is theoretically justified and has interesting connections to fundamental properties such as idempotency and stability, as well as the empirically useful techniques such as structure refinement in AlphaFold. Experiments on protein docking as well as protein backbone generation consistently demonstrate the method's effectiveness, where it outperforms recent baselines of task-associated flow matching and diffusion models, using a similar computational budget.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。