arrow
Return

Tokenization for Molecular Foundation Models

delete2026-01-23
delete0
PRE
AI
A
Alexius Wadell
A
Anoushka Bhutani
V
Venkatasubramanian Viswanathan *
DOI:10.1021/acs.jcim.5c01856delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Text-based foundation models have become an important part of scientific discovery, with molecular foundation models accelerating advancements in material science and molecular design. However, existing models are constrained by closed-vocabulary tokenizers that capture only a fraction of molecular space. In this work, we systematically evaluate 35 tokenizers, including 20 chemistry-specific tokenizers, and reveal significant gaps in their coverage of the SMILES molecular representation. To assess the impact of tokenizer choice, we introduce n-gram language models as a low-cost proxy and validate their effectiveness by pretraining and finetuning 18 RoBERTa-style encoders for molecular property prediction. To overcome the limitations of existing tokenizers, we propose two new tokenizers─Smirk and Smirk-GPE─with full coverage of the OpenSMILES specification. The proposed tokenizers systematically integrate nuclear, electronic, and geometric degrees of freedom, facilitating applications in pharmacology, agriculture, biology, and energy storage. Our results highlight the need for open-vocabulary modeling and chemically diverse benchmarks in cheminformatics.

Journal

Journal of Chemical Information and Modeling cover
Journal of Chemical Information and Modeling
IF:
5.3
Papers:
9.1K
Citations:
4.0W

Organization

U
University of Michigan
Scholars:
6.4W
Papers: 5.3W
Citations: 124
Cited Papers

Cited Papers

E(3)-equivariant graph neural networks for data-efficient and accurate interatomic potentials
err2022-05-04
err648
errOAAI
errBatzner, Simon; Musaelian, Albert; Sun, Lixin; Geiger, Mario; Mailoa, Jonathan P.; Kornbluth, Mordechai; Molinari, Nicola; Smidt, Tess E.; Kozinsky, Boris
errShare
errSave
Found in Translation: predicting outcomes of complex organic chemistry reactions using neural sequence-to-sequence models
err2018-01-01
err281
errOAAI
errSchwaller, Philippe; Gaudin, Theophile; Lanyi, David; Bekas, Costas; Laino, Teodoro
errShare
errSave
err
IF0
err
err0
errOAAI
err
errShare
errSave
Generative models for molecular discovery: Recent advances and challenges
err2022-03-05
err124
PREAI
errBilodeau, Camille; Jin, Wengong; Jaakkola, Tommi; Barzilay, Regina; Jensen, Klavs F.
errShare
errSave
Predicting Electrolyte Conductivity Directly from Molecular-Level Interactions
err2022-10-24
err7
PREAI
errZhang, Yumin; Bier, Imanuel; Viswanathan, Venkatasubramanian
errShare
errSave
researcher View more