English

Thinking Tokens for Language Modeling

Computation and Language 2024-05-15 v1 Artificial Intelligence

Abstract

How much is 56 times 37? Language models often make mistakes in these types of difficult calculations. This is usually explained by their inability to perform complex reasoning. Since language models rely on large training sets and great memorization capability, naturally they are not equipped to run complex calculations. However, one can argue that humans also cannot perform this calculation immediately and require a considerable amount of time to construct the solution. In order to enhance the generalization capability of language models, and as a parallel to human behavior, we propose to use special 'thinking tokens' which allow the model to perform much more calculations whenever a complex problem is encountered.

Keywords

Cite

@article{arxiv.2405.08644,
  title  = {Thinking Tokens for Language Modeling},
  author = {David Herel and Tomas Mikolov},
  journal= {arXiv preprint arXiv:2405.08644},
  year   = {2024}
}

Comments

AITP 2023 (May 10, 2023)

R2 v1 2026-06-28T16:27:01.808Z