A Large Scale Investigation of Scaling Limits in Chemical Language Models
cs.LG
Submitted: 2025-08-19
Updated: 2026-10-01
License: http://creativecommons.org/licenses/by/4.0/
The gist: Chemical Language Models (CLMs) are increasingly used in de novo drug design, driven by recent growth in model scale, compute, and dataset size.
Terminology
Abstract
Chemical Language Models (CLMs) are increasingly used in de novo drug design, driven by recent growth in model scale, compute, and dataset size. However, the relationship between design choices, training dynamics, and downstream generation quality remains poorly understood. We present a compute-controlled scaling study of CLMs comprising more than 30,000 experiments across molecular representations (SMILES, SELFIES, SAFE), tokenizations (atom-level and byte-pair encoding), model scales (0.5M-1B parameters), leakage-controlled datasets (MOSES, ChEMBL, PubChem, ZINC-22), and architectures (decoder-only and encoder-decoder). By fitting IsoFLOP profiles, we establish clear scaling trends in pretraining loss, but find that these improvements do not translate into comparable gains in goal-directed molecular design. Layer-wise probing and sparse autoencoder analysis reveal continued development of chemical representations: chemical syntax saturates early, while semantic properties emerge more slowly and become increasingly accessible with further training and model scale. These representational gains coexist with diminishing improvements in goal-directed generation under the evaluated protocols and oracle budgets. Our resulting suite of models, NovoMolGen, achieves state-of-the-art results, outperforming prior CLMs and specialized generative models in goal-directed molecular generation across drug discovery tasks. These findings expose a disconnect between chemical representation learning and downstream molecular design, motivating the development of pretraining paradigms that more directly learn chemical semantics.
Sources
- BatchTopK Sparse Autoencoders
- MolGAN: An implicit generative model for small molecular graphs
- Sparse Autoencoders Find Highly Interpretable Features in Language Models
- The Llama 3 Herd of Models
- Sample Efficiency Matters: A Benchmark for Practical Molecular Optimization
- OLMo: Accelerating the Science of Language Models
- Saturn: Sample-efficient Generative Molecular Design using Memory Manipulation
- Mistral 7B
- Scaling Laws for Neural Language Models
- GP-MoLFormer: A Foundation Model For Molecular Generation
- Proximal Policy Optimization Algorithms
- GraphVAE: Towards Generation of Small Graphs Using Variational Autoencoders
- Molecular Fingerprints for Robust and Efficient ML-Driven Molecular Generation
- Unveiling Scaling Behaviors in Molecular Language Models: Effects of Model Size, Data, and Representation
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks