Causal Lifting and Link Prediction
Not on the map. Why?
Abstract (arXiv)
Existing causal models for link prediction assume an underlying set of inherent node factors -- an innate characteristic defined at the node's birth -- that governs the causal evolution of links in the graph. In some causal tasks, however, link formation is path-dependent: The outcome of link interventions depends on existing links. Unfortunately, these existing causal methods are not designed for path-dependent link formation, as the cascading functional dependencies between links (arising from path dependence) are either unidentifiable or require an impractical number of control variables. To overcome this, we develop the first causal model capable of dealing with path dependencies in link prediction. In this work we introduce the concept of causal lifting, an invariance in causal models of independent interest that, on graphs, allows the identification of causal link prediction queries using limited interventional data. Further, we show how structural pairwise embeddings exhibit lower bias and correctly represent the task's causal structure, as opposed to existing node embeddings, e.g., graph neural network node embeddings and matrix factorization. Finally, we validate our theoretical findings on three scenarios for causal link prediction tasks: knowledge base completion, covariance matrix estimation and consumer-product recommendations.
Key facts. Cited by 5 papers in the map (data through 7 Oct 2026). Builds on 38 of its 101 references that are in the map. First posted on arXiv 2 Feb 2023.
Builds on
38 of 101 references.
References · 38
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,414
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,148
- 2016node2vec: Scalable Feature Learning for Networkscited by 1,913
- 2014DeepWalk: Online Learning of Social Representationscited by 1,543
- 2014Embedding Entities and Relations for Learning and Inference in Knowledge Basescited by 929
- 2016Complex Embeddings for Simple Link Predictioncited by 897
- 2019RotatE: Knowledge Graph Embedding by Relational Rotation in Complex Spacecited by 893
- 2018Link Prediction Based on Graph Neural Networkscited by 829
- 2020Masked Label Prediction: Unified Message Passing Model for Semi-Supervised Classificationcited by 500
- 2020Can Graph Neural Networks Count Substructures?cited by 235
26 more
- 2019On the equivalence between graph isomorphism testing and function approximation with GNNscited by 172
- 2020Query2box: Reasoning over Knowledge Graphs in Vector Space using Box Embeddingscited by 169
- 2020Labeling Trick: A Theory of Using Graph Neural Networks for Multi-Node Representation Learningcited by 148
- 2017Knowledge Graph Completion via Complex Tensor Factorizationcited by 80
- 2017Causal Embeddings for Recommendationcited by 79
- 2020On the Benefits of Invariance in Neural Networkscited by 78
- 2022Neo-GNNs: Neighborhood Overlap-aware Graph Neural Networks for Link Predictioncited by 78
- 2021Learning from Counterfactual Links for Link Predictioncited by 69
- 2018Sequence-Aware Recommender Systemscited by 58
- 2019Learning Individual Causal Effects from Networked Observational Datacited by 56
- 2013Graph cluster randomization: network exposure to multiple universescited by 42
- 2014Design and analysis of experiments in networks: Reducing bias from interferencecited by 41
- 2018Ontology Reasoning with Deep Neural Networkscited by 40
- 2022OOD Link Prediction Generalization Capabilities of Message-Passing GNNs in Larger Test Graphscited by 38
- 2019Stacking Models for Nearly Optimal Link Prediction in Complex Networkscited by 35
- 2022Learning Causal Effects on Hypergraphscited by 31
- 2015Neural Network Matrix Factorizationcited by 25
- 2019Addressing Marketing Bias in Product Recommendationscited by 23
- 2021Simple, Fast, and Flexible Framework for Matrix Completion with Infinite Width Neural Networkscited by 17
- 2020Causal Network Motifs: Identifying Heterogeneous Spillover Effects in A/B Testscited by 14
- 2021Causal Collaborative Filteringcited by 13
- 2014The Matthew effect in empirical datacited by 10
- 2020TorchKGE: Knowledge Graph Embedding in Python and PyTorchcited by 9
- 2021Node Proximity Is All You Need: Unified Structural and Positional Node and Graph Embeddingcited by 9
- 2022Link recommendations: Their impact on network structure and minoritiescited by 6
- 2014Lifted Probabilistic Inference for Asymmetric Graphical Modelscited by 2
Built on by
Citing papers per year: 2023: 2; 2024: 2; 2025: 1.
5 papers.
- 2023A Multi-Task Perspective for Link Prediction with New Relation Types and Nodescited by 23
- 2024End-To-End Causal Effect Estimation from Unstructured Natural Language Datacited by 18
- 2023Expressive Sign Equivariant Networks for Spectral Geometric Learningcited by 15
- 2024Heterophilic Graph Neural Networks Optimization with Causal Message-passingcited by 4
- 2025Are Heterogeneous Graph Neural Networks Truly Effective for Node Classification? A Causal Perspectivecited by 0
- 2025Are Heterogeneous Graph Neural Networks Truly Effective for Node Classification? A Causal Perspective7 Oct 2025
- 2024Heterophilic Graph Neural Networks Optimization with Causal Message-passing21 Nov 2024
- 2024End-To-End Causal Effect Estimation from Unstructured Natural Language Data9 Jul 2024
- 2023Expressive Sign Equivariant Networks for Spectral Geometric Learning4 Dec 2023
- 2023A Multi-Task Perspective for Link Prediction with New Relation Types and Nodes12 Jul 2023