Causal Interpretation of Neural Network Computations with Contribution Decomposition

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Melander, Joshua Brendan, Alaoui, Zaki, Liu, Shenghua, Ganguli, Surya, Baccus, Stephen A.
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914376002306048
author Melander, Joshua Brendan
Alaoui, Zaki
Liu, Shenghua
Ganguli, Surya
Baccus, Stephen A.
author_facet Melander, Joshua Brendan
Alaoui, Zaki
Liu, Shenghua
Ganguli, Surya
Baccus, Stephen A.
contents Understanding how neural networks transform inputs into outputs is crucial for interpreting and manipulating their behavior. Most existing approaches analyze internal representations by identifying hidden-layer activation patterns correlated with human-interpretable concepts. Here we take a direct approach to examine how hidden neurons act to drive network outputs. We introduce CODEC (Contribution Decomposition), a method that uses sparse autoencoders to decompose network behavior into sparse motifs of hidden-neuron contributions, revealing causal processes that cannot be determined by analyzing activations alone. Applying CODEC to benchmark image-classification networks, we find that contributions grow in sparsity and dimensionality across layers and, unexpectedly, that they progressively decorrelate positive and negative effects on network outputs. We further show that decomposing contributions into sparse modes enables greater control and interpretation of intermediate layers, supporting both causal manipulations of network output and human-interpretable visualizations of distinct image components that combine to drive that output. Finally, by analyzing state-of-the-art models of neural activity in the vertebrate retina, we demonstrate that CODEC uncovers combinatorial actions of model interneurons and identifies the sources of dynamic receptive fields. Overall, CODEC provides a rich and interpretable framework for understanding how nonlinear computations evolve across hierarchical layers, establishing contribution modes as an informative unit of analysis for mechanistic insights into artificial neural networks.
format Preprint
id arxiv_https___arxiv_org_abs_2603_06557
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Causal Interpretation of Neural Network Computations with Contribution Decomposition
Melander, Joshua Brendan
Alaoui, Zaki
Liu, Shenghua
Ganguli, Surya
Baccus, Stephen A.
Machine Learning
Neurons and Cognition
Understanding how neural networks transform inputs into outputs is crucial for interpreting and manipulating their behavior. Most existing approaches analyze internal representations by identifying hidden-layer activation patterns correlated with human-interpretable concepts. Here we take a direct approach to examine how hidden neurons act to drive network outputs. We introduce CODEC (Contribution Decomposition), a method that uses sparse autoencoders to decompose network behavior into sparse motifs of hidden-neuron contributions, revealing causal processes that cannot be determined by analyzing activations alone. Applying CODEC to benchmark image-classification networks, we find that contributions grow in sparsity and dimensionality across layers and, unexpectedly, that they progressively decorrelate positive and negative effects on network outputs. We further show that decomposing contributions into sparse modes enables greater control and interpretation of intermediate layers, supporting both causal manipulations of network output and human-interpretable visualizations of distinct image components that combine to drive that output. Finally, by analyzing state-of-the-art models of neural activity in the vertebrate retina, we demonstrate that CODEC uncovers combinatorial actions of model interneurons and identifies the sources of dynamic receptive fields. Overall, CODEC provides a rich and interpretable framework for understanding how nonlinear computations evolve across hierarchical layers, establishing contribution modes as an informative unit of analysis for mechanistic insights into artificial neural networks.
title Causal Interpretation of Neural Network Computations with Contribution Decomposition
topic Machine Learning
Neurons and Cognition
url https://arxiv.org/abs/2603.06557