Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Lasse Marten Jantsch, Dong-Jae Koh, Seonghyeon Lee, Young-Kyoon Suh• 2026

Related benchmarks

TaskDatasetResultRank
Token Attribution FaithfulnessKnown 1000
Distance5.6
40
Component-level attributionKnown 1000
Discrepancy Score0.00e+0
40
Component-level attributionIOI
Dissimilarity (dis.)0.00e+0
32
Token Attribution FaithfulnessSQuAD v2.0
Disagreement11.93
30
Factual KnowledgeKnown 1000
Disagreement Rate5.79
10
Token Attribution FaithfulnessIMDB
Distance48.21
10
Reading ComprehensionSQuAD v2.0
Disambiguation Score17.73
10
Sentiment AnalysisIMDB
Dis. Score69.34
10
Showing 8 of 8 rows

Other info

Follow for update