mlsanityData to 7 Oct 2026

Where Pretraining writes and Alignment reads: the asymmetry of Transformer weight space

arXiv 2605.16600 ↗First posted 15 May 2026cs.LG cs.AI cs.CLcited by 1

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

  1. 2024Is DPO Superior to PPO for LLM Alignment? A Comprehensive Study16 Apr
  2. 2024ORPO: Monolithic Preference Optimization without Reference Model12 Mar

Other references · 35

  1. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
  2. 2022Locating and Editing Factual Associations in GPTcited by 2,609
  3. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
  4. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
  5. 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,130
  6. 2022Editing Models with Task Arithmeticcited by 1,076
  7. 2023Steering Llama 2 via Contrastive Activation Additioncited by 1,009
  8. 2022Mass-Editing Memory in a Transformercited by 976
  9. 2023Steering Language Models With Activation Engineeringcited by 824
  10. 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
  11. 2023TIES-Merging: Resolving Interference When Merging Modelscited by 795
  12. 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 776
23 more
  1. 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 713
  2. 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 679
  3. 2019Large Batch Optimization for Deep Learning: Training BERT in 76 minutescited by 671
  4. 2017The Implicit Bias of Gradient Descent on Separable Datacited by 649
  5. 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 579
  6. 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 522
  7. 2021Fast Model Editing at Scalecited by 491
  8. 2020Null It Out: Guarding Protected Attributes by Iterative Nullspace Projectioncited by 442
  9. 2023LEACE: Perfect linear concept erasure in closed formcited by 254
  10. 2018Implicit Self-Regularization in Deep Neural Networks: Evidence from Random Matrix Theory and Implications for Learningcited by 211
  11. 2022Language model compression with weighted low-rank factorizationcited by 192
  12. 2020Editable Neural Networkscited by 154
  13. 2024The Geometry of Categorical and Hierarchical Concepts in Large Language Modelscited by 151
  14. 2017Breaking the Softmax Bottleneck: A High-Rank RNN Language Modelcited by 147
  15. 2023The Truth is in There: Improving Reasoning in Language Models with Layer-Selective Rank Reductioncited by 120
  16. 2022Linear Adversarial Concept Erasurecited by 98
  17. 2023Language Models Implement Simple Word2Vec-style Vector Arithmeticcited by 97
  18. 2024Robust LLM safeguarding via refusal feature adversarial trainingcited by 60
  19. 2023The Emergence of Essential Sparsity in Large Pre-trained Models: The Weights that Mattercited by 42
  20. 2019Where is the Information in a Deep Neural Network?cited by 40
  21. 2024Linguistic Collapse: Neural Collapse in (Large) Language Modelscited by 36
  22. 2018Fisher Information and Natural Gradient Learning of Random Deep Networkscited by 28
  23. 2026nabla-Reasoner: LLM Reasoning via Test-Time Gradient Descent in Latent Spacecited by 4

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026Geometric and Behavioral Stratification in Transformer Residual Streams
  2. 2026What Does Layer-Importance Reveal About Transformers and State-Space Models?
  3. 2026What Visual Generators Need from Teachers: Rethinking Representation Alignment
  4. 2026Weight Decay Regimes in Grokking Transformers: Cheap Online Diagnostics
  5. 2025From Memorization to Reasoning in the Spectrum of Loss Curvature
  6. 2026Off-Axis, On Purpose: Where a Transformer Computes Concepts and Why it Does So
  7. 2026Symmetry Breaking in Transformers for Efficient and Interpretable Training
  8. 2025Inverse-Free Wilson Loops for Transformers: A Practical Diagnostic for Invariance and Order Sensitivity