Dual Path Attribution: Efficient Attribution for SwiGLU-Transformers through Layer-Wise Target Propagation
About
Understanding the internal mechanisms of transformer-based large language models (LLMs) is crucial for their reliable deployment and effective operation. While recent efforts have yielded a plethora of attribution methods attempting to balance faithfulness and computational efficiency, dense component attribution remains prohibitively expensive. In this work, we introduce Dual Path Attribution (DPA), a novel framework that faithfully traces information flow on the frozen transformer in one forward and one backward pass without requiring counterfactual examples. DPA analytically decomposes and linearizes the computational structure of the SwiGLU Transformers into distinct pathways along which it propagates a targeted unembedding vector to receive the effective representation at each residual position. This target-centric propagation achieves O(1) time complexity with respect to the number of model components, scaling to long input sequences and dense component attribution. Extensive experiments on standard interpretability benchmarks demonstrate that DPA achieves state-of-the-art faithfulness and unprecedented efficiency compared to existing baselines.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Token Attribution Faithfulness | Known 1000 | Distance5.6 | 40 | |
| Component-level attribution | Known 1000 | Discrepancy Score0.00e+0 | 40 | |
| Component-level attribution | IOI | Dissimilarity (dis.)0.00e+0 | 32 | |
| Token Attribution Faithfulness | SQuAD v2.0 | Disagreement11.93 | 30 | |
| Factual Knowledge | Known 1000 | Disagreement Rate5.79 | 10 | |
| Token Attribution Faithfulness | IMDB | Distance48.21 | 10 | |
| Reading Comprehension | SQuAD v2.0 | Disambiguation Score17.73 | 10 | |
| Sentiment Analysis | IMDB | Dis. Score69.34 | 10 |