Weight-sparse transformers have interpretable circuits

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Gao, Leo, Rajaram, Achyuta, Coxon, Jacob, Govande, Soham V., Baker, Bowen, Mossing, Dan
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866915623283458048
author Gao, Leo
Rajaram, Achyuta
Coxon, Jacob
Govande, Soham V.
Baker, Bowen
Mossing, Dan
author_facet Gao, Leo
Rajaram, Achyuta
Coxon, Jacob
Govande, Soham V.
Baker, Bowen
Mossing, Dan
contents Finding human-understandable circuits in language models is a central goal of the field of mechanistic interpretability. We train models to have more understandable circuits by constraining most of their weights to be zeros, so that each neuron only has a few connections. To recover fine-grained circuits underlying each of several hand-crafted tasks, we prune the models to isolate the part responsible for the task. These circuits often contain neurons and residual channels that correspond to natural concepts, with a small number of straightforwardly interpretable connections between them. We study how these models scale and find that making weights sparser trades off capability for interpretability, and scaling model size improves the capability-interpretability frontier. However, scaling sparse models beyond tens of millions of nonzero parameters while preserving interpretability remains a challenge. In addition to training weight-sparse models de novo, we show preliminary results suggesting our method can also be adapted to explain existing dense models. Our work produces circuits that achieve an unprecedented level of human understandability and validates them with considerable rigor.
format Preprint
id arxiv_https___arxiv_org_abs_2511_13653
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Weight-sparse transformers have interpretable circuits
Gao, Leo
Rajaram, Achyuta
Coxon, Jacob
Govande, Soham V.
Baker, Bowen
Mossing, Dan
Machine Learning
Artificial Intelligence
Finding human-understandable circuits in language models is a central goal of the field of mechanistic interpretability. We train models to have more understandable circuits by constraining most of their weights to be zeros, so that each neuron only has a few connections. To recover fine-grained circuits underlying each of several hand-crafted tasks, we prune the models to isolate the part responsible for the task. These circuits often contain neurons and residual channels that correspond to natural concepts, with a small number of straightforwardly interpretable connections between them. We study how these models scale and find that making weights sparser trades off capability for interpretability, and scaling model size improves the capability-interpretability frontier. However, scaling sparse models beyond tens of millions of nonzero parameters while preserving interpretability remains a challenge. In addition to training weight-sparse models de novo, we show preliminary results suggesting our method can also be adapted to explain existing dense models. Our work produces circuits that achieve an unprecedented level of human understandability and validates them with considerable rigor.
title Weight-sparse transformers have interpretable circuits
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2511.13653