arXiv · 2406.00075
Arbitrary-Length Generalization for Addition in a Tiny Transformer
Abstract
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/.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Alexandre Galvao Patriota. 2024-05-31. Arbitrary-Length Generalization for Addition in a Tiny Transformer. https://arxiv.org/abs/2406.00075
Cite the original work for its findings. Save a collection to share your selection of sources.