Mixture of Link Predictors on Graphs
On the map: Mixture-of-experts → MoE for time series and graphs
Abstract (arXiv)
Link prediction, which aims to forecast unseen connections in graphs, is a fundamental task in graph machine learning. Heuristic methods, leveraging a range of different pairwise measures such as common neighbors and shortest paths, often rival the performance of vanilla Graph Neural Networks (GNNs). Therefore, recent advancements in GNNs for link prediction (GNN4LP) have primarily focused on integrating one or a few types of pairwise information. In this work, we reveal that different node pairs within the same dataset necessitate varied pairwise information for accurate prediction and models that only apply the same pairwise information uniformly could achieve suboptimal performance. As a result, we propose a simple mixture of experts model Link-MoE for link prediction. Link-MoE utilizes various GNNs as experts and strategically selects the appropriate expert for each node pair based on various types of pairwise information. Experimental results across diverse real-world datasets demonstrate substantial performance improvement from Link-MoE. Notably, Link-MoE achieves a relative improvement of 18.71% on the MRR metric for the Pubmed dataset and 9.59% on the Hits@100 metric for the ogbl-ppa dataset, compared to the best baselines.
Key facts. Cited by 10 papers in the map (data through 7 Oct 2026). Placed in MoE for time series and graphs, one of 12 directions in Mixture-of-experts. Builds on 30 of its 56 references that are in the map. First posted on arXiv 13 Feb 2024.
Builds on
30 of 56 references.
Closest ancestors · 1
Same direction, newest first
Other references · 29
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2017Graph Attention Networkscited by 8,854
- 2017Inductive Representation Learning on Large Graphscited by 6,284
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,148
- 2016node2vec: Scalable Feature Learning for Networkscited by 1,913
- 2016Variational Graph Auto-Encoderscited by 1,373
- 2016Revisiting Semi-Supervised Learning with Graph Embeddingscited by 970
- 2014Embedding Entities and Relations for Learning and Inference in Knowledge Basescited by 929
- 2021GLaM: Efficient Scaling of Language Models with Mixture-of-Expertscited by 890
- 2018Link Prediction Based on Graph Neural Networkscited by 829
- 2021Scaling Vision with Sparse Mixture of Expertscited by 740
17 more
- 2019Multi-scale Attributed Node Embeddingcited by 543
- 2018Session-based Recommendation with Graph Neural Networkscited by 312
- 2021Neural Bellman-Ford Networks: A General Graph Neural Network Framework for Link Predictioncited by 273
- 2020Labeling Trick: A Theory of Using Graph Neural Networks for Multi-Node Representation Learningcited by 148
- 2022Equivariant and Stable Positional Encoding for More Powerful Graph Neural Networkscited by 94
- 2022Graph Neural Networks for Link Prediction with Subgraph Sketchingcited by 81
- 2022Neo-GNNs: Neighborhood Overlap-aware Graph Neural Networks for Link Predictioncited by 78
- 2023Evaluating Graph Neural Networks for Link Prediction: Current Pitfalls and New Benchmarkingcited by 69
- 2023Neural Common Neighbor with Completion for Link Predictioncited by 64
- 2022Towards Understanding Mixture of Experts in Deep Learningcited by 52
- 2019Stacking Models for Nearly Optimal Link Prediction in Complex Networkscited by 35
- 2023Revisiting Link Prediction: A Data Perspectivecited by 29
- 2022Beyond Real-world Benchmark Datasets: An Empirical Study of Node Classification with GNNscited by 23
- 2023LPFormer: An Adaptive Graph Transformer for Link Predictioncited by 17
- 2020Personalized PageRank to a Target Node, Revisitedcited by 16
- 2022Link Prediction on Heterophilic Graphs via Disentangled Representation Learningcited by 13
- 2017Leveraging Node Attributes for Incomplete Relational Datacited by 10
Built on by
Citing papers per year: 2024: 2; 2025: 5; 2026: 3.
10 papers.
- 2025GiGL: Large-Scale Graph Neural Networks at Snapchatcited by 8
- 2024DA-MoE: Addressing Depth-Sensitivity in Graph-Level Analysis through Mixture of Expertscited by 7
- 2026TFMLinker: Universal Link Predictor by Graph In-Context Learning with Tabular Foundation Modelscited by 5
- 2024Reconsidering the Performance of GAE in Link Predictioncited by 4
- 2026Graph2Video: Leveraging Video Models to Model Dynamic Graph Evolutioncited by 1
- 2025Heuristic Methods are Good Teachers to Distill MLPs for Graph Link Predictioncited by 0
- 2025H2GFM: Towards unifying Homogeneity and Heterogeneity on Text-Attributed Graphscited by 0
- 2025Enhancing Contrastive Link Prediction With Edge Balancing Augmentationcited by 0
- 2025Adaptive Graph Mixture of Residual Experts: Unsupervised Learning on Diverse Graphs with Heterogeneous Specializationcited by 0
- 2026Chimaera: A Mixture-of-Graph-Experts Architecture for Cross-Task and Cross-Dataset Graph Learningcited by 0
- 2026Chimaera: A Mixture-of-Graph-Experts Architecture for Cross-Task and Cross-Dataset Graph Learning8 Sep 2026
- 2026Graph2Video: Leveraging Video Models to Model Dynamic Graph Evolution9 Mar 2026
- 2026TFMLinker: Universal Link Predictor by Graph In-Context Learning with Tabular Foundation Models9 Feb 2026
- 2025Adaptive Graph Mixture of Residual Experts: Unsupervised Learning on Diverse Graphs with Heterogeneous Specialization24 Oct 2025
- 2025Enhancing Contrastive Link Prediction With Edge Balancing Augmentation20 Aug 2025
- 2025H2GFM: Towards unifying Homogeneity and Heterogeneity on Text-Attributed Graphs10 Jun 2025
- 2025Heuristic Methods are Good Teachers to Distill MLPs for Graph Link Prediction8 Apr 2025
- 2025GiGL: Large-Scale Graph Neural Networks at Snapchat20 Feb 2025
- 2024Reconsidering the Performance of GAE in Link Prediction6 Nov 2024
- 2024DA-MoE: Addressing Depth-Sensitivity in Graph-Level Analysis through Mixture of Experts5 Nov 2024
Similar, not cited
Close in wording, no citation either way
- 2025A Scalable Pretraining Framework for Link Prediction with Efficient Adaptation
- 2025Heuristic-Informed Mixture of Experts for Link Prediction in Multilayer Networks
- 2024Mixture of Experts for Node Classification
- 2025Training Diverse Graph Experts for Ensembles: A Systematic Empirical Study
- 2024Node-wise Filtering in Graph Neural Networks: A Mixture of Experts Approach
- 2026Early-Exit Graph Neural Networks for Link Prediction
- 2025GMoPE:A Prompt-Expert Mixture Framework for Graph Foundation Models
- 2026Learning How Much to Think: Difficulty-Aware Dynamic MoEs for Graph Node Classification