DLR: Zero-Inference-Cost Latent Residuals for Low-Rank Pre-Training
About
Large language models have driven recent progress in language and multimodal AI, yet pre-training them at scale is prohibitively expensive. Low-rank pre-training, which factorizes each weight matrix into a rank-r product to reduce both parameters and FLOPs, is a promising response but typically lags full-rank training in quality. We propose Duplicated Latent Residual (DLR), a training-only, parameter-free, foldable plug-in for low-rank pre-training. DLR augments the standard low-rank output Bz with a fixed structured residual alpha/sqrt(K) * Expand_K(z) that replicates each latent coordinate K = ceil(d_out/r) times across the output. With alpha fixed, DLR adds zero learnable parameters per layer; after training, it is absorbed into the up-projection in closed form, B* = B + alpha/sqrt(K) R^T, so deployment parameter count, FLOPs and memory match the underlying low-rank backbone exactly. Across LLaMA models from 60M to 7B parameters, DLR strengthens low-rank pre-training on C4 validation perplexity in most settings, with the clearest gains at 130M and above; folded checkpoints transfer cleanly to supervised fine-tuning on standard benchmarks.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Language Modeling | C4 (val) | PPL14.26 | 908 | |
| Question Answering | PIQA | Accuracy66.81 | 589 | |
| Sentence Completion | HellaSwag | Accuracy34.2 | 440 | |
| Commonsense Question Answering | WinoGrande | Accuracy50.69 | 89 | |
| Question Answering | ARC-C | Accuracy (ARC-C)22.29 | 67 | |
| Commonsense Reasoning | HellaSwag | HellaSwag Zero-shot Accuracy35.19 | 25 | |
| Multi-task Language Understanding | MMLU | Zero-shot Accuracy22.95 | 14 | |
| Multiple-choice Question Answering | MMLU | MMLU Accuracy22.95 | 4 | |
| Physical Interaction Question Answering | PIQA | Zero-shot Accuracy67.5 | 3 | |
| Science Question Answering | ARC Challenge | Zero-shot Accuracy23.27 | 3 |