mCLM: A Modular Chemical Language Model that Generates Functional and Makeable Molecules
Carl Edwards, Chi Han, Gawon Lee, Thao Nguyen, Sara Szymkuć, Chetan Kumar Prasad, Bowen Jin, Jiawei Han, Ying Diao, Ge Liu, Hao Peng, Bartosz A. Grzybowski, Martin D. Burke, Heng Ji
TL;DR
mCLM addresses the gap between in silico molecule design and practical synthesis by tokenizing molecules into function-infused, automation-friendly building blocks and jointly modeling them with natural language in a bilingual, multimodal Transformer. It introduces a synthesis-aware vocabulary and a two-tokenization strategy, enabling automated assembly on modular synthesis platforms and grounding reasoning in real-world makeability. The model is trained with a unified loss over both modalities and employs critical chemical reasoning to iteratively refine designs toward multi-objective targets, validated through ADMET oracle evaluations, FDA-drug improvements, and case studies on fallen angels. The results demonstrate high validity and makeability, improved property profiles, and a scalable pathway to AI-guided, automated laboratory discovery with potential for autonomous, iterative drug and material development, while outlining future multimodal and systems-level enhancements.$\mathcal{L}=H(P(\mathbf{x}), P_\theta(\mathbf{x}))$.$
Abstract
Despite their ability to understand chemical knowledge, large language models (LLMs) remain limited in their capacity to propose novel molecules with desired functions (e.g., drug-like properties). In addition, the molecules that LLMs propose can often be challenging to make, and are almost never compatible with automated synthesis approaches. To better enable the discovery of functional small molecules, LLMs need to learn a new molecular language that is more effective in predicting properties and inherently synced with automated synthesis technology. Current molecule LLMs are limited by representing molecules based on atoms. In this paper, we argue that just like tokenizing texts into meaning-bearing (sub-)word tokens instead of characters, molecules should be tokenized at the level of functional building blocks, i.e., parts of molecules that bring unique functions and serve as effective building blocks for real-world automated laboratory synthesis. This motivates us to propose mCLM, a modular Chemical-Language Model that comprises a bilingual language model that understands both natural language descriptions of functions and molecular blocks. mCLM front-loads synthesizability considerations while improving the predicted functions of molecules in a principled manner. mCLM, with only 3B parameters, achieves improvements in synthetic accessibility relative to 7 other leading generative AI methods including GPT-5. When tested on 122 out-of-distribution medicines using only building blocks/tokens that are compatible with automated modular synthesis, mCLM outperforms all baselines in property scores and synthetic accessibility. mCLM can also reason on multiple functions and iteratively self-improve to rescue drug candidates that failed late in clinical trials ("fallen angels").
