Cheminformatics

Atomic Design Transformer: Scaffold-Conditioned 3D Molecule Generation via xTB-Reward Reinforcement Learning

Takao Kotani

Featured August 8, 2026

This analysis was generated by SciGrove. Upload your own PDFs or enter a DOI — and get the same AI breakdown on any paper.

Get started

AI-generated analysis — This is SciGrove's AI interpretation of the paper, not peer-reviewed content. Always refer to the original paper.

Simply

By describing 3D molecules as a sequence of simple instructions for placing atoms in relation to their neighbors, a standard AI can learn to build new molecules, then improve itself by checking if its creations are physically stable.

In depth

The Atomic Design Transformer (ADT) introduces a novel SE(3)-invariant tokenization scheme, encoding 3D atomic positions as discrete tokens in local coordinate frames. This allows a standard causal transformer to autoregressively generate 3D molecules atom-by-atom. The method is further enhanced by Reinforcement Learning from Verifiable xTB Reward (RLVR), which significantly improves the physical validity and topology preservation of generated molecules without external data.

Key Takeaways

1
The paper introduces a fully discrete autoregressive transformer (ADT) for 3D molecule generation, achieving SE(3) invariance through local coordinate frame tokenization rather than equivariant network layers.
2
A novel metric, the xTB topology-preservation rate (XTP), is proposed to quantitatively evaluate whether the generated 3D molecular structures maintain their declared bond topology after quantum-mechanical relaxation.
3
The Reinforcement Learning from Verifiable xTB Reward (RLVR) approach enables the ADT model to self-improve its generation quality against physical rules, dramatically increasing the yield of valid, topology-preserving molecules to ~95%.

Conceptual Flow

HIGH LEVEL

Methodology: Building Molecules with Local Instructions

The system turns a 3D molecule into a list of simple instructions, uses a smart AI to make new instruction lists, and then turns those lists back into new 3D molecules.

Real 3D Molecule

Turn into Instructions

Instruction List

Results: Learning to Make Better Molecules

At first, the AI makes okay molecules, but after practicing and getting feedback from a physics checker, it learns to make much more stable and correct molecules.

AI's First Try

Physics Checker

Learn and Improve

AI's Better Molecules