mlsanityData to 7 Oct 2026

Learning a Generative Meta-Model of LLM Activations

arXiv 2602.06964 ↗First posted 6 Feb 2026cs.LG cs.AI cs.CLcited by 8

Not on the map. Why?

Abstract (arXiv)

Existing approaches for analyzing neural network activations, such as PCA and sparse autoencoders, rely on strong structural assumptions. Generative models offer an alternative: they can uncover structure without such assumptions and act as priors that improve intervention fidelity. We explore this direction by training diffusion models on one billion residual stream activations, creating "meta-models" that learn the distribution of a network's internal states. We find that diffusion loss decreases smoothly with compute and reliably predicts downstream utility. In particular, applying the meta-model's learned prior to steering interventions improves fluency, with larger gains as loss decreases. Moreover, the meta-model's neurons increasingly isolate concepts into individual units, with sparse probing scores that scale as loss decreases. These results suggest generative meta-models offer a scalable path toward interpretability without restrictive structural assumptions. Project page: https://generative-latent-prior.github.io.

Key facts. Cited by 8 papers in the map (data through 7 Oct 2026). Builds on 56 of its 86 references that are in the map. First posted on arXiv 6 Feb 2026.

Builds on

56 of 86 references.

References · 56

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  3. 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
  4. 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
  5. 2020Scaling Laws for Neural Language Modelscited by 6,411
  6. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
  7. 2022Flow Matching for Generative Modelingcited by 5,426
  8. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
  9. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
  10. 2013Distributed Representations of Words and Phrases and their Compositionalitycited by 3,462
  11. 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,286
  12. 2013Visualizing and Understanding Convolutional Networkscited by 2,278
38 more
  1. 2021SDEdit: Guided Image Synthesis and Editing with Stochastic Differential Equationscited by 1,903
  2. 2020GLU Variants Improve Transformercited by 1,724
  3. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
  4. 2016Understanding intermediate layers using linear classifier probescited by 1,279
  5. 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,104
  6. 2022Editing Models with Task Arithmeticcited by 1,076
  7. 2022Diffusion-LM Improves Controllable Text Generationcited by 1,006
  8. 2016HyperNetworkscited by 1,003
  9. 2022Building Normalizing Flows with Stochastic Interpolantscited by 964
  10. 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 831
  11. 2023Steering Language Models With Activation Engineeringcited by 824
  12. 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 712
  13. 2023The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasetscited by 673
  14. 2024Scaling and evaluating sparse autoencoderscited by 530
  15. 2023Emergent Correspondence from Image Diffusioncited by 437
  16. 2021DExperts: Decoding-Time Controlled Text Generation with Experts and Anti-Expertscited by 434
  17. 2024Gemma Scope: Open Sparse Autoencoders Everywhere All At Once on Gemma 2cited by 381
  18. 2023Finding Neurons in a Haystack: Case Studies with Sparse Probingcited by 328
  19. 2023In-Context Learning Creates Task Vectorscited by 302
  20. 2023Your Diffusion Model is Secretly a Zero-Shot Classifiercited by 301
  21. 2023A Tale of Two Features: Stable Diffusion Complements DINO for Zero-Shot Semantic Correspondencecited by 297
  22. 2023Function Vectors in Large Language Modelscited by 292
  23. 2020Understanding the Role of Individual Units in a Deep Neural Networkcited by 254
  24. 2023Diffusion Hyperfeatures: Searching Through Time and Space for Semantic Correspondencecited by 211
  25. 2025AxBench: Steering LLMs? Even Simple Baselines Outperform Sparse Autoencoderscited by 199
  26. 2023HyperDiffusion: Generating Implicit Neural Fields with Weight-Space Diffusioncited by 152
  27. 2024Llama Scope: Extracting Millions of Features from Llama-3.1-8B with Sparse Autoencoderscited by 133
  28. 2023Unsupervised Semantic Correspondence Using Stable Diffusioncited by 120
  29. 2025Are Sparse Autoencoders Useful? A Case Study in Sparse Probingcited by 109
  30. 2023Return of Unconditional Generation: A Self-supervised Representation Generation Methodcited by 95
  31. 2022Learning to Learn with Generative Models of Neural Network Checkpointscited by 85
  32. 2024SelfIE: Self-Interpretation of Large Language Model Embeddingscited by 63
  33. 2025Text-to-LoRA: Instant Transformer Adaptioncited by 50
  34. 2024LatentQA: Teaching LLMs to Decode Activations Into Natural Languagecited by 49
  35. 2025Angular Steering: Behavior Control via Rotation in Activation Spacecited by 45
  36. 2024NNsight and NDIF: Democratizing Access to Open-Weight Foundation Model Internalscited by 43
  37. 2024Interpreting the Weight Space of Customized Diffusion Modelscited by 35
  38. 2025Training Language Models to Explain Their Own Computationscited by 27

50 most cited of 56 All 56 (JSON)

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 8.

8 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Look But Don't Touch with Sparse Autoencoders for Unlearning in Diffusion Models
  2. 2023Bridging the Gap: Addressing Discrepancies in Diffusion Model Training for Classifier-Free Guidance
  3. 2024Diverse capability and scaling of diffusion and auto-regressive models when learning abstract rules
  4. 2026CASL: Concept-Aligned Sparse Latents for Interpreting Diffusion Models
  5. 2026Mechanism Shift During Post-training from Autoregressive to Masked Diffusion Language Models
  6. 2025Hyper-Transforming Latent Diffusion Models
  7. 2024Controlling Language and Diffusion Models by Transporting Activations
  8. 2026Why Instruction-Based Unlearning Fails in Diffusion Models?