English

AtomComposer: Discovering Chemical Space from First Principles with Reinforcement Learning

Machine Learning 2026-05-28 v1 Materials Science

Abstract

Discovering novel stable molecules without training data remains a grand scientific challenge. Current molecular generative models are trained on large, pre-curated datasets, which introduce biases and limit exploration of novel chemistry. In contrast, we propose a new paradigm: autonomous, generalized agents capable of mapping vast, unknown chemical spaces without any pretraining. For the first time, we present AtomComposer, a self-guided agent that autonomously constructs valid 3D isomers under stoichiometric constraints and is trained exclusively online using reinforcement learning. Unlike existing approaches that generally overfit to a specific chemical formula, we establish a multi-composition training scheme that enables a broad generalization across diverse chemistry, guided by energy- and validity-based rewards. Our agent can discover up to an order of magnitude more valid isomers on unseen test formulas than existing single-composition reinforcement-learning baselines trained with per-step energy rewards. These results fulfill the promise of online reinforcement learning as a powerful paradigm for scalable, from-scratch exploration of chemical configuration space.

Keywords

Cite

@article{arxiv.2605.28287,
  title  = {AtomComposer: Discovering Chemical Space from First Principles with Reinforcement Learning},
  author = {Bjarke Hastrup and Francois Cornet and Tejs Vegge and Arghya Bhowmik},
  journal= {arXiv preprint arXiv:2605.28287},
  year   = {2026}
}
R2 v1 2026-07-22T07:36:53.899Z