Diversity Beats Size Scaling for Chemical Language Models

Read full text
Authors: Borja Medina, Alessandro Tibo, Jiazhen He, Jon Paul Janet, Nicklas Österbacka
Cited by: 0
References: 36

Abstract

Chemical language models, such as transformers trained on SMILES strings, are increasingly used in drug design and have seen rapid growth in both model capacity and training dataset size. The impact of this scaling on practical downstream performance remains unclear, however. We systematically evaluate how model size and dataset size affect encoder-decoder transformers trained on paired textual molecular representations. We find that, beyond a minimal threshold, further model scaling yields no gain in hit generation rate, while dataset scaling gives diminishing returns. We further introduce a dataset diversification strategy that substantially increases hit diversity. These results suggest that, for molecular hit discovery, data curation and diversity may be more impactful than continued scaling of model size or dataset volume, and they motivate a shift from scale-first to diversity-first training paradigms.

Fields of Study & Subjects

Machine Learning in Materials Science Computational Drug Discovery Methods Topic Modeling

Publication details

Journal / source
npj Drug Discovery.
Publisher
Springer Science and Business Media LLC
Volume
3
Issue
1
Publication date
4 August 2026
Language
EN