English

Leviathan: Decoupling Input and Output Representations in Language Models

Computation and Language 2026-05-08 v2 Artificial Intelligence Machine Learning

Abstract

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×2.1\times 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.

Keywords

Cite

@article{arxiv.2601.22040,
  title  = {Leviathan: Decoupling Input and Output Representations in Language Models},
  author = {Reza T. Batley and Sourav Saha},
  journal= {arXiv preprint arXiv:2601.22040},
  year   = {2026}
}
R2 v1 2026-07-01T09:26:14.253Z