Where Pretraining writes and Alignment reads: the asymmetry of Transformer weight space
On the map: LLM alignment / RLHF / preference optimisation → LLM preference optimization methods
Abstract (arXiv)
Cross-entropy pretraining and preference alignment update the same transformer weights, but leave geometrically distinct traces. We characterise this asymmetry with a relative-subspace-fraction probe that tracks how weight deltas align with residual-stream activation subspaces and with the prediction subspace defined by the unembedding. Alignment deltas concentrate in the read pathway (WQ, WK), along principal directions of attention-input activations, while remaining near-isotropic in the write pathway (WO, W2) relative to the prediction subspace. We explain this pattern through anisotropic gradient accumulation: updates to a matrix W are sums of outer products δt at^top, and inherit directional structure from whichever side has concentrated covariance. For read-pathway matrices, this side is the input activation at, whose covariance is spiked in trained transformers and therefore produces objective-agnostic concentration. For write-pathway matrices, the relevant side is the upstream gradient δt, whose anisotropy depends on the loss. Cross-entropy supplies the canonical sharp per-sample signal, inducing write-pathway prediction geometry during pretraining; alignment objectives typically add little further write-side concentration. We support this explanation with a within-checkpoint trajectory, a graded contrastive-objective control, and a closed-form rank-1 intervention with matched direction controls, providing causal evidence for the proposed weight-space geometry.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in LLM preference optimization methods, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 37 of its 44 references that are in the map. First posted on arXiv 15 May 2026.
Builds on
37 of 44 references.
Closest ancestors · 2
Same direction, newest first
- 2024Is DPO Superior to PPO for LLM Alignment? A Comprehensive Study16 Apr
- 2024ORPO: Monolithic Preference Optimization without Reference Model12 Mar
Other references · 35
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2022Locating and Editing Factual Associations in GPTcited by 2,609
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,130
- 2022Editing Models with Task Arithmeticcited by 1,076
- 2023Steering Llama 2 via Contrastive Activation Additioncited by 1,009
- 2022Mass-Editing Memory in a Transformercited by 976
- 2023Steering Language Models With Activation Engineeringcited by 824
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
- 2023TIES-Merging: Resolving Interference When Merging Modelscited by 795
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 776
23 more
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 713
- 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 679
- 2019Large Batch Optimization for Deep Learning: Training BERT in 76 minutescited by 671
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 649
- 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 579
- 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 522
- 2021Fast Model Editing at Scalecited by 491
- 2020Null It Out: Guarding Protected Attributes by Iterative Nullspace Projectioncited by 442
- 2023LEACE: Perfect linear concept erasure in closed formcited by 254
- 2018Implicit Self-Regularization in Deep Neural Networks: Evidence from Random Matrix Theory and Implications for Learningcited by 211
- 2022Language model compression with weighted low-rank factorizationcited by 192
- 2020Editable Neural Networkscited by 154
- 2024The Geometry of Categorical and Hierarchical Concepts in Large Language Modelscited by 151
- 2017Breaking the Softmax Bottleneck: A High-Rank RNN Language Modelcited by 147
- 2023The Truth is in There: Improving Reasoning in Language Models with Layer-Selective Rank Reductioncited by 120
- 2022Linear Adversarial Concept Erasurecited by 98
- 2023Language Models Implement Simple Word2Vec-style Vector Arithmeticcited by 97
- 2024Robust LLM safeguarding via refusal feature adversarial trainingcited by 60
- 2023The Emergence of Essential Sparsity in Large Pre-trained Models: The Weights that Mattercited by 42
- 2019Where is the Information in a Deep Neural Network?cited by 40
- 2024Linguistic Collapse: Neural Collapse in (Large) Language Modelscited by 36
- 2018Fisher Information and Natural Gradient Learning of Random Deep Networkscited by 28
- 2026nabla-Reasoner: LLM Reasoning via Test-Time Gradient Descent in Latent Spacecited by 4
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026Geometric and Behavioral Stratification in Transformer Residual Streams
- 2026What Does Layer-Importance Reveal About Transformers and State-Space Models?
- 2026What Visual Generators Need from Teachers: Rethinking Representation Alignment
- 2026Weight Decay Regimes in Grokking Transformers: Cheap Online Diagnostics
- 2025From Memorization to Reasoning in the Spectrum of Loss Curvature
- 2026Off-Axis, On Purpose: Where a Transformer Computes Concepts and Why it Does So
- 2026Symmetry Breaking in Transformers for Efficient and Interpretable Training
- 2025Inverse-Free Wilson Loops for Transformers: A Practical Diagnostic for Invariance and Order Sensitivity