提出新模型FlowMol-CTMC,提升3D分子生成质量与效率。
Exploring Discrete Flow Matching for 3D De Novo Molecule Generation
- 基于离散流匹配构建分子生成框架,支持原子元素等离散结构生成。
- 在3D小分子生成任务中达到当前最优性能,参数量更少。
- 引入新评估指标,发现现有模型易生成异常功能基团。
深度生成模型在生成新型分子结构方面具有推动化学发现的潜力。流匹配是一种近期提出的生成建模框架,在生物分子结构等任务中表现优异。但原始流匹配仅适用于连续数据,而从头分子设计需生成原子元素或氨基酸序列等离散数据。为此,近期已有若干离散流匹配方法被提出。本文对现有离散流匹配方法在3D从头小分子生成任务中的性能进行基准测试,并解释其行为差异。结果提出FlowMol-CTMC,一个开源模型,在3D从头设计中实现当前最优性能,且可学习参数更少。此外,提出衡量分子质量的新指标,超越局部化学价约束,关注更高阶结构特征。结果显示,尽管满足基本化学约束,模型仍倾向于生成训练数据分布外的异常、潜在问题的功能基团。代码与训练模型可在https://github.com/dunni3/FlowMol获取。
原文摘要 · Abstract (English)
Deep generative models that produce novel molecular structures have the potential to facilitate chemical discovery. Flow matching is a recently proposed generative modeling framework that has achieved impressive performance on a variety of tasks including those on biomolecular structures. The seminal flow matching framework was developed only for continuous data. However, de novo molecular design tasks require generating discrete data such as atomic elements or sequences of amino acid residues. Several discrete flow matching methods have been proposed recently to address this gap. In this work we benchmark the performance of existing discrete flow matching methods for 3D de novo small molecule generation and provide explanations of their differing behavior. As a result we present FlowMol-CTMC, an open-source model that achieves state of the art performance for 3D de novo design with fewer learnable parameters than existing methods. Additionally, we propose the use of metrics that capture molecule quality beyond local chemical valency constraints and towards higher-order structural motifs. These metrics show that even though basic constraints are satisfied, the models tend to produce unusual and potentially problematic functional groups outside of the training data distribution. Code and trained models for reproducing this work are available at \url{https://github.com/dunni3/FlowMol}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。