用分子表示引导生成,提升逆合成反应预测精度与效率
Representation-Guided Discrete Molecular Graph Retrosynthesis

- 通过预训练分子表示指导生成过程,显式引入化学语义
- 在USPTO-50k上达到87.1%的top-10准确率,多样性提升至15.5
- 减少35%训练轮次,适合追求高效高质逆合成设计的研究者
基于随机过程的分子图生成器已成为无模板单步逆合成的最新方法。然而,这些模型仅在产物-反应物对上训练,化学相关表示的获取是间接且隐式的。近期计算机视觉进展表明,向生成器提供表示引导可有效将预训练编码器中的语义信息注入到DiTs中,显著提升收敛速度与生成质量。该方法是否适用于逆合成任务,以及如何进行图结构特定的设计,仍是开放问题。为此,我们系统性地研究了一个统一的设计空间,涵盖教师分子表示、终点与粒度选择、去噪器中的注入深度、对应策略及引导方案。基于此,我们提出图导向表示引导(GRG),在USPTO-50k数据集上实现58.6 / 77.2 / 83.4 / 87.1的top-1 / 3 / 5 / 10准确率,多样性达15.5,显著优于基线生成器。值得注意的是,GRG在分布外设置下持续提升所有top-k指标,表明表示引导有助于获取内在化学语义。此外,表示引导使训练轮次减少35%,壁钟时间缩短30%即可达到相当性能。我们还引入一种基于表示相似性的简单重排序机制,无需额外训练验证器即可进一步提升排名前列结果。
原文摘要 · Abstract (English)
Stochastic process-based molecular graph generators have become the state of the art for template-free single-step retrosynthesis. However, these models are typically trained only on product-reactant pairs, thereby acquiring chemistry-relevant representations in an indirect and implicit manner. Meanwhile, recent advances in computer vision demonstrate that offering representation guidance to a generator can effectively distill semantics from pretrained encoders into DiTs, substantially improving both convergence and generation quality. Whether similar gains extend to the retrosynthesis task, and what graph-specific design choices can make them work, remains an open question. To address these questions, we conduct a systematic empirical study over a unified design space spanning teacher molecular representations, endpoint and granularity choices, injection depths in the denoiser, correspondence strategies and guidance scheme. Guided by these considerations, we develop Graph-oriented Representation Guidance (GRG), which achieves 58.6 / 77.2 / 83.4 / 87.1 top-1 / 3 / 5 / 10 accuracy on USPTO-50k, while increasing diversity to 15.5, both substantially outperforming the adopted base generator. Notably, GRG consistently improves all top-k metrics in out-of-distribution settings, suggesting that representation guidance facilitates the acquisition of intrinsic chemical semantics. Meanwhile, the introduced representation guidance reduces the number of epochs by 35% and the wall-clock time by 30% to reach comparable performance. In addition, we introduce a simple yet effective representation-similarity-based reranking mechanism, which further improves the top of the ranked list without training an additional verifier.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。