Efficient Pre-Training of LLMs through Truncated SVD Representations
cs.LG, cs.AI
Submitted: 2026-05-27
Updated: 2026-09-29
License: http://creativecommons.org/licenses/by-nc-sa/4.0/
The gist: LLM pretraining is extremely costly; therefore, parameter-efficient LLM architectures have recently emerged as a compelling research direction.
Terminology
Abstract
LLM pretraining is extremely costly; therefore, parameter-efficient LLM architectures have recently emerged as a compelling research direction. One such promising approach is to represent the parameters as orthonormal low-rank weight matrices. However, maintaining orthonormality during training is computationally expensive, making it impractical. This paper presents the TSVD (Truncated Singular Value Decomposition) framework which efficiently maintains orthonormality through QR decomposition and caching. Furthermore, a spectral energy heuristic is introduced to select the rank of the resulting low-rank weight matrices. Empirical evaluations across model sizes show that TSVD matches or outperforms full-parameter baselines at a fraction of the compute cost. TSVD thus provides a scalable, computationally efficient foundation for LLM pretraining.
Sources
- Unitary Evolution Recurrent Neural Networks
- Can We Gain More from Orthogonality Regularizations in Training Deep CNNs?
- Documenting Large Webtext Corpora: A Case Study on the Colossal Clean Crawled Corpus
- Full-Rank No More: Low-Rank Weight Training for Modern Speech Recognition Models
- The Llama 3 Herd of Models
- SLTrain: a sparse plus low-rank approach for parameter and memory efficient pretraining
- Training Compute-Optimal Large Language Models
- LoRA: Low-Rank Adaptation of Large Language Models
- Orthogonal Weight Normalization: Solution to Optimization over Multiple Dependent Stiefel Manifolds in Deep Neural Networks
- Scaling Laws for Neural Language Models
- Initialization and Regularization of Factorized Neural Layers
- LOST: Low-rank and Sparse Pre-training for Large Language Models
- ReLoRA: High-Rank Training Through Low-Rank Updates
- CoLA: Compute-Efficient Pre-Training of LLMs via Low-Rank Activation
- Olmo 3
- Reparameterized LLM Training via Orthogonal Equivalence Transformation
- Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer
- Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layer
- LLaMA: Open and Efficient Foundation Language Models
- Llama 2: Open Foundation and Fine-Tuned Chat Models
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks