Competition is the key: A Game Theoretic Causal Discovery Approach
Not on the map. Why?
Abstract (arXiv)
Causal discovery remains a central challenge in machine learning, yet existing methods face a fundamental gap: algorithms like GES and GraN-DAG achieve strong empirical performance but lack finite-sample guarantees, while theoretically principled approaches fail to scale. We close this gap by introducing a game-theoretic reinforcement learning framework for causal discovery, where a DDQN agent directly competes against a strong baseline (GES or GraN-DAG), always warm-starting from the opponent's solution. This design yields three provable guarantees: the learned graph is never worse than the opponent, warm-starting strictly accelerates convergence, and most importantly, with high probability the algorithm selects the true best candidate graph. To the best of our knowledge, our result makes a first-of-its-kind progress in explaining such finite-sample guarantees in causal discovery: on synthetic SEMs (30 nodes), the observed error probability decays with n, tightly matching theory. On real-world benchmarks including Sachs, Asia, Alarm, Child, Hepar2, Dream, and Andes, our method consistently improves upon GES and GraN-DAG while remaining theoretically safe. Remarkably, it scales to large graphs such as Hepar2 (70 nodes), Dream (100 nodes), and Andes (220 nodes). Together, these results establish a new class of RL-based causal discovery algorithms that are simultaneously provably consistent, sample-efficient, and practically scalable, marking a decisive step toward unifying empirical performance with rigorous finite-sample theory.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 9 of its 24 references that are in the map. First posted on arXiv 23 Oct 2025.
Builds on
9 of 24 references.
References · 9
- 2015Deep Reinforcement Learning with Double Q-learningcited by 1,704
- 2018DAGs with NO TEARS: Continuous Optimization for Structure Learningcited by 745
- 2019DAG-GNN: DAG Structure Learning with Graph Neural Networkscited by 329
- 2019Gradient-Based Neural DAG Learningcited by 227
- 2020On the Role of Sparsity and DAG Constraints for Learning Linear DAGscited by 186
- 2019Causal Discovery with Reinforcement Learningcited by 145
- 2021gCastle: A Python Toolbox for Causal Discoverycited by 60
- 2021Ordering-Based Causal Discovery with Reinforcement Learningcited by 55
- 2021On the Role of Entropy-based Loss for Learning Causal Structures with Continuous Optimizationcited by 5
Built on by
Not cited yet.