MoleBridge: Synthetic Space Projecting with Discrete Markov Bridges

Rongchao Zhang (peking university) · Yu Huang (University of Pennsylvania) · Yongzhi Cao (Peking University) · Hanpin Wang (Peking University)
application scenarioschemical spacesconditional probability transitionsde novo molecular designdeep generative modeldenoising separation approachdiscrete reaction postfix notationsdiscrete state spacegenerative probability pathsmarkov bridge approachmolecular graphsmolecular synthetic spacenoisy token blocksnotation token optimizationsynthesis pathway predictionsynthetic postfix notations

Molecular synthetic space projecting is a critical technique in de novo molecular design, which aims to rectify molecules without synthesizability guarantee by converting them into synthetic postfix notations. However, the vast synthesizable chemical space and the discrete data modalities involved pose significant challenges to postfix notation conversion benchmarking. In this paper, we exploit conditional probability transitions in discrete state space and introduce MoleBridge, a deep generative model built on the Markov bridge approach for designing postfix notations of molecular synthesis pathways. MoleBridge consists of two iterative optimizations: i) Autoregressive extending of notation tokens from molecular graphs, and ii) generation of discrete reaction postfix notations through Markov bridge, where noisy token blocks are progressively denoised over multi-step iterations. For the challenging second iteration, which demands sensitivity to incorrect generative probability paths within intricate chemical spaces, we employ a thinking and denoising separation approach to denoise. Empirically, we find that MoleBridge is capable of accurately predicting synthesis pathways while exhibiting excellent performance in a variety of application scenarios.