Score: 0

Pre-trained Language Models Learn Remarkably Accurate Representations of Numbers

Published: June 10, 2025 | arXiv ID: 2506.08966v1

By: Marek Kadlčík , Michal Štefánik , Timothee Mickus and more

Potential Business Impact:

Fixes math mistakes in smart computer programs.

Business Areas:
Natural Language Processing Artificial Intelligence, Data and Analytics, Software

Pretrained language models (LMs) are prone to arithmetic errors. Existing work showed limited success in probing numeric values from models' representations, indicating that these errors can be attributed to the inherent unreliability of distributionally learned embeddings in representing exact quantities. However, we observe that previous probing methods are inadequate for the emergent structure of learned number embeddings with sinusoidal patterns. In response, we propose a novel probing technique that decodes numeric values from input embeddings with near-perfect accuracy across a range of open-source LMs. This proves that after the sole pre-training, LMs represent numbers with remarkable precision. Finally, we find that the embeddings' preciseness judged by our probe's accuracy explains a large portion of LM's errors in elementary arithmetic, and show that aligning the embeddings with the pattern discovered by our probe can mitigate these errors.

Page Count
8 pages

Category
Computer Science:
Computation and Language