Skip to content

Training Chemical Plausibility-Aware Large Language Models for Single-Step Retrosynthesis

Aug 2026 · 0 citations · 40 references
Computer Science

TL;DR

This work introduces Top-K prompting as a robust training and inference paradigm to better capture diverse, plausible reaction predictions and establishes Top-K, plausibility-aware training as a practical new direction for robust future LLM-based synthesis planning.

Abstract

Single-step retrosynthesis is a central component of computer-aided synthesis planning, yet its intrinsically one-to-many nature is poorly captured by single-answer evaluation and benchmarking protocols. To address this, we introduce Top-K prompting as a robust training and inference paradigm to better capture diverse, plausible reaction predictions. We compile CREED-CCV-2+USPTO-XL, an ultra-large-scale dataset of ~45.6 million verified reactions to train the C3LM (Chemistry Constraint-Consistent Language Model). By integrating fine-tuning with ChemCensor-based and novelty-oriented rewards, our model achieves state-of-the-art performance on the OOD URSA-expert-2026 benchmark. Further analysis of reaction uniqueness shows that LLMs and conventional models explore complementary reaction spaces, motivating ensemble-based retrosynthesis systems. Overall, our results establish Top-K, plausibility-aware training as a practical new direction for robust future LLM-based synthesis planning.

View source

Similar papers

Book Open access Aug 2026

Chem-R: Learning to Reason as a Chemist

The proposed Chem-R, a general Chemical Reasoning model designed to emulate the deliberative processes of chemists, achieves state-of-the-art performance on comprehensive benchmarks, surpassing leading LLMs, including Gemini-3-Pro and Kimi-k2.5.

Weida Wang, Benteng Chen, Di Zhang et al. · 0 citations
Preprint Jul 2026

URSA: Chemistry-Aware Benchmark for Utilitarian Retrosynthesis Assessment

Synthesis planning aiming to find pathways of reactions for a target molecule is one of the most important and challenging tasks in drug discovery. Recent progress has produced both specialized deep-learning retrosynthesis systems and general-purpose large language models, but objective comparison remains difficult due to the lack of flexible, chemically interpretable benchmarking protocols. In the current study, we are introducing the URSA (Utilitarian RetroSynthesis Assessment) evaluation framework that provides the opportunity to benchmark the synthetic routes not only from a formal perspective, such as convergence to commercially available starting materials, but also from a chemical plausibility perspective, mimicking the way expert chemists evaluate the reactions and routes. The study covers a comprehensive evaluation of both conventional end-to-end retrosynthesis solutions and LLMs for the synthesis planning task on a set of novel, diverse target molecules with undisclosed synthetic routes, which represent realistic tasks in the daily drug design routine. We find that while LLMs can support high-level strategic planning, they currently underperform specialized retrosynthesis models in reliably solving synthesis planning tasks.

B. Zagribelnyy, Ivan D. Ilin, N. Bondarev et al. · 0 citations
Open access Jul 2026

RetroMPA: A Molecular Property-Aware Auxiliary Framework for Enhancing Retrosynthesis Prediction.

Retrosynthesis is a cornerstone of drug discovery and organic synthesis. While data-driven deep learning models have shown remarkable progress, they are designed to autonomously learn reaction patterns from extensive retrosynthesis data sets with limited explicit integration of established chemical knowledge as priors. To address this limitation, we introduce RetroMPA, a molecular property-aware, posthoc enhancement module that injects chemical knowledge into the retrosynthesis pipeline. Rather than functioning as an independent, standalone SMILES sequence generator from scratch, RetroMPA is conceptualized as a broadly applicable, model-agnostic chemical filter designed to recalibrate and optimize the predictive pathways of various existing algorithms. This plug-and-play framework can be seamlessly integrated with a range of existing data-driven retrosynthesis methods, enhancing model outputs without necessitating any modifications to the original model architecture or requiring resource-intensive, model-specific retraining procedures. By operating at the molecular level and leveraging a property-aware latent embedding space, RetroMPA consistently improves top-1 accuracy across eight representative retrosynthesis models by an average of 5.50% on USPTO-50K. Furthermore, we demonstrate its scalability by validating its performance on the large-scale USPTO-Full data set, achieving an average improvement of about 2.03% across both template-based and template-free architectures. In addition, wet-lab experiments provide preliminary support for the practical utility of the framework. These syntheses confirmed viable, previously unreported substrate combinations for established, classic reaction paradigms─specifically, the Suzuki-Miyaura coupling, the Bucherer reaction, and the Friedel-Crafts acylation, thereby suggesting that RetroMPA can operate beyond mere data fitting. The code is open-sourced at https://github.com/MengzhouLu/RetroMPA.

Mianzhi Liu, Fan Xiao, Zhi-Qiang Yu et al. · 0 citations
Book Open access Aug 2026

Beyond Reaction Data: Learning Chemical Knowledge from Large-Scale Molecules for Retrosynthesis

Retrosynthesis, the process of predicting reactants from products, remains a critical challenge in computational chemistry and drug discovery. While recent deep learning methods have shown strong performance, they remain overly reliant on reaction datasets, which are limited in availability and quality. Large-scale unlabeled molecular data encode rich structural patterns that can be leveraged to learn transferable chemical knowledge, but remain largely unexplored. In this work, we propose KnowRetro (Knowledge-Guided Retrosynthesis Prediction), a chemically-aware framework that learns chemical knowledge from large-scale unlabeled molecules to enhance the accuracy and diversity of retrosynthesis prediction. Specifically, KnowRetro first builds a hierarchical knowledge graph from millions of unlabeled molecules, which captures transformation-relevant relationships among molecules, substructures, and functional groups. It then employs chemically guided pre-training based on substructure decomposition to encourage the model to capture fundamental reaction patterns, followed by fine-tuning with an adapter designed to inject task-relevant knowledge into reactant generation. Extensive experiments demonstrate that KnowRetro achieves high accuracy with improved robustness and diversity in reactant generation. Our code is available at https://github.com/chenyujie1127/KnowRetro.

Yujie Chen, Tengfei Ma, Zhou Yu et al. · 0 citations
Open access Aug 2026

Robust out-of-distribution prediction of Buchwald-Hartwig reactions.

A framework that systematically standardizes and integrates multiple reaction datasets into a high-quality, unique-structure-per-entity dataset, coupled with active learning to strategically expand chemical space is introduced, establishing a blueprint for robust machine learning in synthetic chemistry.

Paulo Neves, Bo Hao, Santeri Aikonen et al. · 1 citation

Related blog posts

MIT News · Artificial Intelligence Aug 27, 2026

Looking beyond natural sequences

A new machine-learning framework aims to improve the success rate of computational protein design while moving away from results that reproduce sequences found in nature.