ADAG: Automatically Describing Attribution Graphs

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Arora, Aryaman, Wu, Zhengxuan, Steinhardt, Jacob, Schwettmann, Sarah
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866915926184558592
author Arora, Aryaman
Wu, Zhengxuan
Steinhardt, Jacob
Schwettmann, Sarah
author_facet Arora, Aryaman
Wu, Zhengxuan
Steinhardt, Jacob
Schwettmann, Sarah
contents In language model interpretability research, \textbf{circuit tracing} aims to identify which internal features causally contributed to a particular output and how they affected each other, with the goal of explaining the computations underlying some behaviour. However, all prior circuit tracing work has relied on ad-hoc human interpretation of the role that each feature in the circuit plays, via manual inspection of data artifacts such as the dataset examples the component activates on. We introduce \textbf{ADAG}, an end-to-end pipeline for describing these attribution graphs which is fully automated. To achieve this, we introduce \textit{attribution profiles} which quantify the functional role of a feature via its input and output gradient effects. We then introduce a novel clustering algorithm for grouping features, and an LLM explainer--simulator setup which generates and scores natural-language explanations of the functional role of these feature groups. We run our system on known human-analysed circuit-tracing tasks and recover interpretable circuits, and further show ADAG can find steerable clusters which are responsible for a harmful advice jailbreak in Llama 3.1 8B Instruct.
format Preprint
id arxiv_https___arxiv_org_abs_2604_07615
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle ADAG: Automatically Describing Attribution Graphs
Arora, Aryaman
Wu, Zhengxuan
Steinhardt, Jacob
Schwettmann, Sarah
Computation and Language
I.2.7
In language model interpretability research, \textbf{circuit tracing} aims to identify which internal features causally contributed to a particular output and how they affected each other, with the goal of explaining the computations underlying some behaviour. However, all prior circuit tracing work has relied on ad-hoc human interpretation of the role that each feature in the circuit plays, via manual inspection of data artifacts such as the dataset examples the component activates on. We introduce \textbf{ADAG}, an end-to-end pipeline for describing these attribution graphs which is fully automated. To achieve this, we introduce \textit{attribution profiles} which quantify the functional role of a feature via its input and output gradient effects. We then introduce a novel clustering algorithm for grouping features, and an LLM explainer--simulator setup which generates and scores natural-language explanations of the functional role of these feature groups. We run our system on known human-analysed circuit-tracing tasks and recover interpretable circuits, and further show ADAG can find steerable clusters which are responsible for a harmful advice jailbreak in Llama 3.1 8B Instruct.
title ADAG: Automatically Describing Attribution Graphs
topic Computation and Language
I.2.7
url https://arxiv.org/abs/2604.07615