A Game Theoretic Free Energy Analysis of Higher Order Synergy in Attention Heads of Large Language Models
Not on the map. Why?
Abstract (arXiv)
Large language models rely on multihead attention, but interactions among heads remain poorly understood. We apply the Game Theoretic Free Energy Principle (GTFEP): a framework casting multiagent systems as distributed variational inference to analyze attention heads as bounded rational agents. According to GTFEP, each head minimizes its variational free energy, and collective behavior follows a Gibbs distribution over coalition structures whose energy is decomposed into Harsanyi dividends. Using a tractable approximation (uniform prior, deterministic dynamics), coalition free energy reduces to joint Shannon entropy of discretized head outputs (argmax key index). Pairwise dividends become mutual information (nonnegative), while triple dividends correspond to interaction information and can be negative. On BERT, GPT2, and Llama with GSM8K, triple dividends are consistently negative, revealing higher order redundancy. The Nash FEP correspondence guarantees that stationary points of collective free energy are epsilon Nash equilibria; thus, heads with negligible contribution can be pruned with minimal performance loss. Pruning heads with low marginal contribution reduces computational cost with minimal performance loss: for example, pruning 20% of heads in GPT2 reduces FLOPs by 18%, increases throughput by 22%, and raises perplexity only modestly (from 28.4 to 33.4 on GSM8K). Our work shows GTFEP provides a principled foundation for analyzing and optimizing transformer architectures.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 20 of its 34 references that are in the map. First posted on arXiv 10 May 2026.
Builds on
20 of 34 references.
References · 20
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 844
- 2024BlockPruner: Fine-grained Pruning for Large Language Modelscited by 35
- 2022Natural Language Syntax Complies with the Free-Energy Principlecited by 7
- 2026Garbage Attention in Large Language Models: BOS Sink Heads and Sink-aware Pruningcited by 7
- 2025Distributionally Robust Free Energy Principle for Decision-Makingcited by 4
- 2025Structured Pruning for Diverse Best-of-N Reasoning Optimizationcited by 4
- 2024CLOVER: Cross-Layer Orthogonal Vectors Pruning and Fine-Tuningcited by 3
- 2025Active inference and artificial reasoningcited by 3
- 2025Beyond Shapley Values: Cooperative Games for the Interpretation of Machine Learning Modelscited by 2
- 2026D2Prune: Sparsifying Large Language Models via Dual Taylor Expansion and Attention Distribution Awarenesscited by 2
8 more
- 2026Multi-Head Attention Is a Multi-Player Gamecited by 2
- 2024Bayesian Mechanics of Synaptic Learning under the Free Energy Principlecited by 1
- 2025Beyond Parallelism: Synergistic Computational Graph Effects in Multi-Head Attentioncited by 1
- 2025SAP: Syntactic Attention Pruning for Transformer-based Language Modelscited by 1
- 2025SHRP: Specialized Head Routing and Pruning for Efficient Encoder Compressioncited by 1
- 2026TensorLens: End-to-End Transformer Analysis via High-Order Attention Tensorscited by 1
- 2026NeuroGame Transformer: Gibbs-Inspired Attention Driven by Game Theory and Statistical Physicscited by 1
- 2026A Collective Variational Principle Unifying Bayesian Inference, Game Theory, and Thermodynamicscited by 1
Built on by
Not cited yet.