Learning a Generative Meta-Model of LLM Activations

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Luo, Grace, Feng, Jiahai, Darrell, Trevor, Radford, Alec, Steinhardt, Jacob
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910014387519488
author Luo, Grace
Feng, Jiahai
Darrell, Trevor
Radford, Alec
Steinhardt, Jacob
author_facet Luo, Grace
Feng, Jiahai
Darrell, Trevor
Radford, Alec
Steinhardt, Jacob
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2602_06964
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Learning a Generative Meta-Model of LLM Activations
Luo, Grace
Feng, Jiahai
Darrell, Trevor
Radford, Alec
Steinhardt, Jacob
Machine Learning
Artificial Intelligence
Computation and Language
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.
title Learning a Generative Meta-Model of LLM Activations
topic Machine Learning
Artificial Intelligence
Computation and Language
url https://arxiv.org/abs/2602.06964