GDiffRetro: Retrosynthesis Prediction with Dual Graph Enhanced Molecular Representation and Diffusion Generation

Authors

  • Shengyin Sun Department of Computer Science, City University of Hong Kong
  • Wenhao Yu Department of Computer Science and Engineering, The Chinese University of Hong Kong
  • Yuxiang Ren Advance Computing and Storage Lab, Huawei Technologies
  • Weitao Du Academy of Mathematics and Systems Science, Chinese Academy of Sciences
  • Liwei Liu Advance Computing and Storage Lab, Huawei Technologies
  • Xuecang Zhang Advance Computing and Storage Lab, Huawei Technologies
  • Ying Hu Department of Electronic Engineering, Tsinghua University
  • Chen Ma Department of Computer Science, City University of Hong Kong

DOI:

https://doi.org/10.1609/aaai.v39i12.33373

Abstract

Retrosynthesis prediction focuses on identifying reactants capable of synthesizing a target product. Typically, the retrosynthesis prediction involves two phases: Reaction Center Identification and Reactant Generation. However, we argue that most existing methods suffer from two limitations in the two phases: (i) Existing models do not adequately capture the ``face'' information in molecular graphs for the reaction center identification. (ii) Current approaches for the reactant generation predominantly use sequence generation in a 2D space, which lacks versatility in generating reasonable distributions for completed reactive groups and overlooks molecules' inherent 3D properties. To overcome the above limitations, we propose GDiffRetro. For the reaction center identification, GDiffRetro uniquely integrates the original graph with its corresponding dual graph to represent molecular structures, which helps guide the model to focus more on the faces in the graph. For the reactant generation, GDiffRetro employs a conditional diffusion model in 3D to further transform the obtained synthon into a complete reactant. Our experimental findings reveal that GDiffRetro outperforms state-of-the-art semi-template models across various evaluative metrics.

Published

2025-04-11

How to Cite

Sun, S., Yu, W., Ren, Y., Du, W., Liu, L., Zhang, X., … Ma, C. (2025). GDiffRetro: Retrosynthesis Prediction with Dual Graph Enhanced Molecular Representation and Diffusion Generation. Proceedings of the AAAI Conference on Artificial Intelligence, 39(12), 12595–12603. https://doi.org/10.1609/aaai.v39i12.33373

Issue

Section

AAAI Technical Track on Data Mining & Knowledge Management II