Training a Scientific Reasoning Model for Chemistry
This work addresses the need for data-efficient and accurate AI models in scientific domains like chemistry, offering a method that could be applied broadly across various scientific fields.
The paper tackled the problem of whether language model reasoning generalizes to chemistry by post-training a reasoning model without domain pretraining, achieving higher accuracy and data efficiency than general-purpose and specialized models, with concrete results including exceeding human experts on molecular design tasks.
Reasoning models are large language models that emit a long chain-of-thought 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 for chemistry 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 640,730 experimentally-grounded chemistry problems across 375 tasks ranging from synthesizability, to blood-brain barrier permeability, to human receptor activity, to scent. Our model exceeds general-purpose chemistry models, frontier models, and human experts on molecular design tasks. It is also more data efficient relative to specialized models. We anticipate that this method can be applied to train data-efficient language models specialized for tasks across a wide variety of scientific domains.