Leviathan: Decoupling Input and Output Representations in Language Models
Not on the map. Why?
Abstract (arXiv)
Modern language models use a single matrix for input embedding and output projection. This couples two distinct objectives: token representation and discrimination over a vocabulary. This work introduces Leviathan, a Transformer architecture that replaces the input embedding matrix with learned embedding vectorization (LEV), a compact continuous mapping from token indices to embeddings. Leviathan's output head remains untied for a parameter increase of as low as 0.2%. Under controlled comparisons with identical Transformer backbones, Leviathan consistently improves language modeling performance over standard tied-embedding baselines across a 200M-1.2B parameter regime on The Pile with gains that grow during training. At 1.2B scale, Leviathan reduces validation perplexity by 9%, requires 2.1× fewer training tokens to reach the tied baseline's final loss, and improves on all six downstream benchmarks evaluated, including a 30% reduction in LAMBADA perplexity. Frequency-stratified analysis reveals gains to be concentrated in rare tokens, where continuous parameterization reduces perplexity by 81%, falling to near zero for the most frequent.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 15 of its 20 references that are in the map. First posted on arXiv 29 Jan 2026.
Builds on
15 of 20 references.
References · 15
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,772
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
- 2018Deep contextualized word representationscited by 2,285
- 2020GLU Variants Improve Transformercited by 1,724
- 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 960
- 2016Using the Output Embedding to Improve Language Modelscited by 266
- 2019Representation Degeneration Problem in Training Natural Language Generation Modelscited by 250
- 2018Adaptive Input Representations for Neural Language Modelingcited by 226
- 2017Breaking the Softmax Bottleneck: A High-Rank RNN Language Modelcited by 147
3 more
- 2015Character-Aware Neural Language Modelscited by 129
- 2016Tying Word Vectors and Word Classifiers: A Loss Framework for Language Modelingcited by 87
- 2017Hash Embeddings for Efficient Word Representationscited by 54
Built on by
Citing papers per year: 2026: 2.
2 papers.