Do Language Models Need a Trainable Input Embedding Table? Fixed Minimal Token Codes at 1.7B-Class Scale
Abstract
A trainable input embedding table assigns each vocabulary item an independently adjustable vector. We investigate whether this token-specific parameterization is required for substantial language-modeling capability, or whether a shared Transformer can learn from fixed token identities. We compare three decoder-only language models trained from scratch with the same tokenizer, contextual backbone, untied output-head architecture, and training recipe, with a target budget of 100 billion prediction tokens per model. Their input interfaces are a learned table, canonical 16-bit token-ID codes, and one fixed invertible recoding over GF(2). The fixed codes are repeated to model width without an additional trainable input projection. Both fixed-code models acquire substantial capabilities: canonical codes achieve 52.40% HellaSwag normalized accuracy, 70.51% PIQA accuracy, and 42.75% LAMBADA accuracy. The learned-input control performs better on several evaluations, including HellaSwag and LAMBADA, so these results establish viability rather than performance parity. The fixed interfaces remove 100.7 million trainable parameters, yielding 1.711B-parameter models, but parameter reduction is not the central result. These single-run experiments distinguish architectural necessity from empirical utility: independently trainable token-specific input vectors are not required for the observed capabilities. A fixed identity interface also provides a controlled setting for studying representation learning downstream of an immutable input, without establishing where particular capabilities are localized.
Figures & tables
| Quantity | Learned | Binary16 | GF2 |
|---|---|---|---|
| Trainable input parameters | 100,663,296 | 0 | 0 |
| Trainable body parameters | 1,610,713,088 | 1,610,713,088 | 1,610,713,088 |
| Trainable output parameters | 100,663,296 | 100,663,296 | 100,663,296 |
| Total trainable parameters | 1,812,039,680 | 1,711,376,384 | 1,711,376,384 |
| Persistent input-buffer entries | 0 | 786,432 | 786,704 |
| Metric | Learned | Binary16 | GF2 |
|---|---|---|---|
| HellaSwag, normalized accuracy | |||
| ARC-Easy, normalized accuracy | |||
| ARC-Challenge, normalized accuracy | |||
| PIQA, accuracy | |||
| WinoGrande, accuracy | |||
| OpenBookQA, normalized accuracy |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Metric | Learned | Binary16 | GF2 | SmolLM2-135M | SmolLM2-360M | SmolLM2-1.7B |
|---|---|---|---|---|---|---|
| HellaSwag acc | ||||||
| HellaSwag acc_norm | ||||||
| ARC-Easy acc | ||||||
| ARC-Easy acc_norm | ||||||
| ARC-Challenge acc | ||||||
| ARC-Challenge acc_norm |
| Setting | Value |
|---|---|
| Vocabulary size | 49,152 |
| Decoder blocks | 24 |
| Hidden width | 2048 |
| Attention heads | 32 |
| Head dimension | 64 |
| SwiGLU intermediate width | 8192 |
| Task group | Likelihood requests | Marked truncated |
|---|---|---|
| Multiple choice, 0-shot | 68,671 | 0 |
| MMLU, 0-shot | 56,168 | 0 |
| MMLU, 5-shot | 56,168 | 1,508 |
| LAMBADA | 5,153 | 0 |
| WikiText rolling likelihood | 173 | 0 |
LLM Training
Large language model training studies the data, objectives, optimizers, and systems used to build and adapt language models.
See all 338 papers