atlas

Research map / Probabilistic and scientific modeling / Molecules and proteins

RetroBridge: Modeling Retrosynthesis with Markov Bridges

ICLR 2024 · Probabilistic and scientific modeling › Molecules and proteins · approach: Structure-based 3D molecule generation · 4 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes the Markov Bridge Model, a discrete generative framework that learns the dependency between two intractable discrete distributions, and applies it as RetroBridge for template-free single-step retrosynthesis on USPTO-50k.

Problem

Single-step retrosynthesis requires predicting sets of reactants for a product molecule; the product-reactant dependency is modeled as a mapping between two intractable discrete distributions, which standard diffusion models (designed for one intractable distribution from a tractable noise prior) fi

Approach

Model the product-to-reactant mapping as a Markov bridge pinned at a product start point and a reactant endpoint, using a categorical transition kernel with a learned predictor of the final state trained by a variational lower bound; sampling starts from the product itself rather than from noise, and confidence scores come from the sample frequency of each reactant set.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use the Markov Bridge formulation instead of DiGress with product contextretrosynthesis prediction · USPTO-50k validationtop-5 exact-match accuracy: 73.93 -> 79.44Helped
add product context at each sampling step (ablation)RetroBridge-VLB · USPTO-50k validationtop-5 exact-match accuracy: 75.21 -> 79.44Helped
replace VLB with CE loss (ablation)RetroBridge with product context · USPTO-50k validationtop-5 exact-match accuracy: 79.44 -> 76.58Hurt
replace VLB with CE loss (ablation)RetroBridge without product context · USPTO-50k validationtop-5 exact-match accuracy: 75.21 -> 72.33Hurt
add molecular weight and atom valency features (ablation)RetroBridge · USPTO-50k validationtop-5 exact-match accuracyHurt
reduce sampling steps from T=500 to T=250 (ablation)RetroBridge with CE loss · USPTO-50k validationtop-k accuracyNo clear effect
reduce sampling steps by 5-fold (ablation)RetroBridge with CE loss · USPTO-50k validationtop-k accuracyHurt
reduce the number of generated samples by 5-fold (ablation)RetroBridge with VLB loss · USPTO-50k testtop-k accuracyNo clear effect
use one-shot graph prediction instead of iterative sampling (ablation)graph transformer retrosynthesis model · USPTO-50k validationexact-match accuracyHurt
use product context with DiGress (ablation)DiGress · USPTO-50k validationtop-5 exact-match accuracyHelped
use RetroBridge instead of template-free baselinesretrosynthesis prediction · USPTO-50k testtop-5 exact-match accuracy: 78.1 -> 80.6Helped
use RetroBridge instead of LocalRetroretrosynthesis prediction · USPTO-50k testtop-5 round-trip coverage: 94.7 -> 97.1Helped

Surprising findings

Limitations

Experimental setup

Datasets: USPTO-50k

Benchmarks: USPTO-50k

Metrics: top-k exact match accuracy round-trip accuracy round-trip coverage

Baselines: GLN LocalRetro GraphRetro MEGAN G2G Augmented Transformer SCROP Graph2SMILES Retroformer DiGress

Builds on: DiGress discrete diffusion models Schrödinger bridge Graph Transformer

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy