Learning a Generative Meta-Model of LLM Activations
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
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2013Distributed Representations of Words and Phrases and their Compositionalitycited by 3,462
- 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,286
- 2013Visualizing and Understanding Convolutional Networkscited by 2,278
38 more
- 2021SDEdit: Guided Image Synthesis and Editing with Stochastic Differential Equationscited by 1,903
- 2020GLU Variants Improve Transformercited by 1,724
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
- 2016Understanding intermediate layers using linear classifier probescited by 1,279
- 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,104
- 2022Editing Models with Task Arithmeticcited by 1,076
- 2022Diffusion-LM Improves Controllable Text Generationcited by 1,006
- 2016HyperNetworkscited by 1,003
- 2022Building Normalizing Flows with Stochastic Interpolantscited by 964
- 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 831
- 2023Steering Language Models With Activation Engineeringcited by 824
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 712
- 2023The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasetscited by 673
- 2024Scaling and evaluating sparse autoencoderscited by 530
- 2023Emergent Correspondence from Image Diffusioncited by 437
- 2021DExperts: Decoding-Time Controlled Text Generation with Experts and Anti-Expertscited by 434
- 2024Gemma Scope: Open Sparse Autoencoders Everywhere All At Once on Gemma 2cited by 381
- 2023Finding Neurons in a Haystack: Case Studies with Sparse Probingcited by 328
- 2023In-Context Learning Creates Task Vectorscited by 302
- 2023Your Diffusion Model is Secretly a Zero-Shot Classifiercited by 301
- 2023A Tale of Two Features: Stable Diffusion Complements DINO for Zero-Shot Semantic Correspondencecited by 297
- 2023Function Vectors in Large Language Modelscited by 292
- 2020Understanding the Role of Individual Units in a Deep Neural Networkcited by 254
- 2023Diffusion Hyperfeatures: Searching Through Time and Space for Semantic Correspondencecited by 211
- 2025AxBench: Steering LLMs? Even Simple Baselines Outperform Sparse Autoencoderscited by 199
- 2023HyperDiffusion: Generating Implicit Neural Fields with Weight-Space Diffusioncited by 152
- 2024Llama Scope: Extracting Millions of Features from Llama-3.1-8B with Sparse Autoencoderscited by 133
- 2023Unsupervised Semantic Correspondence Using Stable Diffusioncited by 120
- 2025Are Sparse Autoencoders Useful? A Case Study in Sparse Probingcited by 109
- 2023Return of Unconditional Generation: A Self-supervised Representation Generation Methodcited by 95
- 2022Learning to Learn with Generative Models of Neural Network Checkpointscited by 85
- 2024SelfIE: Self-Interpretation of Large Language Model Embeddingscited by 63
- 2025Text-to-LoRA: Instant Transformer Adaptioncited by 50
- 2024LatentQA: Teaching LLMs to Decode Activations Into Natural Languagecited by 49
- 2025Angular Steering: Behavior Control via Rotation in Activation Spacecited by 45
- 2024NNsight and NDIF: Democratizing Access to Open-Weight Foundation Model Internalscited by 43
- 2024Interpreting the Weight Space of Customized Diffusion Modelscited by 35
- 2025Training Language Models to Explain Their Own Computationscited by 27
50 most cited of 56 All 56 (JSON)
Built on by
Citing papers per year: 2026: 8.
8 papers.
- 2026UniSteer: Text-Guided Flow Matching in Activation Space for Versatile LLM Steeringcited by 1
- 2026Tensor Product Representation Probes Reveal Shared Structure Across Linear Directionscited by 0
- 2026Reasoning Fine-Tuning Induces Persistent Latent Policy Statescited by 0
- 2026Scaling Interpretable Transformers with Parity Bottleneck Layerscited by 0
- 2026Off-Manifold Collapse in Guided Protein Language Modelscited by 0
- 2026Persona Dosing: Calibrated Activation Steering for Graded Trait Controlcited by 0
- 2026MetaSteer: Context-Conditioned, nonlinear Steering via Attention-Projection Adaptationcited by 0
- 2026Right Answer, Wrong Mechanism: Detecting Pernicious Divergence in Causal Interventionscited by 0
- 2026Right Answer, Wrong Mechanism: Detecting Pernicious Divergence in Causal Interventions30 Sep 2026
- 2026MetaSteer: Context-Conditioned, nonlinear Steering via Attention-Projection Adaptation30 Sep 2026
- 2026Persona Dosing: Calibrated Activation Steering for Graded Trait Control28 Sep 2026
- 2026Off-Manifold Collapse in Guided Protein Language Models19 Aug 2026
- 2026Scaling Interpretable Transformers with Parity Bottleneck Layers22 Jul 2026
- 2026Reasoning Fine-Tuning Induces Persistent Latent Policy States20 Jul 2026
- 2026UniSteer: Text-Guided Flow Matching in Activation Space for Versatile LLM Steering28 May 2026
- 2026Tensor Product Representation Probes Reveal Shared Structure Across Linear Directions11 May 2026
Similar, not cited
Close in wording, no citation either way
- 2026Look But Don't Touch with Sparse Autoencoders for Unlearning in Diffusion Models
- 2023Bridging the Gap: Addressing Discrepancies in Diffusion Model Training for Classifier-Free Guidance
- 2024Diverse capability and scaling of diffusion and auto-regressive models when learning abstract rules
- 2026CASL: Concept-Aligned Sparse Latents for Interpreting Diffusion Models
- 2026Mechanism Shift During Post-training from Autoregressive to Masked Diffusion Language Models
- 2025Hyper-Transforming Latent Diffusion Models
- 2024Controlling Language and Diffusion Models by Transporting Activations
- 2026Why Instruction-Based Unlearning Fails in Diffusion Models?