mlsanityData to 14 Oct 2026

Arbitrary-Length Generalization for Addition in a Tiny Transformer

arXiv 2406.00075 ↗First posted 31 May 2024cs.LG stat.AP stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

This paper introduces a novel training methodology that enables a Transformer model to generalize the addition of two-digit numbers to numbers with unseen lengths of digits. The proposed approach employs an autoregressive generation technique, processing from right to left, which mimics a common manual method for adding large numbers. To the best of my knowledge, this methodology has not been previously explored in the literature. All results are reproducible, and the corresponding R code is available at github.com/AGPatriota/ALGA-R/.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 5 of its 6 references that are in the map. First posted on arXiv 31 May 2024.

Builds on

5 of 6 references.

References · 5

  1. 2017Attention Is All You Needcited by 60,361
  2. 2021Show Your Work: Scratchpads for Intermediate Computation with Language Modelscited by 886
  3. 2021Investigating the Limitations of Transformers with Simple Arithmetic Taskscited by 139
  4. 2023Teaching Arithmetic to Small Transformerscited by 123
  5. 2024Transformers Can Do Arithmetic with the Right Embeddingscited by 78

Built on by

Not cited yet.