Kronecker Embeddings: Byte-Level Structured Token Representations for Parameter-Efficient Language Models
Not on the map. Why?
Abstract (arXiv)
Large language models route every input through a learned embedding table of shape |V| x d_model, consuming hundreds of millions to billions of trainable parameters at frontier scale. We introduce Kronecker Embeddings, a deterministic byte-level character-position factorization that replaces this table with a fixed encoder and a single learned projection, compatible with standard BPE tokenizers, eliminating 91--94% of input-side trainable parameters at frontier scale. We provide five contributions. First, a cross-model probe across six LMs (135M-671B parameters) shows trained input embeddings cluster typographic variants of the probe word far more than morphological relatives; Kronecker escapes this clustering at the embedding layer. Second, a controlled three-seed comparison on nanoGPT GPT-2 124M over 2.5B tokens of FineWeb-Edu shows Kronecker reaching 2.5 +- 0.2% lower validation loss than the BPE-tied baseline (gap 0.083 +- 0.007 nats, ~9% lower perplexity), needing ~1.43x fewer steps to reach BPE's converged loss. Third, a spelling-robustness probe over 110 clean/typo pairs shows Kronecker preserves the top-1 prediction on 55.5% of pairs vs. 47.3% for BPE (+8.2 pp) and lowers KL by 7.6%, winning or tying in 10 of 11 categories; a generation probe shows Kronecker echoes byte-novel strings and typos through generation where BPE forgets them. Fourth, BPE embedding norm drifts during training while Kronecker projection norm stays near 1.0, consistent with a stable representational target. Fifth, an on-the-fly runtime variant reconstructs embeddings from a 4.5 MB byte buffer rather than a 2.15 GB table at vocabulary 131,072, with 0.01--0.24% step-time overhead. Byte-level locality has a tradeoff: byte-similar but semantically distant pairs (compute/commute, nation/notion) cluster together, shifting disambiguation to early attention layers.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 22 of its 28 references that are in the map. First posted on arXiv 28 May 2026.
Builds on
22 of 28 references.
References · 22
- 2017Attention Is All You Needcited by 60,361
- 2025Qwen3 Technical Reportcited by 8,820
- 2015Neural Machine Translation of Rare Words with Subword Unitscited by 2,918
- 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,790
- 2018SentencePiece: A simple and language independent subword tokenizer and detokenizer for Neural Text Processingcited by 1,989
- 2022GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformerscited by 1,919
- 2025Gemma 3 Technical Reportcited by 1,824
- 2025gpt-oss-120b & gpt-oss-20b Model Cardcited by 1,386
- 2019How Contextual are Contextualized Word Representations? Comparing the Geometry of BERT, ELMo, and GPT-2 Embeddingscited by 784
- 2021ByT5: Towards a token-free future with pre-trained byte-to-byte modelscited by 509
- 2025SmolLM2: When Smol Goes Big -- Data-Centric Training of a Small Language Modelcited by 335
- 2016Using the Output Embedding to Improve Language Modelscited by 272
10 more
- 2019Representation Degeneration Problem in Training Natural Language Generation Modelscited by 255
- 2021CANINE: Pre-training an Efficient Tokenization-Free Encoder for Language Representationcited by 232
- 2017All-but-the-Top: Simple and Effective Postprocessing for Word Representationscited by 212
- 2023MEGABYTE: Predicting Million-byte Sequences with Multiscale Transformerscited by 170
- 2021Charformer: Fast Character Transformers via Gradient-based Subword Tokenizationcited by 160
- 2024Byte Latent Transformer: Patches Scale Better Than Tokenscited by 146
- 2015Character-Aware Neural Language Modelscited by 132
- 2021Between words and characters: A Brief History of Open-Vocabulary Modeling and Tokenization in NLPcited by 132
- 2024MYTE: Morphology-Driven Byte Encoding for Better and Fairer Multilingual Language Modelingcited by 40
- 2026Weight Tying Biases Token Embeddings Towards the Output Spacecited by 3
Built on by
Citing papers per year: 2026: 1.
1 paper.