Review date: 2026-06-19 Review author: Zhongzhu Zhou Paper reviewed: “The Truth is in There: Improving Reasoning in Language Models with Layer-Selective Rank Reduction” Paper authors: Pratyusha Sharma, Jordan T. Ash, Dipendra Misra arXiv: 2312.13558 Venue: ICLR 2024
Short Answer
Here is the central surprise of this paper: you take a 7-billion-parameter language model, find one of its weight matrices deep in the network, replace it with a rank-40 approximation (out of a possible rank of 4096), and the model starts answering factual questions 18 percentage points more accurately — with zero additional training.
This is LASER, which stands for LAyer-SElective Rank-Reduction. The idea is almost embarrassingly simple: compute the SVD of a chosen weight matrix, keep only the top- singular components, discard everything else, and plug the compressed matrix back into the model. No gradients, no data, no training loop. Just a handful of matrix multiplications.
The result challenges a deep assumption in the field — that a model’s trained weights are precious and that removing information from them is harmful. LASER shows that for certain layers and certain tasks, the opposite is true: the weight matrices are storing information that actively hurts performance. Removing it, rather than adding to it, is the fix. This review unpacks how the method works mechanically, why the authors believe it works conceptually, how well it actually performs across benchmarks, and — critically — where it falls short.
Prerequisites: What You Need to Know First
Before diving into the method, let us build up the mathematical and architectural background. This section covers three things: Singular Value Decomposition (SVD), how transformers store and process information in their weight matrices, and the idea of low-rank approximation as a way to capture the “most important” directions in a matrix.
Singular Value Decomposition (SVD) From First Principles
Every real matrix can be written as a product of three matrices:
where:
- is an orthogonal matrix (columns are orthonormal: ) whose columns are called the left singular vectors
- is a diagonal matrix (or block-diagonal with a zero block when ) with non-negative diagonal entries , called the singular values, where
- is an orthogonal matrix whose columns are the right singular vectors
This decomposition always exists and is unique (up to sign flips and handling of repeated singular values). Geometrically, rotates the input space, scales along the axes, and rotates into the output space.
Equivalent outer-product form. An equivalent and more intuitive form expands the product as a sum of rank-1 matrices:
Each term is a rank-1 matrix scaled by . The singular values are sorted in decreasing order, so the first few terms dominate the matrix — they capture the “coarse structure.” Later terms, with small , capture fine detail, high-frequency variation, or noise.
Computing SVD in practice. Modern libraries (NumPy, PyTorch) compute SVD efficiently using the Golub-Reinsch algorithm or randomized methods. For a weight matrix in a large transformer (e.g., 4096 × 4096), full SVD takes time. In practice, PyTorch’s torch.linalg.svd is used in the LASER codebase.
# LASER's core SVD computation (simplified from matrix_utils.py)
def low_rank_approximation(W, rank_fraction):
U, S, Vt = torch.linalg.svd(W, full_matrices=False)
k = int(rank_fraction * min(W.shape)) # number of singular values to keep
# Keep only top-k components
U_k = U[:, :k]
S_k = S[:k]
Vt_k = Vt[:k, :]
return U_k @ torch.diag(S_k) @ Vt_k # W_k: the low-rank approximation
The Eckart–Young Theorem: Why Truncated SVD is Optimal
A key mathematical fact that justifies using truncated SVD is the Eckart–Young theorem (1936). It states:
Among all rank- matrices , the one that minimizes the Frobenius norm distance to is the truncated SVD .
More precisely:
The reconstruction error is exactly the sum of squared discarded singular values:
This theorem holds not just for the Frobenius norm but also for the spectral (operator) norm and any unitarily invariant norm. It means that truncated SVD is not just a heuristic — it is the provably optimal way to compress a matrix to a given rank.
flowchart LR
A["W ∈ ℝ^(m×n)\nFull weight matrix"] --> B["SVD\nW = UΣVᵀ"]
B --> C["σ₁ ≥ σ₂ ≥ … ≥ σᵣ\nSingular values sorted"]
C --> D["Keep top-k\nW_k = U_k Σ_k Vk^T"]
D --> E["Eckart-Young:\nBest rank-k approx\nin Frobenius norm"]
C --> F["Discard σ_{k+1}…σ_r\n'Higher-order components'"]
F --> G["Error = Σᵢ₌ₖ₊₁ σᵢ²\n(can be large if σᵢ small)"]
style D fill:#4CAF50,color:#fff
style F fill:#FF5722,color:#fff
Figure 1: The truncated SVD pipeline. The Eckart-Young theorem guarantees that is the closest rank- matrix to under the Frobenius norm.
Transformer Weight Matrices: A Map of Where Information Lives
A transformer layer consists of two main sub-modules: a multi-head self-attention (MHSA) block and a feed-forward network (FFN/MLP) block. Each has several weight matrices:
Attention weights (in a model with hidden dimension and heads):
- Query projection:
- Key projection:
- Value projection:
- Output projection:
MLP weights (standard 2-layer FFN with intermediate dimension ):
- First linear: (fc_in)
- Second linear: (fc_out)
- (For SwiGLU/LLaMA variants) Gate: (fc_up)
Research over the past few years (notably Geva et al. 2021, Meng et al. 2022 ROME) has established that factual knowledge and world-model information is predominantly stored in the MLP layers, specifically in and . The attention layers are better thought of as routing and retrieval mechanisms, while the MLP layers act as “key-value memories.”
flowchart TD
subgraph "Transformer Layer l"
X["Input x_l ∈ ℝ^d"]
ATTN["Multi-Head Attention\nW_Q, W_K, W_V, W_O\n(routing & retrieval)"]
NORM1["LayerNorm"]
MLP["MLP Block\nW₁ (fc_in): d→d_ff\nW₂ (fc_out): d_ff→d\n(knowledge storage)"]
NORM2["LayerNorm"]
OUT["Output x_{l+1}"]
end
X --> NORM1 --> ATTN --> ADD1["+"] --> NORM2 --> MLP --> ADD2["+"] --> OUT
X --> ADD1
NORM1 --> ADD2
subgraph "LASER Target"
T1["LASER typically targets\nW₁ or W₂ in late MLP blocks\n(τ = fc_in or fc_out)"]
end
MLP -.-> T1
style T1 fill:#FF9800,color:#000
style MLP fill:#2196F3,color:#fff
Figure 2: Architecture of a transformer layer. LASER primarily targets the MLP weight matrices in late layers, consistent with the finding that factual knowledge is stored in FFN layers.
Rank Deficiency and Information in Neural Networks
A surprising empirical fact is that large trained weight matrices are often effectively low-rank: most of their singular values are tiny, with only the top few carrying significant information. This phenomenon is exploited by methods like LoRA for parameter-efficient fine-tuning, which adds only low-rank update matrices.
LASER takes the opposite viewpoint: rather than adding low-rank components, it removes the high-rank (small-singular-value) components. The question is: which components are the “signal” and which are “noise”?
Motivation: Why Would Rank Reduction Help?
The standard view of model weights is that they encode learned knowledge and any reduction degrades performance. What reason could there be to expect the opposite?
Overfitting to spurious patterns. During pretraining on large web corpora, LLMs are exposed to vast amounts of text that includes noise, contradictions, stereotypes, and misleading co-occurrences. The model may learn to associate certain tokens with spurious patterns (e.g., a question that contains a proper noun may cause the model to pattern-match toward factual recall rather than logical reasoning). These spurious associations may be encoded in the higher-order (small ) components of weight matrices because:
- Robust, high-frequency patterns are captured in the top singular directions (high )
- Idiosyncratic, task-specific, or noisy patterns get relegated to smaller singular directions
The “knowledge bottleneck” view. Another perspective: a model may contain more task-relevant knowledge than it can easily “access” due to interfering representations. Removing the high-rank noise creates a cleaner bottleneck that forces the model to rely on robust, generalizable features — effectively acting as a post-hoc regularizer.
Relation to model editing. ROME (Meng et al. 2022) showed that specific factual associations can be localized to specific MLP layers. LASER can be seen as a complementary finding: beyond individual facts, the quality of reasoning is affected by the accumulated noise in MLP weight matrices, particularly in later layers.
The LASER Method
Formal Definition
A single LASER intervention is defined by three hyperparameters:
where:
- is the parameter type — which matrix to target
- is the layer index ( = total number of transformer layers)
- is the rank retention fraction — what fraction of the maximum rank to keep ( components are retained)
The intervention replaces the target weight matrix with its rank- truncated SVD:
where , , are the top- components of the SVD of .
Multiple interventions can be composed by applying them independently on different layers or parameter types.
Step-by-Step Algorithm
Algorithm 1: Single LASER Intervention
Input:
LLM with weight matrices {W^(l,τ) : l = 0..L-1, τ ∈ Params}
Hyperparameters (τ*, l*, ρ*)
Target task with validation set D_val
Step 1: Load pre-trained LLM weights (no fine-tuning needed)
Step 2: Extract the target weight matrix:
W ← W^(l*, τ*) ∈ ℝ^(m × n)
Step 3: Compute full SVD:
U, S, Vt ← svd(W, full_matrices=False)
# U ∈ ℝ^(m×m), S ∈ ℝ^min(m,n), Vt ∈ ℝ^(n×n)
Step 4: Compute rank k:
k ← floor(ρ* × min(m, n))
Step 5: Truncate to top-k components:
U_k ← U[:, :k] ∈ ℝ^(m × k)
S_k ← diag(S[:k]) ∈ ℝ^(k × k)
Vt_k ← Vt[:k, :] ∈ ℝ^(k × n)
Step 6: Reconstruct low-rank approximation:
W_k ← U_k @ S_k @ Vt_k ∈ ℝ^(m × n)
Step 7: Replace in the model:
W^(l*, τ*) ← W_k
Step 8: Evaluate on D_val → accuracy(l*, τ*, ρ*)
Output: Modified LLM with W^(l*, τ*) replaced by W_k
Algorithm 2: Hyperparameter Search
Input: LLM, task T with split D_val (20%) / D_test (80%)
Search space: τ ∈ Params, l ∈ {0..L-1}, ρ ∈ {0.01, 0.1, 0.2, 0.4, 0.8, 0.9, 0.99}
best_acc ← base model accuracy on D_val
best_params ← None
for each τ, l, ρ in search space:
modified_LLM ← apply_LASER(LLM, τ, l, ρ)
acc_val ← evaluate(modified_LLM, D_val)
if acc_val > best_acc:
best_acc ← acc_val
best_params ← (τ, l, ρ)
if best_params is not None:
final_LLM ← apply_LASER(LLM, best_params)
return evaluate(final_LLM, D_test)
else:
return evaluate(base_LLM, D_test) # no improvement found
The search space is not exhaustive — the paper typically scans a grid of values and sweeps layers. The number of SVD computations required is , which is affordable since SVD of a weight matrix (e.g., for LLaMA-2) takes a few seconds on GPU.
Computational Cost Analysis
SVD computation cost. For a matrix with :
- Full SVD via Golub-Reinsch: time
- For a typical LLaMA-2 weight matrix (): ~ billion FLOPs — expensive but done once offline
Memory cost after compression. If we store in factored form:
- Storage: floats vs. for the original
- For , this is floats vs — a 73× reduction
However, the current LASER codebase does not store in factored form — it stores the full reconstructed matrix. This means no memory savings at inference time. The authors acknowledge this gap and mark it as future work.
flowchart LR
subgraph "Memory Tradeoff"
A["Original W\nm×n floats\n(e.g., 4096×11008 = 45M)"]
B["Factored form\nU_k + S_k + Vt_k\nk(m+n) floats\n(k=41 → 619K, 73× smaller)"]
C["Reconstructed W_k\nSame m×n floats\n(current LASER code)"]
end
A --> B
A --> C
B -. "NOT implemented\nin current code" .-> D["73× memory savings\nat inference"]
C --> E["No memory savings\nbut same compute graph"]
style B fill:#4CAF50,color:#fff
style D fill:#9E9E9E,color:#fff
style C fill:#FF5722,color:#fff
Figure 3: Memory implications of LASER. The factored form would yield 73× memory savings, but the current implementation reconstructs the full matrix and stores it, gaining no memory benefit.
The Noise Hypothesis: Why Does This Work?
The key question is: why does removing the higher-order singular components improve performance? The authors propose the noise hypothesis:
Central claim. The higher-order components of MLP weight matrices (those corresponding to small singular values) predominantly encode:
- Task-spurious memorization (co-occurrence patterns in pretraining data that are not causally related to the reasoning task)
- Over-fitted idiosyncratic patterns that interfere with generalization
- Statistical noise from the data distribution
The lower-order components (large singular values) encode:
- Robust, compositional patterns (semantics, syntax, logical structure)
- General-purpose feature detectors
- The “true” signal that underlies correct reasoning
Supporting evidence from the experiments:
- Improvements are much larger for QA and reasoning tasks (where spurious associations are harmful) than for straightforward retrieval
- The effective rank at which improvements peak is extremely low (, i.e., 1% of rank), suggesting that the “real” useful information is concentrated in very few singular directions
- Improvements are concentrated in late MLP layers (the last few transformer layers), consistent with the view that early layers extract features while late layers perform task-specific reasoning and association
The signal-vs-noise decomposition. We can think of the weight matrix as:
When the model processes an input , the output of layer is (ignoring residual connections for simplicity):
The second term adds “noise” to the hidden representation that may push the model toward wrong answers. LASER eliminates this term by substituting .
flowchart TD
subgraph "Information content of a weight matrix W"
A["W (full rank r)"]
B["Top-k components\nσ₁…σ_k (large)\n→ Robust features\n→ General patterns\n→ Compositional semantics"]
C["Higher-order components\nσ_{k+1}…σ_r (small)\n→ Spurious co-occurrences\n→ Dataset-specific noise\n→ Memorized idiosyncrasies"]
end
A --> B
A --> C
B --> D["Helps correct reasoning"]
C --> E["Interferes with reasoning\n→ LASER removes this"]
style B fill:#4CAF50,color:#fff
style C fill:#FF5722,color:#fff
style E fill:#FF5722,color:#fff
style D fill:#4CAF50,color:#fff
Figure 4: The noise hypothesis. Higher-order singular components (small ) encode spurious patterns that interfere with reasoning. LASER removes them, leaving only the robust signal.
Alternative explanations. The authors also consider two other hypotheses:
-
Regularization hypothesis: Rank reduction acts like regularization, preventing the weight from fitting task-irrelevant patterns. This is consistent with the data but hard to distinguish from the noise hypothesis.
-
Implicit data augmentation: By making the weight matrix “smoother” (lower rank), the model generalizes better to the test distribution even without seeing new data.
Neither alternative is ruled out. The paper acknowledges the mechanism is not fully understood.
Experimental Setup
Models Evaluated
| Model | Size | Architecture | Layers |
|---|---|---|---|
| RoBERTa | 355M | Encoder-only BERT variant | 12 |
| GPT-J | 6B | Decoder-only, 28 layers | 28 |
| LLaMA-2 | 7B | Decoder-only, 32 layers, SwiGLU | 32 |
Benchmarks
| Dataset | Task Type | Description |
|---|---|---|
| CounterFact | Factual recall QA | Multiple-choice, tests factual associations |
| HotpotQA | Multi-hop reasoning | Requires reasoning over multiple facts |
| FEVER | Fact verification | Binary claim verification |
| Bios-Gender | Demographic prediction | Predicting gender from professional bio |
| Bios-Profession | Occupation prediction | Predicting profession from bio |
| TruthfulQA | Truthfulness | Tests whether model gives truthful vs plausible answers |
| BigBench-Epistemic | Epistemic reasoning | Logical reasoning about beliefs and knowledge |
| BigBench-WikidataQA | Factual retrieval | Multi-choice QA using Wikidata triples |
Evaluation Protocol
- Split: 20% of each dataset as validation (for hyperparameter search), 80% as test set
- Metric: Accuracy via log-likelihood: the answer with the highest sequence probability under the model wins
- Baseline comparison: Base model (no intervention) and random low-rank projection (random orthogonal basis at same rank as optimal LASER)
- LASER grid search: Over , fc_in, fc_out, fc_up, mlp, k_proj, v_proj, q_proj, out_proj,
Key Results
Headline Numbers
The table below reproduces the key results from the LASER website leaderboard (vanilla LASER, from the paper):
| Model | Dataset | Base Accuracy | LASER Accuracy | Gain (pp) | Best |
|---|---|---|---|---|---|
| GPT-J (6B) | CounterFact | 13.1% | 24.0% | +10.9 | (Uin, 27, 0.01) |
| GPT-J (6B) | Bios-Gender | 70.9% | 97.5% | +26.6 | (Uin, 14, 0.01) |
| GPT-J (6B) | Bios-Profession | 75.6% | 82.1% | +6.5 | (Uin, 18, 0.01) |
| GPT-J (6B) | FEVER | 50.2% | 56.2% | +6.0 | (Uin, 24, 0.01) |
| GPT-J (6B) | BigBench-WikidataQA | 51.8% | 65.9% | +14.1 | (Uin, 27, 0.01) |
| LLaMA-2 (7B) | CounterFact | 35.6% | 37.6% | +2.0 | (Uin, 28, 0.05) |
| LLaMA-2 (7B) | Bios-Gender | 75.5% | 88.4% | +12.9 | (Uin, 24, 0.01) |
| LLaMA-2 (7B) | FEVER | 59.3% | 64.5% | +5.2 | (Uin, 30, 0.2) |
| LLaMA-2 (7B) | TruthfulQA | 50.5% | 56.2% | +5.7 | (Uin, 30, 0.05) |
| LLaMA-2 (7B) | BigBench-Epistemic | 44.8% | 63.4% | +18.6 | (Uout, 28, 0.01) |
| LLaMA-2 (7B) | BigBench-WikidataQA | 59.5% | 62.0% | +2.5 | (Uin, 27, 0.01) |
| RoBERTa (355M) | CounterFact | 17.3% | 19.3% | +2.0 | (Uin, 8, 0.8) |
| RoBERTa (355M) | Bios-Profession | 64.5% | 72.5% | +8.0 | (Uin, 3, 0.9) |
Dominant patterns:
- (keeping only 1% of rank) is optimal in the vast majority of cases
- (the first MLP layer / fc_in) is consistently targeted
- The optimal layer is almost always in the last 25% of the network (layers 24–30 for LLaMA-2/32 layers; 26–27 for GPT-J/28 layers)
flowchart TD
subgraph "LASER Accuracy Gain (GPT-J 6B, pct points)"
CF["CounterFact\n+10.9 pp\n13.1pct to 24.0pct"]
BG["Bios-Gender\n+26.6 pp (best)\n70.9pct to 97.5pct"]
BP["Bios-Profession\n+6.5 pp\n75.6pct → 82.1pct"]
FV["FEVER\n+6.0 pp\n50.2pct → 56.2pct"]
WD["BigBench-Wikidata\n+14.1 pp\n51.8pct → 65.9pct"]
end
style BG fill:#4CAF50,color:#fff
style WD fill:#66BB6A,color:#000
style CF fill:#FFA726,color:#000
style BP fill:#FFA726,color:#000
style FV fill:#FFA726,color:#000
flowchart LR
subgraph "LASER vs Baselines (LLaMA-2 7B)"
A["BigBench-Epistemic\nBase: 44.8pct\nRandom rank-reduction: ~44pct\nLASER: 63.4pct (best)"]
B["Bios-Gender\nBase: 75.5pct\nRandom rank-reduction: ~75pct\nLASER: 88.4pct (best)"]
C["TruthfulQA\nBase: 50.5pct\nRandom rank-reduction: ~50pct\nLASER: 56.2pct (best)"]
end
style A fill:#4CAF50,color:#fff
style B fill:#4CAF50,color:#fff
style C fill:#4CAF50,color:#fff
Figure 5: LASER vs baselines. Random low-rank projection (same rank, random orthogonal basis) does NOT improve performance — the SVD-based truncation is what matters, not merely reducing rank.
Evaluation Metric Deep Dive: Log-Likelihood Ranking
All experiments use a log-likelihood ranking protocol for evaluation, which is worth understanding in detail. For a -way multiple-choice question with candidate answers :
where .
This metric has an important property: it is purely about the relative ordering of candidates, not about the model generating the correct answer without a hint. A model that generates fluent-but-wrong answers to open-ended questions might score well on log-likelihood ranking if the correct answer still has higher probability than the distractors. This is a fundamentally easier task than open-ended generation.
The consequence for interpreting LASER results: the measured improvements (+2pp to +27pp) may not translate directly to equivalent improvements on open-ended generation benchmarks. LASER is changing which of several pre-specified options the model prefers — not whether the model can generate the correct answer from scratch.
The Crucial Ablation: Random vs. SVD-Based Truncation
One of the paper’s most important controls is comparing LASER to random rank reduction — using a random orthogonal basis instead of the SVD-based one, at the same target rank . If simply reducing rank were responsible for the improvement, both would work equally. The result: random rank reduction gives no improvement, sometimes degrading performance. The improvement comes specifically from choosing the SVD directions — i.e., the particular choice of which components to keep matters deeply.
This rules out simple regularization-as-compression as the sole explanation and supports the view that there is something special about the specific low-rank structure revealed by SVD.
Layer-Level Analysis: Why Late Layers?
A systematic sweep over layers reveals that LASER improvements are strongly concentrated in the final 20–30% of transformer layers. Early layers (embedding + initial processing) show little or no improvement when rank-reduced; middle layers show mixed results; late layers (especially the last 2–5) are where the biggest gains occur.
This aligns with a mechanistic view of transformer computation:
- Early layers extract low-level features (token identities, syntax)
- Middle layers build up contextualized representations
- Late layers perform final “decision-making” for the next-token prediction — and this is where task-specific noise is most concentrated and harmful
The finding also connects to model editing research: ROME (Meng et al. 2022) showed that factual associations can be inserted or removed by editing MLP weights in mid-to-late layers. LASER’s effectiveness in late layers suggests a related phenomenon — the late layers accumulate spurious task-specific associations that can be cleaned up by low-rank approximation.
Deep Dive: The Singular Value Spectrum of Transformer Weights
To truly understand why LASER works, it helps to examine what the singular value distribution of transformer weight matrices actually looks like in practice. This section provides a detailed analysis.
Typical Spectral Shape in Pretrained LLMs
The singular values of weight matrices in well-trained LLMs exhibit a characteristic heavy-tailed distribution: a few very large singular values, followed by a rapid drop-off, then a long tail of many small but non-zero values.
Concretely, for a weight matrix in LLaMA-2-7B (e.g., , ), the spectrum looks approximately like:
- Top-1% (41 components): values in the range
- Top-10% (410 components): values in the range
- Remaining 90% (3686 components): values in the range — a long, nearly flat tail
The effective rank — measured by the entropy of the normalized squared singular values, where — is typically much smaller than the matrix’s true rank. For many MLP weight matrices in pretrained LLMs, –, meaning the effective “intrinsic dimensionality” is 5–15% of the full rank.
This is not coincidental — it reflects the low-dimensional structure of natural language, combined with gradient-descent training that tends to concentrate information in a relatively small subspace.
Why Small Singular Values Are “Noise”
During training with stochastic gradient descent, the model updates its weights based on batches of data. These updates have a noise component from mini-batch sampling. Over millions of update steps:
- High-signal directions are reinforced across many training examples → they accumulate large singular values
- Noise directions receive conflicting updates from different examples → they drift randomly, producing small but non-zero singular values
The result is that the small singular values of are roughly proportional to the effective noise floor of training. The optimal rank for inference ( empirically) may thus be reflecting the genuine intrinsic dimensionality of the task-relevant information.
Quantifying How Much Information is Retained
The fraction of the Frobenius norm (equivalently, the squared variance) retained by the top- components is:
For the heavy-tailed distribution typical in LLMs, with (keeping 1% of components):
This means keeping only the top-1% of singular components still retains 60–80% of the Frobenius norm! The long tail of small singular values has high count but collectively low norm, which is why extreme truncation still preserves most of the “mass” of the matrix while removing the noisy tail.
flowchart LR
subgraph "Singular Value Spectrum (Schematic)"
A["Top 1pct (k~41)\nLarge singular values: 50-200\nRetains ~70pct variance"]
B["Top 10pct (k~410)\nMedium singular values: 5-50\nAdditional ~20pct variance"]
C["Bottom 90pct (k~3686)\nSmall singular values: 0-5\nOnly ~10pct variance\nLASER removes this"]
end
style A fill:#4CAF50,color:#fff
style B fill:#FF9800,color:#000
style C fill:#FF5722,color:#fff
Figure 6: Schematic of the singular value spectrum for a typical LLaMA-2 MLP weight matrix. The top 1% of components retains ~70% of the Frobenius norm, explaining why extreme truncation () is viable.
Connection to LoRA and the Low-Rank Paradigm
LASER belongs to a broader family of methods that exploit the low-rank structure of transformer weights. Understanding how it relates to LoRA and other low-rank methods reveals both its uniqueness and its limitations.
LoRA vs. LASER: Opposite Directions
LoRA (Hu et al. 2022) and LASER both use low-rank matrix factorizations, but their approaches are diametrically opposite:
| Aspect | LoRA | LASER |
|---|---|---|
| When applied | During fine-tuning | After pretraining (no training) |
| What is modified | Adds to existing | Replaces with |
| Data required | Full fine-tuning dataset | Small validation set for HP search |
| Direction of change | Adds low-rank update | Removes high-rank components |
| Motivation | Parameter efficiency | Noise removal |
| Memory footprint | Adds params per layer | Replaces with (if factored) |
A key insight: LoRA adds a low-rank update where and , with . This update has exactly rank by construction. LASER, on the other hand, replaces with , which retains the largest singular components.
The difference in direction is illuminating: LoRA assumes the pre-trained is a good starting point and adds task-specific refinement. LASER assumes already contains all necessary task-relevant information, but also contains interfering noise that should be removed.
Connection to Spectral Regularization
LASER can be interpreted as a form of spectral regularization applied post-hoc. In the training literature, spectral regularization penalizes large Frobenius norms or nuclear norms of weight matrices, encouraging low-rank solutions during training. LASER achieves a similar effect after training by hard-thresholding the singular values.
This connection suggests an interesting alternative: could spectral regularization during pretraining reduce or eliminate the need for LASER afterward? If the noise hypothesis is correct, enforcing a lower effective rank during training might produce models that are cleaner and more capable without post-hoc intervention. This is an important open research direction not addressed in the paper.
AdaLoRA and Dynamic Rank
AdaLoRA (Zhang et al. 2023) allocates different ranks to different layers and parameters based on their importance, using SVD-based adaptive rank selection during fine-tuning. This connects to LASER’s finding that different layers have very different optimal values. If LASER found for some layers and for others, an adaptive approach to rank selection during fine-tuning (as in AdaLoRA) would naturally align with LASER’s post-hoc findings.
When Does LASER Work? Conditions for Success
Based on the experimental results, several empirical conditions predict when LASER is likely to succeed:
Condition 1: Late-layer MLP target. LASER consistently works on the second-to-last or last few MLP layers. Early layers provide little or no benefit, and sometimes degrade performance.
Condition 2: Extreme rank reduction. Optimal is almost always — not , not . This extreme compression removes far more than just “noise” by any conventional definition. It suggests the task-relevant information is concentrated in very few dimensions.
Condition 3: The task involves multi-step reasoning or fact retrieval. Benchmarks like BigBench-Epistemic and CounterFact (which require multi-hop reasoning or specific factual recall) show the largest gains. Simpler tasks (like TruthfulQA) show more modest improvements, possibly because they don’t require the same depth of compositional reasoning that the noise interferes with.
Condition 4: The model family matters. GPT-J shows larger gains on most benchmarks than LLaMA-2. This may reflect architectural differences (GPT-J uses parallel attention+MLP while LLaMA-2 uses sequential with SwiGLU and RoPE). The spectral properties of weight matrices differ across architectures.
Condition 5: Absence of strong instruction tuning. The experiments use base pretrained models. Instruction tuning and RLHF align the model specifically for helpfulness and factual accuracy — they may already “clean up” some of the noise that LASER targets. Whether LASER adds value on top of instruction tuning is unknown.
Reproducibility Notes
The LASER codebase is open source under the MIT license at github.com/pratyushasharma/laser. However, reproducing the results requires attention to several practical details:
1. Dataset access. CounterFact requires a separate download script (scripts/get_counterfact.py). All other datasets are on HuggingFace. Some BigBench tasks require the bigbench package which has its own installation requirements.
2. Model download. GPT-J-6B and LLaMA-2-7B require HuggingFace access. LLaMA-2 requires a Meta AI access request. GPT-J is freely available.
3. Parameter mapping. Each model requires a custom mapping from the LASER parameter type names (fc_in, fc_out, etc.) to the actual PyTorch parameter names in the HuggingFace implementation. For LLaMA-2, this mapping is provided in src/laser/llama2_laser.py. For a new model, this mapping must be created manually.
4. Hyperparameter search. The paper uses a specific grid of values. Reproducing the exact results requires running this grid and evaluating on the 20% validation split. Minor differences in the validation split (random seed) can shift the optimal hyperparameters.
5. GPU memory requirements. Loading LLaMA-2-7B in FP16 requires ~14 GB GPU memory. Running the full SVD and evaluation requires an additional ~8 GB, so a 24 GB GPU (e.g., A5000, RTX 3090) is needed. GPT-J-6B has similar requirements.
# Minimal example to reproduce LASER on GPT-J + FEVER
# (adapted from the LASER codebase)
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
from laser.LaserWrapper import LaserWrapper
model_name = "EleutherAI/gpt-j-6B"
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16)
tokenizer = AutoTokenizer.from_pretrained(model_name)
# Apply LASER: fc_in at layer 24, keep 1% of rank
laser = LaserWrapper(model)
laser.apply_laser(
layer_num=24,
param_name="fc_in",
rank_fraction=0.01
)
# Evaluate on FEVER validation set
# ... (dataset loading and evaluation code)
Generalization Beyond LLMs: Decision Transformers
To test whether the finding is specific to language models, the authors apply LASER to Decision Transformers (DT) — transformer-based agents for offline reinforcement learning. A Decision Transformer predicts actions conditioned on state history and target return; it has no language modeling objective.
The result: LASER also improves DT performance on RL tasks. This suggests the phenomenon is a property of transformer training dynamics in general, not specific to language modeling or next-token prediction.
This is an important negative result in a positive direction — it generalizes the finding and increases confidence that something fundamental about transformer weight structure is at play, not merely an artifact of pretraining data distribution.
Limitations and Boundary Conditions
Several important limitations constrain where and how LASER can be applied:
1. Task-specific hyperparameter tuning is required. LASER is not zero-shot. Choosing requires a labeled validation set for the target task. This is a significant practical limitation: for a new task, you must first collect/annotate a small validation set and run the sweep before benefiting from LASER.
2. The sweep is cheap but not free. Running SVD on each target matrix plus evaluating the model on the validation set requires compute. For LLaMA-2-7B, with ~32 layers × ~8 parameter types × ~8 rank values = ~2048 evaluated configurations, each requiring a forward pass over the validation set. This is manageable but non-trivial.
3. No theoretical guarantees. The noise hypothesis is a post-hoc explanation. There is no mathematical theorem predicting when LASER will help or how much. The method fails for some model-task combinations.
4. Tested only on small models (≤ 7B). It is unclear whether the findings scale to 70B, 140B, or frontier-scale models. The distribution of singular values in larger models may differ significantly, and the task-specificity of later layers may be more complex.
5. Only classification/ranking tasks are evaluated. All benchmarks are multiple-choice or binary classification. LASER’s effect on open-ended generation (free-form text generation) is unexplored.
6. No memory efficiency implemented. Despite the theoretical memory savings from factored-form storage, the current codebase stores the full reconstructed matrix, providing no inference speedup or memory reduction.
7. Improvements are highly variable. Gains range from +2 to +27 percentage points across settings. There is no reliable predictor of how much improvement to expect before running the sweep.
8. Evaluation scope is narrow. Every benchmark uses log-likelihood ranking over a fixed set of candidate answers. This is a restricted form of evaluation that does not capture whether the model generates better free-form text, reasons more coherently across long contexts, or performs better in interactive scenarios. The improvements in log-likelihood ranking may not translate directly to real-world improvements in how users experience the model.
9. Dataset-specific hyperparameters may not transfer. The paper searches for (, , ) on a per-task basis. It is not clear whether the optimal parameters for one task (say, CounterFact) transfer to a closely related task (say, WikidataQA). If they don’t, the practical utility of LASER is limited to settings where task labels are available.
10. The effect on multi-task performance is unknown. Applying LASER optimized for task A may hurt performance on task B. There is no multi-task LASER analysis in the paper, making it unclear whether LASER is usable in systems that must serve multiple tasks simultaneously.
The LASER Mechanism: A Worked Numerical Example
To make the SVD truncation concrete, let us trace through a small numerical example that mirrors what happens inside a transformer MLP layer.
Suppose we have a tiny weight matrix (in practice, these are , but the principle is the same):
The singular values might look like: , , .
With (keeping 1 out of 3 components — roughly 1% for a large matrix):
The reconstruction error is . Most of this error comes from , not the small . In a real model, and beyond have tiny values () while and are in the range . So keeping only the top-1% captures the dominant variance.
What happens to the MLP output? Given an input , the MLP computes:
- Original: , which includes the projection onto all singular directions
- After LASER: , which projects onto only the top- right singular vectors and reconstructs the output in the top- left singular directions
The discarded terms represent the “noise contribution” — the part of the hidden representation derived from spurious associations. Removing it leaves a cleaner representation for subsequent layers.
Layer-by-Layer Propagation of LASER Effects
An important subtlety: LASER modifies a single weight matrix at layer , but the effects propagate through all subsequent layers. The output of layer feeds into layer , and so on. This means the “noise removal” has compounding effects: by cleaning up the representation at layer , the model’s computation in layers through is also affected.
This propagation is likely why late layers show the largest improvements from LASER: when you apply LASER at the last MLP layer (), you directly affect the final prediction without the compounding of further noise. When you apply it at an early layer, the cleaned representation must still pass through many noisy late layers, reducing the benefit.
Formally, if represents the composition of transformer computations from layer to the output, and is replaced by , the output changes by:
The sensitivity of the output to changes in is captured by the Jacobian of the composition, which generally decreases as decreases (for well-conditioned networks). Late-layer modifications thus have the largest direct effect on the output.
Critical Assessment: Weaknesses & Improvements
Weaknesses & Flaws
(a) Hyperparameter regime is suspiciously extreme. The optimal (keeping only 1% of rank) appears across the vast majority of experiments. For a 4096-dimensional weight matrix, this means keeping rank ~41 out of 4096. This raises a question the paper does not adequately address: at such extreme compression, is the model essentially computing a degenerate projection, and is the improvement because the task only needs a very simple decision boundary? The paper does not show ablations comparing (even more extreme) or ask why is special.
(b) Validation set requirement is underplayed. The paper presents LASER as requiring “no additional parameters or data,” but the hyperparameter search does use labeled data (the 20% validation split). For some benchmarks, this validation set may itself be non-trivial to obtain. The framing as “training-free” is partially misleading.
(c) Absent baselines. The only comparison is to the base model and random rank reduction. There is no comparison to:
- LoRA fine-tuning on the same 20% validation set — it would be revealing to see whether task-specific fine-tuning outperforms LASER given the same data budget
- Full fine-tuning on the validation set
- Prompt engineering / few-shot prompting using the validation examples
- Quantization applied to the same layers
Without these, the claim that LASER is competitive among adaptation methods is unsupported.
(d) No variance reported. Results are reported as single-run point estimates. For accuracy-based benchmarks on held-out sets, there is inherent variance from the test set composition. Without error bars or repeated trials, the significance of small gains (e.g., +2.5pp for LLaMA-2 on WikidataQA) cannot be judged.
(e) Counterfactual gap. The paper shows that SVD-based truncation outperforms random truncation, but doesn’t ask: what if we fine-tune after truncation? The combination of LASER with subsequent fine-tuning could be a natural research direction the paper doesn’t explore.
Limitations the Authors Understate
(a) The “no training required” framing obscures the validation-set dependency. In practice, if you want LASER’s best performance, you need labeled examples for the target task — a validation set. This is a form of task-specific adaptation. The paper acknowledges it but treats it as minor; in production deployment contexts, labeled validation data for each new task is a real cost.
(b) The mechanism remains essentially unexplained. The noise hypothesis is intuitive but not rigorously established. The paper provides correlational evidence (improvements are in late MLP layers, consistent with the knowledge-storage literature) but no causal analysis. Specifically: which spurious associations are being removed? Can we inspect the higher-order components to see what information they encode? This interpretability gap is not acknowledged as a major limitation.
(c) Scalability is not addressed. The paper is from December 2023, when 70B+ models were not yet widely studied. The results on 355M (RoBERTa), 6B (GPT-J), and 7B (LLaMA-2) span less than an order of magnitude in scale. Whether the same phenomenon holds at frontier model scale is entirely open, and this uncertainty is not communicated clearly in the paper.
Concrete Improvement Suggestions
-
Compare against fine-tuning on the same validation set. The most important missing baseline is “what if we just fine-tune on the 20% validation set used for hyperparameter search?” Showing that LASER outperforms fine-tuning without gradient descent would be a much stronger result. Showing it doesn’t would clarify that LASER is a cheap alternative rather than a superior one.
-
Automatic hyperparameter selection using spectral properties. Instead of grid search over , explore whether the spectral gap (the ratio ) or effective rank (entropy of singular value distribution) of a weight matrix predicts which matrices benefit most from LASER. This could make LASER applicable without any labeled validation data.
-
Implement and benchmark factored-form inference. Computing instead of uses storage instead of , and costs FLOPs instead of . For , this is a 50× speedup in that layer’s FLOP count. Implementing and benchmarking this would make LASER practically useful for deployment.
-
Test on instruction-tuned models and RLHF models. The experiments use base pretrained models. Instruction tuning and RLHF significantly alter the weight structure. LASER on models like LLaMA-2-Chat, Mistral Instruct, or GPT-4 (in a research context) may show very different results.
-
Interpretability of removed components. Inspect the discarded singular vectors () and ask: what input patterns have high projection onto these directions? If the noise hypothesis is correct, these directions should correspond to spurious vocabulary biases or stereotyped associations. Showing this interpretability evidence would greatly strengthen the theory.
-
Extend to generation tasks. All evaluations use log-likelihood ranking (effectively treating generation as multiple-choice). Evaluating LASER on free-form generation (MMLU, HELM, GPQA) would significantly broaden the relevance of the findings.
Broader Implications for the Field
What LASER Teaches Us About Trained Models
LASER’s success has implications beyond the method itself. It suggests several things about how large models store and process information:
1. Trained models are over-parameterized even after training. We knew this abstractly (neural tangent kernel theory, lottery ticket hypothesis), but LASER provides a concrete manifestation: even a single weight matrix in a 7B-parameter model can be dramatically compressed at test time without hurting — and sometimes improving — performance. The trained network is storing far more than it needs for any given task.
2. Task-relevant information is concentrated in a tiny subspace. The optimal rank for LASER (, i.e., rank ~41 for a 4096-dimensional weight matrix) means that the task-relevant information in that weight matrix fits into a ~41-dimensional subspace. This is consistent with the intrinsic dimensionality literature, which finds that fine-tuning trajectories for LLMs live in surprisingly low-dimensional subspaces (Aghajanyan et al. 2021).
3. Training introduces systematic noise that hurts test performance. The fact that removing components improves performance is a direct signal that training is not clean. Gradient descent on large corpora with mini-batches introduces structured noise into weight matrices, and this noise is concentrated in the higher-order singular components. LASER’s success is, in some sense, a diagnostic of training imperfection.
Connections to Model Interpretability
LASER’s finding that the higher-order components of MLP matrices encode “noise” opens an interpretability question: what exactly is stored in those components?
One approach is to examine the input-output behavior of the discarded components. If we define the “noise subspace” as , we can ask: for which inputs is large? These are the inputs most affected by LASER. Preliminary analysis would involve computing the right singular vectors , finding vocabulary tokens with high projection onto these directions, and inspecting the resulting token sets for spurious patterns or stereotypes. If the noise hypothesis is correct, these tokens should correspond to stereotyped associations and noisy co-occurrences from the pretraining corpus.
This is a concrete, actionable interpretability experiment that is surprisingly absent from the paper — making it an excellent direction for follow-up work.
Practical Usage Guide
For practitioners who want to experiment with LASER on their own models and tasks, here is a step-by-step practical guide:
Step 1: Identify your target task and collect a validation set. LASER requires approximately 500–2000 labeled examples for reliable hyperparameter selection. Fewer examples increase variance in the optimal (, , ) choice.
Step 2: Choose a starting search strategy. Based on the paper’s findings, prioritize:
- Parameter types: fc_in (Uin) and fc_out (Uout) before attention matrices
- Layer range: the last 25% of transformer layers (e.g., layers 24–32 for a 32-layer model)
- Rank fraction: — start with
Step 3: Run the SVD and evaluate. For each (, , ) combination in your grid, compute torch.linalg.svd(W), truncate to top- components, and evaluate on the validation set. Record validation accuracy.
Step 4: Apply the best hyperparameters to the test set. Use the (, , ) that maximized validation accuracy. Report test set accuracy.
Step 5: Consider combining interventions. Once you have the best single intervention, try adding a second LASER on a different layer or parameter type. Combinations are applied independently and may compound the benefit.
Expected compute cost: For LLaMA-2-7B with a 200-configuration grid and a 1000-example validation set: approximately 4–8 hours on a single A100 GPU (80 GB). The SVD itself is fast; the bottleneck is running forward passes for evaluation.
Summary of Key Findings
To close the technical analysis, here is a structured summary of what the LASER paper established, what it didn’t, and what remains open:
Established findings:
- SVD-based rank reduction of specific weight matrices in specific transformer layers can significantly improve performance on reasoning benchmarks
- The optimal regime is extremely aggressive: (1% rank retention) in the last ~25% of MLP layers (specifically fc_in / Uin)
- Random rank reduction of the same aggressive amount does NOT improve performance — the SVD direction is essential, not mere compression
- The effect generalizes beyond LLMs to Decision Transformers in reinforcement learning
- The improvement exists across models spanning 355M to 7B parameters
What is NOT established:
- A theoretical explanation for why SVD-based truncation helps
- Whether the findings scale to 70B+ models
- Whether the effect holds for instruction-tuned or RLHF models
- Whether LASER is competitive with fine-tuning on the same validation data
- Whether the effect is task-specific (i.e., does the optimal LASER for task A harm performance on task B?)
Open questions with high research value:
- Can spectral properties of weight matrices predict which layers will benefit from LASER without labeled validation data?
- Is there a connection between the singular value spectrum and training dynamics (learning rate, batch size, etc.)?
- Can LASER and quantization be combined for compounding benefits?
- What does the “noise subspace” (discarded singular directions) actually encode, in terms of interpretable linguistic or factual content?
Conclusion
LASER is a genuinely surprising result with a simple implementation. The finding that selectively truncating SVD components of specific weight matrices can improve reasoning performance without any training challenges the assumption that all information in a trained network is useful. The method works best in late MLP layers at extremely aggressive compression ratios (), suggesting that LLMs store substantial task-interfering “noise” in their higher-order weight components.
From a practical standpoint, LASER occupies a niche: it requires a labeled validation set (so it’s not truly zero-shot) but no gradient computation (so it’s cheaper than fine-tuning). For settings where fine-tuning is too expensive but targeted adaptation is needed, LASER provides a lightweight option. Its effectiveness on multiple models and diverse benchmarks makes it a credible tool in the model-editing toolkit.
The critical gaps — the lack of fine-tuning baselines, the unexplained mechanism, the restriction to small models and classification tasks, and the absence of memory/compute benefits in the current implementation — leave substantial room for follow-up work. The most impactful extension would be understanding why SVD-based truncation helps specifically in late MLP layers, which would open the door to principled methods for identifying which components to remove and by how much, without requiring task-specific validation data.
The truth, as the paper’s title suggests, really is in there — buried under a surprisingly large amount of noise that training inadvertently introduces.
For practitioners, the take-away is pragmatic: if you have a small labeled validation set for your target task and a trained LLM, running LASER is a very cheap experiment. The SVD computation and grid search take hours on a single GPU, require no gradient computation, and may yield meaningful improvements. It is worth trying before reaching for fine-tuning.
For researchers, the deeper take-away is mechanistic: trained LLMs appear to store task-relevant information in an extremely low-dimensional subspace of their MLP weight matrices, with the remaining dimensions filled by structured noise. Understanding the origin of this noise (overfitting? gradient stochasticity? data contamination?), its distribution across model families, and its relationship to model capabilities is a rich and largely open research agenda. LASER is a clean probe into this structure, and its simplicity is both its strength and the reason it opens so many unanswered questions.
The field of model compression and efficient inference tends to focus on matching the performance of the original model. LASER’s most provocative contribution is to ask a different question: what if the original model is not the right target to match? What if, by removing the right components, we can do better?
References
- Sharma, P., Ash, J. T., & Misra, D. (2023). The Truth is in There: Improving Reasoning in Language Models with Layer-Selective Rank Reduction. arXiv:2312.13558 [ICLR 2024]
- Meng, K., Bau, D., Andonian, A., & Belinkov, Y. (2022). Locating and Editing Factual Associations in GPT. NeurIPS 2022 (ROME)
- Geva, M., Schuster, R., Berant, J., & Levy, O. (2021). Transformer Feed-Forward Layers Are Key-Value Memories. EMNLP 2021
- Hu, E. J., et al. (2022). LoRA: Low-Rank Adaptation of Large Language Models. ICLR 2022
- Eckart, C., & Young, G. (1936). The approximation of one matrix by another of lower rank. Psychometrika
- Decision Transformer: Chen, L., et al. (2021). Decision Transformer: Reinforcement Learning via Sequence Modeling. NeurIPS 2021