FlowMol3用三招改进生成分子的精度与稳定性,无需改架构。
FlowMol3: Flow Matching for 3D De Novo Small-Molecule Generation
- 通过自条件、假原子和训练时几何扰动提升生成质量。
- 生成药物分子时接近100%有效率,且结构更贴近真实数据。
- 参数量少一倍,适合高效部署于新药研发场景。
能够采样具有目标属性的真实分子的生成模型,可加速化学发现。为此,研究聚焦于联合生成分子拓扑与三维结构的模型。我们提出FlowMol3,一个开源的多模态流匹配模型,在全原子小分子生成任务上达到当前最佳性能。相比前代FlowMol版本,其性能显著提升,但未改变图神经网络架构或底层流匹配公式。改进源于三项架构无关的技术:自条件、假原子和训练时几何畸变,计算开销极低。FlowMol3在含显式氢的药物类分子上实现近100%分子有效性,更准确还原训练数据的功能团组成与几何特征,且学习参数量比同类方法少一个数量级。我们推测这些技术缓解了基于传输的生成模型普遍存在的分布漂移问题,使推理过程中可检测并修正分布偏差。结果表明,这些策略简单、可迁移,能显著提升扩散与流模型的稳定性和生成质量。
原文摘要 · Abstract (English)
A generative model capable of sampling realistic molecules with desired properties could accelerate chemical discovery across a wide range of applications. Toward this goal, significant effort has focused on developing models that jointly sample molecular topology and 3D structure. We present FlowMol3, an open-source, multi-modal flow matching model that advances the state of the art for all-atom, small-molecule generation. Its substantial performance gains over previous FlowMol versions are achieved without changes to the graph neural network architecture or the underlying flow matching formulation. Instead, FlowMol3's improvements arise from three architecture-agnostic techniques that incur negligible computational cost: self-conditioning, fake atoms, and train-time geometry distortion. FlowMol3 achieves nearly 100% molecular validity for drug-like molecules with explicit hydrogens, more accurately reproduces the functional group composition and geometry of its training data, and does so with an order of magnitude fewer learnable parameters than comparable methods. We hypothesize that these techniques mitigate a general pathology affecting transport-based generative models, enabling detection and correction of distribution drift during inference. Our results highlight simple, transferable strategies for improving the stability and quality of diffusion- and flow-based molecular generative models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。