A Unifying Relational Perspective on Expressive Lottery Tickets
Lorenz Kummer, Samir Moustafa, Anatol Ehrlich, Franka Bause, Marco Nennstiel, Przemysław Wałęga, Nils M. Kriege
Abstract
Graph neural networks (GNNs) are widely used, but how parameter sparsity affects the expressivity of relational (RGNNs) and temporal (TGNNs) variants is poorly understood. The Strong Expressive Lottery Ticket Hypothesis (SELTH) posits the existence of sparse GNNs that preserve Weisfeiler-Leman (WL) expressivity on static graphs. We generalize this existence result to a probabilistic statement for multi-relational and temporal domains via the relational WL (RWL). We prove that sufficiently parameterized RGNNs contain sparse subnetworks that maintain 1-RWL expressivity and derive a lower bound on the probability that a random pruning yields such a subnetwork. We show that common TGNNs and cross-graph message passing schemes admit RGNN reformulations such that they inherit these guarantees and, moreover, that the expressivity of a sparse RGNN is connected to its optimization behavior under common update regimes. Experiments instantiate the bound, compare it to empirical probabilities on synthetic data, and study how pre-training expressivity relates to optimization and prediction quality metrics on temporal and molecular benchmarks.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 2805232d-68f0-469d-90f2-e3cfd4956e7aBuilds on16
- Composition-based Multi-Relational Graph Convolutional NetworksShikhar Vashishth, Soumya Sanyal, Vikram Nitin, Partha P. TalukdarICLR 2020 · 1,105 citations
- Inductive representation learning on temporal graphsDa Xu, Chuanwei Ruan, Evren Körpeoglu, Sushant Kumar et al.ICLR 2020 · 901 citations
- Proving the Lottery Ticket Hypothesis: Pruning is All You NeedEran Malach, Gilad Yehudai, Shai Shalev-Shwartz, Ohad ShamirICML 2020 · 327 citations
- A Unified Lottery Ticket Hypothesis for Graph Neural NetworksTianlong Chen, Yongduo Sui, Xuxi Chen, Aston Zhang et al.ICML 2021 · 208 citations
- Gradient Flow in Sparse Neural Networks and How Lottery Tickets WinUtku Evci, Yani Ioannou, Cem Keskin, Yann N. DauphinAAAI 2022 · 106 citations
Related papers
- Weisfeiler and Leman Go Gambling: Why Expressive Lottery Tickets WinLorenz Kummer, Samir Moustafa, Anatol Ehrlich, Franka Bause et al.ICML 2025
- Rethinking Graph Lottery Tickets: Graph Sparsity MattersBo Hui, Da Yan, Xiaolong Ma, Wei-Shinn KuICLR 2023 · 2 citations
- Searching Lottery Tickets in Graph Neural Networks: A Dual PerspectiveKun Wang, Yuxuan Liang, Pengkun Wang, Xu Wang et al.ICLR 2023
- Graph Lottery Ticket AutomatedGuibin Zhang, Kun Wang, Wei Huang, Yanwei Yue et al.ICLR 2024 · 17 citations
- Pre-Training Identification of Graph Winning Tickets in Adaptive Spatial-Temporal Graph Neural NetworksWenying Duan, Tianxiang Fang, Hong Rao, Xiaoxi HeKDD 2024 · 1 citation
