English

The Tokenization Bottleneck: How Vocabulary Extension Improves Chemistry Representation Learning in Pretrained Language Models

Computation and Language 2025-11-19 v1 Artificial Intelligence

Abstract

The application of large language models (LLMs) to chemistry is frequently hampered by a "tokenization bottleneck", where tokenizers tuned on general-domain text tend to fragment chemical representations such as SMILES into semantically uninformative sub-tokens. This paper introduces a principled methodology to resolve this bottleneck by unifying the representation of natural language and molecular structures within a single model. Our approach involves targeted vocabulary extension-augmenting a pretrained LLM's vocabulary with chemically salient tokens, followed by continued pretraining on chemistry-domain text to integrate this new knowledge. We provide an empirical demonstration of the effectiveness of this strategy, showing that our methodology leads to superior performance on a range of downstream chemical tasks.

Keywords

Cite

@article{arxiv.2511.14365,
  title  = {The Tokenization Bottleneck: How Vocabulary Extension Improves Chemistry Representation Learning in Pretrained Language Models},
  author = {Prathamesh Kalamkar and Ned Letcher and Meissane Chami and Sahger Lad and Shayan Mohanty and Prasanna Pendse},
  journal= {arXiv preprint arXiv:2511.14365},
  year   = {2025}
}