arXiv:2411.12078cs.LG2024-11NeurIPS被引 34

通过检索增强生成新分子,突破已有片段限制。

Molecule Generation with Fragment Retrieval Augmentation

  • 基于预训练模型检索硬/软片段,引导生成新分子。
  • 迭代更新片段库,生成高质量新颖分子。
  • 适合药物发现中需要突破现有化学空间的场景。

基于片段的药物发现通过组装分子片段生成具有理想生化性质的新分子已取得显著进展。然而,许多片段生成方法仅限于数据库中现有片段的重组或微调,难以拓展新化学空间。为此,本文提出一种带检索增强的片段生成框架f-RAG。该框架基于预训练分子生成模型,在输入片段基础上检索两类片段:(1) 硬片段,作为将被显式包含在新分子中的构建块;(2) 软片段,通过可训练的片段注入模块指导新片段生成。为实现对现有片段库的外推,f-RAG通过迭代精炼过程更新片段词汇表,并结合事后遗传片段修改进一步优化。该方法通过维护并扩展高质量片段池,利用强生成先验实现了探索与利用之间的更好平衡。

原文摘要 · Abstract (English)

Fragment-based drug discovery, in which molecular fragments are assembled into new molecules with desirable biochemical properties, has achieved great success. However, many fragment-based molecule generation methods show limited exploration beyond the existing fragments in the database as they only reassemble or slightly modify the given ones. To tackle this problem, we propose a new fragment-based molecule generation framework with retrieval augmentation, namely Fragment Retrieval-Augmented Generation (f-RAG). f-RAG is based on a pre-trained molecular generative model that proposes additional fragments from input fragments to complete and generate a new molecule. Given a fragment vocabulary, f-RAG retrieves two types of fragments: (1) hard fragments, which serve as building blocks that will be explicitly included in the newly generated molecule, and (2) soft fragments, which serve as reference to guide the generation of new fragments through a trainable fragment injection module. To extrapolate beyond the existing fragments, f-RAG updates the fragment vocabulary with generated fragments via an iterative refinement process which is further enhanced with post-hoc genetic fragment modification. f-RAG can achieve an improved exploration-exploitation trade-off by maintaining a pool of fragments and expanding it with novel and high-quality fragments through a strong generative prior.

分子生成片段检索药物发现

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