Papers › The Information Pathways Hypothesis: Transformers are Dynamic Self-Ensembles

The Information Pathways Hypothesis: Transformers are Dynamic Self-Ensembles

2 Jun 2023arXiv:2306.01705archive 2025-07-28

Md Shamim Hussain, Mohammed J. Zaki, Dharmashankar Subramanian

Transformers use the dense self-attention mechanism which gives a lot of flexibility for long-range connectivity. Over multiple layers of a deep transformer, the number of possible connectivity patterns increases exponentially. However, very few of these contribute to the performance of the network, and even fewer are essential. We hypothesize that there are sparsely connected sub-networks within a transformer, called information pathways which can be trained independently. However, the dynamic (i.e., input-dependent) nature of these pathways makes it difficult to prune dense self-attention during training. But the overall distribution of these pathways is often predictable. We take advantage of this fact to propose Stochastically Subsampled self-Attention (SSA) - a general-purpose training strategy for transformers that can reduce both the memory and computational cost of self-attention by 4 to 8 times during training while also serving as a regularization method - improving generalization over dense training. We show that an ensemble of sub-models can be formed from the subsampled pathways within a network, which can achieve better performance than its densely attended counterpart. We perform experiments on a variety of NLP, computer vision and graph learning tasks in both generative and discriminative settings to provide empirical evidence for our claims and show the effectiveness of the proposed method.

PaperPDFCode

In Syntology Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

Code

shamim-hussain/ssa officialmentioned in papermentioned on GitHub report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

Graph LearningGraph RegressionImage ClassificationImage GenerationLanguage Modelling

Results from the paper archive 2025-07-28

TaskDatasetModelMetricValueRank at snapshotLeaderboardReport
Graph Regression PCQM4Mv2-LSC EGT+SSA+Self-ensemble Validation MAE 0.0865 #14 of 20 Archive leaderboard report
Graph Regression PCQM4Mv2-LSC EGT+SSA Validation MAE 0.0876 #16 of 20 Archive leaderboard report
Image Classification ImageNet Swin-T+SSA Top 1 Accuracy 81.89% #601 of 1060 Archive leaderboard report
Language Modelling WikiText-103 Transformer+SSA+Self-ensemble Test perplexity 17.18 #23 of 89 Archive leaderboard report
Language Modelling WikiText-103 Transformer+SSA+Self-ensemble Validation perplexity 16.54 #23 of 89 Archive leaderboard report
Language Modelling WikiText-103 Transformer+SSA Test perplexity 17.60 #27 of 89 Archive leaderboard report
Language Modelling WikiText-103 Transformer+SSA Validation perplexity 16.91 #27 of 89 Archive leaderboard report
Language Modelling enwik8 Transformer+SSA Bit per Character (BPC) 1.024 #22 of 42 Archive leaderboard report

Ranks are positions in the archive's leaderboards as they stood at the 2025-07-28 snapshot. Results published since then are not among these rows, so a rank here is not a current standing.

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections