Convergent Evolution: How Different Language Models Learn Similar Number Representations
Abstract
Transformers and other language models exhibit periodic numerical representations in their Fourier domains, with some models developing geometrically separable features for linear classification of numbers modulo T, though Fourier sparsity alone is insufficient for this separability.
Language models trained on natural text learn to represent numbers using periodic features with dominant periods at T=2, 5, 10. In this paper, we identify a two-tiered hierarchy of these features: while Transformers, Linear RNNs, LSTMs, and classical word embeddings trained in different ways all learn features that have period-T spikes in the Fourier domain, only some learn geometrically separable features that can be used to linearly classify a number mod-T. To explain this incongruity, we prove that Fourier domain sparsity is necessary but not sufficient for mod-T geometric separability. Empirically, we investigate when model training yields geometrically separable features, finding that the data, architecture, optimizer, and tokenizer all play key roles. In particular, we identify two different routes through which models can acquire geometrically separable features: they can learn them from complementary co-occurrence signals in general language data, including text-number co-occurrence and cross-number interaction, or from multi-token (but not single-token) addition problems. Overall, our results highlight the phenomenon of convergent evolution in feature learning: A diverse range of models learn similar features from different training signals.
Community
Convergent Evolution: How Different Language Models Learn Similar Number Representations
This paper is an awesome read. Every question i found myself having was answered somewhere further in the paper. Great job. Hope you do more
Get this paper in your agent:
hf papers read 2604.20817 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper