mlsanityData to 7 Oct 2026

Leviathan: Decoupling Input and Output Representations in Language Models

arXiv 2601.22040 ↗First posted 29 Jan 2026cs.CL cs.AI cs.LGcited by 2

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

  1. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
  2. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
  3. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
  4. 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,772
  5. 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
  6. 2018Deep contextualized word representationscited by 2,285
  7. 2020GLU Variants Improve Transformercited by 1,724
  8. 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 960
  9. 2016Using the Output Embedding to Improve Language Modelscited by 266
  10. 2019Representation Degeneration Problem in Training Natural Language Generation Modelscited by 250
  11. 2018Adaptive Input Representations for Neural Language Modelingcited by 226
  12. 2017Breaking the Softmax Bottleneck: A High-Rank RNN Language Modelcited by 147
3 more
  1. 2015Character-Aware Neural Language Modelscited by 129
  2. 2016Tying Word Vectors and Word Classifiers: A Loss Framework for Language Modelingcited by 87
  3. 2017Hash Embeddings for Efficient Word Representationscited by 54

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.