Training a Scientific Reasoning Model for Chemistry

Siddharth Narayanan (Edison Scientific) · James Braza (Edison Scientific) · Ryan-Rhys Griffiths (University of Cambridge) · Albert Bou (FutureHouse / Edison Scientific) · Geemi Wellawatte (FutureHouse) · Mayk Caldas Ramos (University of Rochester) · Ludovico Mitchener (FutureHouse) · Michael Pieler (Edison Scientific (prev. FutureHouse)) · Sam Rodriques (FutureHouse) · Andrew White (University of Rochester)
chemistry tasksdata efficiencydomain-specific modelsexperimental groundinggeneral-purpose modelsgenerative taskslanguage model generalizationorganic moleculesparameter scalingpost-trainingpredictive tasksreasoning modelsreinforcement learningscientific domainsthought tokens

Reasoning models are large language models that use extra "thought tokens" before answering, providing both higher accuracy and explicit reasoning for their response. A major question has been whether language model reasoning generalizes beyond mathematics, programming, and logic, where most previous work has focused. We demonstrate that reasoning models can be post-trained in scientific domains without additional domain pretraining, and require substantially less data compared to contemporary domain-specific models. We report ether0, a 24B parameter LLM (based on Mistral-Small-24B) that can reason in natural language and respond with chemical structures. This reasoning model was trained with reinforcement learning on 577,790 experimentally-grounded chemistry tasks involving synthesized organic molecules. Our model outperforms all previous general-purpose chemistry models, frontier models, and humans, and is more data efficient relative to specialized models. We anticipate that this method can be applied to train highly data-efficient language models specialized for predictive and generative tasks across a wide variety of scientific domains.