Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Dong Wang, Wenwu Tang, Yun Cheng, Olga Saukh• 2026

Related benchmarks

TaskDatasetResultRank
Language ModelingC4 (val)
PPL14.26
908
Question AnsweringPIQA
Accuracy66.81
589
Sentence CompletionHellaSwag
Accuracy34.2
440
Commonsense Question AnsweringWinoGrande
Accuracy50.69
89
Question AnsweringARC-C
Accuracy (ARC-C)22.29
67
Commonsense ReasoningHellaSwag
HellaSwag Zero-shot Accuracy35.19
25
Multi-task Language UnderstandingMMLU
Zero-shot Accuracy22.95
14
Multiple-choice Question AnsweringMMLU
MMLU Accuracy22.95
4
Physical Interaction Question AnsweringPIQA
Zero-shot Accuracy67.5
3
Science Question AnsweringARC Challenge
Zero-shot Accuracy23.27
3
Showing 10 of 11 rows

Other info

Follow for update