When Should Graph Attention Be Sparse? Learning a Per-Edge Tsallis Index

📅 2026-08-03
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
This work addresses the limitation of conventional graph attention mechanisms, which rely on fixed softmax normalization and struggle to simultaneously accommodate the divergent attention requirements of homophilic and heterophilic graph structures. To overcome this, the authors propose the Learnable Tsallis Graph Attention (LTGA) layer, which introduces a learnable Tsallis entropy index \( q \) into the attention mechanism for the first time. This enables dynamic, edge-wise modulation of attention distributions, allowing continuous interpolation between heavy-tailed and compactly supported forms. The method facilitates edge-level sparsification and interpretable attention pruning, and is end-to-end trainable via a Tsallis entropy–based differentiable reparameterization. Evaluated across eight benchmark datasets, LTGA achieves the best average rank (2.75), automatically zeroes out 42% of erroneous edge attentions, and ablating these pruned edges results in a 7.1-point performance drop—significantly outperforming random pruning.
📝 Abstract
Graph attention normalizes neighborhood scores with softmax, the maximum-entropy choice under Shannon statistics. But homophilic and heterophilic graphs want different attention shapes, and one fixed normalization cannot serve both. We propose \textbf{LTGA} (\textbf{L}earnable \textbf{T}sallis \textbf{G}raph \textbf{A}ttention), a graph attention layer whose Tsallis entropic index $q$ is learned jointly with the weights, interpolating continuously between heavy-tailed ($q\!<\!1$), softmax ($q\!=\!1$) and compact-support ($q\!>\!1$) attention at four granularities from a global scalar to a per-edge index, under a bounded reparameterization that starts every model at the GAT baseline. Across eight benchmarks at ten seeds, LTGA-Edge takes the best average rank ($2.75$), but the omnibus test does not reject ($p\!=\!0.199$) and learning $q$ does not beat searching it: a validation-tuned frozen grid reaches $61.4\%$, tuned $α$-entmax $62.2\%$ and a capacity-matched $q\!\equiv\!1$ control $62.0\%$, against $61.7\%$ for LTGA-Edge. What the learned index buys is one run instead of a grid, and an interpretable mechanism: where $q$ leaves $1$, it prunes $42\%$ of attention coefficients to exactly zero, and those edges are selectively the wrong ones, restoring them costs $7.1$ points, while random pruning at the same rate costs $13.0$ more. Project page: https://kleyt0n.github.io/ltga
Problem

Research questions and friction points this paper is trying to address.

graph attention
homophily
heterophily
attention sparsity
Tsallis entropy
Innovation

Methods, ideas, or system contributions that make the work stand out.

Tsallis entropy
graph attention
learnable sparsity
heterophilic graphs
adaptive normalization
🔎 Similar Papers
2024-02-16Nature CommunicationsCitations: 2
💼 Related Jobs
No related jobs found.
K
Kleyton da Costa
University College London & Holistic AI
B
Bernardo Modenesi
University of Utah