Gradient-Based Neural DAG Learning
Not on the map. Why?
Abstract (arXiv)
We propose a novel score-based approach to learning a directed acyclic graph (DAG) from observational data. We adapt a recently proposed continuous constrained optimization formulation to allow for nonlinear relationships between variables using neural networks. This extension allows to model complex interactions while avoiding the combinatorial nature of the problem. In addition to comparing our method to existing continuous optimization methods, we provide missing empirical comparisons to nonlinear greedy search methods. On both synthetic and real-world data sets, this new method outperforms current continuous methods on most tasks, while being competitive with existing greedy search methods on important metrics for causal inference.
Key facts. Cited by 227 papers in the map (data through 7 Oct 2026). First posted on arXiv 5 Jun 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 24; 2022: 39; 2023: 55; 2024: 39; 2025: 33; 2026: 37.
227 papers.
Sort:
By area:
- 2021D'ya like DAGs? A Survey on Structure Learning and Causal Discoverycited by 209
- 2021Discrete Graph Structure Learning for Forecasting Multiple Time Seriescited by 169
- 2021Beware of the Simulated DAG! Causal Discovery Benchmarks May Be Easy To Gamecited by 159
- 2022DAGMA: Learning DAGs via M-matrices and a Log-Determinant Acyclicity Characterizationcited by 135
- 2022Score matching enables causal discovery of nonlinear additive noise modelscited by 108
- 2021DiBS: Differentiable Bayesian Structure Learningcited by 105
- 2022Deep End-to-end Causal Inferencecited by 93
- 2022Amortized Inference for Causal Structure Learningcited by 89
- 2023Identifiability Guarantees for Causal Disentanglement from Soft Interventionscited by 84
- 2021Efficient Neural Causal Discovery without Acyclicity Constraintscited by 82
- 2022Learning to Induce Causal Structurecited by 68
- 2022Diffusion Models for Causal Discovery via Topological Orderingcited by 64
38 more
- 2021gCastle: A Python Toolbox for Causal Discoverycited by 60
- 2021Ordering-Based Causal Discovery with Reinforcement Learningcited by 55
- 2021DAGs with No Curl: An Efficient DAG Structure Learning Approachcited by 53
- 2022Large-Scale Differentiable Causal Discovery of Factor Graphscited by 47
- 2022Differentiable DAG Samplingcited by 46
- 2022Causal Structure Learning: a Combinatorial Perspectivecited by 46
- 2021Variational Causal Networks: Approximate Bayesian Inference over Causal Structurescited by 43
- 2023Causal Discovery from Temporal Data: An Overview and New Perspectivescited by 43
- 2023Scalable Causal Discovery with Score Matchingcited by 42
- 2023Causal Discovery with Score Matching on Additive Models with Arbitrary Noisecited by 41
- 2021Learning Neural Causal Models with Active Interventionscited by 40
- 2023A Survey on Causal Discovery Methods for I.I.D. and Time Series Datacited by 35
- 2023Structure Learning with Continuous Optimization: A Sober Look and Beyondcited by 34
- 2021NTS-NOTEARS: Learning Nonparametric DBNs With Prior Knowledgecited by 33
- 2023Discovering Causal Relations and Equations from Datacited by 33
- 2023BayesDAG: Gradient-Based Posterior Inference for Causal Discoverycited by 33
- 2021Towards Federated Bayesian Network Structure Learning with Continuous Optimizationcited by 31
- 2022Bayesian learning of Causal Structure and Mechanisms with GFlowNets and Variational Bayescited by 30
- 2023Stable Differentiable Causal Discoverycited by 30
- 2022Improving Multi-Task Generalization via Regularizing Spurious Correlationcited by 28
- 2023Assumption violations in causal discovery and the robustness of score matchingcited by 28
- 2023DynGFN: Towards Bayesian Inference of Gene Regulatory Networks with GFlowNetscited by 27
- 2022BaCaDI: Bayesian Causal Discovery with Unknown Interventionscited by 26
- 2021Simultaneously Reconciled Quantile Forecasting of Hierarchically Related Time Seriescited by 24
- 2022Truncated Matrix Power Iteration for Differentiable DAG Learningcited by 23
- 2022Causality-driven Hierarchical Structure Discovery for Reinforcement Learningcited by 23
- 2023Self-Compatibility: Evaluating Causal Discovery without Ground Truthcited by 22
- 2024A Meta-Learning Approach to Bayesian Causal Discoverycited by 22
- 2022Differentiable Causal Discovery Under Latent Interventionscited by 21
- 2022MissDAG: Causal Discovery in the Presence of Missing Data with Continuous Additive Noise Modelscited by 21
- 2024Position: Understanding LLMs Requires More Than Statistical Generalizationcited by 20
- 2024Sample, estimate, aggregate: A recipe for causal discovery foundation modelscited by 19
- 2023Automated Scientific Discovery: From Equation Discovery to Autonomous Discovery Systemscited by 18
- 2023textttcausalAssembly: Generating Realistic Production Data for Benchmarking Causal Discoverycited by 18
- 2021Structured Sparsity Inducing Adaptive Optimizers for Deep Learningcited by 17
- 2021FedDAG: Federated DAG Structure Learningcited by 17
- 2023Optimizing NOTEARS Objectives via Topological Swapscited by 17
- 2023Towards Causal Foundation Model: on Duality between Causal Inference and Attentioncited by 17
50 most cited of 227 All 227 (JSON)
- 2026MARCEDES: Score-based causal discovery under non-Gaussianity with continuous optimization25 Sep 2026
- 2026Decoupled Causal Discovery20 Sep 2026
- 2026Guide, Not Bind: Why Defeasible Priors Fail in Augmented Lagrangian Causal Discovery3 Sep 2026
- 2026From Causal Plausibility to Causal Reliability: Evaluating LLMs as Calibrated Direct Causal-Edge Classifiers24 Aug 2026
- 2026Resilient Concurrent Causal Discovery for Topological Event Sequences22 Aug 2026
- 2026SVI-DAG: A Structured Variational Inference Approach to Bayesian Causal Discovery5 Aug 2026
- 2026DAG-FM: A Foundation Model for Causal Discovery under Heterogeneous Causal Mechanisms13 Jul 2026
- 2026Order-based Causal Discovery for Multistage Processes4 Jul 2026
- 2026A Survey on Federated Causal Discovery and Inference21 Jun 2026
- 2026DCD-PFN: A Decoupling-Aware Foundation Model for Causal Discovery19 Jun 2026
- 2026Latent Confounded Causal Discovery via Lie Bracket Geometry17 Jun 2026
- 2026FoundCause: Causal Discovery with Latent Confounders from Observational Data16 Jun 2026
38 more
- 2026polyDAG: Polynomial Acyclicity Constraints for Efficient Continuous Causal Discovery in Visual Semantic Graphs5 Jun 2026
- 2026EML-CD: Causal Mechanism Recovery via EML Symbolic Trees in Structure Learning4 Jun 2026
- 2026Do Real-World Datasets Contain Natural Experiments? An Empirical Study Using Causal Feature Selection2 Jun 2026
- 2026Causal Density Functions30 May 2026
- 2026Test Time Training for Supervised Causal Learning28 May 2026
- 2026Evolving Causal Regulatory Networks (ECR-Net)24 May 2026
- 2026Learning Causal Orderings for In-Context Tabular Prediction21 May 2026
- 2026Stable Causal Discovery via Directed Acyclic Graph Aggregation18 May 2026
- 2026PACER: Acyclic Causal Discovery from Large-Scale Interventional Data14 May 2026
- 2026A Topological Sorting Criterion for Random Causal Directed Acyclic Graphs7 May 2026
- 2026When Does Gene Regulatory Network Inference Break? A Controlled Diagnostic Study of Causal and Correlational Methods on Single-Cell Data6 May 2026
- 2026PRCD-MAP: Learning How Much to Trust Imperfect Priors in Causal Discovery3 May 2026
- 2026Proactive Dialogue Model with Intent Prediction30 Apr 2026
- 2026Optimization-Free Topological Sort for Causal Discovery via the Schur Complement of Score Jacobians28 Apr 2026
- 2026DAGAF: A directed acyclic generative adversarial framework for joint structure learning and tabular data synthesis5 Apr 2026
- 2026Smoothing the Landscape: Causal Structure Learning via Diffusion Denoising Objectives2 Apr 2026
- 2026A Knowledge-Informed Pretrained Model for Causal Discovery21 Mar 2026
- 2026Scalable Contrastive Causal Discovery under Unknown Soft Interventions3 Mar 2026
- 2026Coarse-to-Fine Learning of Dynamic Causal Structures26 Feb 2026
- 2026Sparse Additive Model Pruning for Order-Based Causal Structure Learning17 Feb 2026
- 2026Federated Learning of Nonlinear Temporal Dynamics with Graph Attention-based Cross-Client Interpretability13 Feb 2026
- 2026CauScale: Neural Causal Discovery at Scale9 Feb 2026
- 2026Causal Graph Learning via Distributional Invariance of Cause-Effect Relationship3 Feb 2026
- 2026Efficient Causal Structure Learning via Modular Subgraph Integration28 Jan 2026
- 2026Ordering-based Causal Discovery via Generalized Score Matching22 Jan 2026
- 2025CauSight: Learning to Supersense for Visual Causal Discovery1 Dec 2025
- 2025Skewness-Robust Causal Discovery in Location-Scale Noise Models18 Nov 2025
- 2025Robust Causal Discovery under Imperfect Structural Constraints10 Nov 2025
- 2025Higher-Order Causal Structure Learning with Additive Models5 Nov 2025
- 2025Linear Causal Discovery with Interventional Constraints30 Oct 2025
- 2025Group Interventions on Deep Networks for Causal Discovery in Subsystems27 Oct 2025
- 2025Competition is the key: A Game Theoretic Causal Discovery Approach23 Oct 2025
- 2025The Robustness of Differentiable Causal Discovery in Misspecified Scenarios14 Oct 2025
- 2025Score-based Greedy Search for Structure Identification of Partially Observed Linear Causal Models5 Oct 2025
- 2025DAG DECORation: Continuous Optimization for Structure Learning under Hidden Confounding2 Oct 2025
- 2025Differentiable Structure Learning and Causal Discovery for General Binary Data25 Sep 2025
- 2025Causality-Induced Positional Encoding for Transformer-Based Representation Learning of Non-Sequential Features20 Sep 2025
- 2025Effects of Distributional Biases on Gradient-Based Causal Discovery in the Bivariate Categorical Case1 Sep 2025
50 newest of 227 All 227 (JSON)
- 2021D'ya like DAGs? A Survey on Structure Learning and Causal Discoverycited by 209
- 2021Discrete Graph Structure Learning for Forecasting Multiple Time Seriescited by 169
- 2021Beware of the Simulated DAG! Causal Discovery Benchmarks May Be Easy To Gamecited by 159
- 2022DAGMA: Learning DAGs via M-matrices and a Log-Determinant Acyclicity Characterizationcited by 135
- 2022Score matching enables causal discovery of nonlinear additive noise modelscited by 108
- 2021DiBS: Differentiable Bayesian Structure Learningcited by 105
- 2022Deep End-to-end Causal Inferencecited by 93
- 2022Amortized Inference for Causal Structure Learningcited by 89
- 2023Identifiability Guarantees for Causal Disentanglement from Soft Interventionscited by 84
- 2021Efficient Neural Causal Discovery without Acyclicity Constraintscited by 82
- 2022Learning to Induce Causal Structurecited by 68
- 2021gCastle: A Python Toolbox for Causal Discoverycited by 60
12 most cited of 223 All 223 (JSON)
- 2026MARCEDES: Score-based causal discovery under non-Gaussianity with continuous optimization25 Sep 2026
- 2026Decoupled Causal Discovery20 Sep 2026
- 2026Guide, Not Bind: Why Defeasible Priors Fail in Augmented Lagrangian Causal Discovery3 Sep 2026
- 2026From Causal Plausibility to Causal Reliability: Evaluating LLMs as Calibrated Direct Causal-Edge Classifiers24 Aug 2026
- 2026Resilient Concurrent Causal Discovery for Topological Event Sequences22 Aug 2026
- 2026SVI-DAG: A Structured Variational Inference Approach to Bayesian Causal Discovery5 Aug 2026
- 2026DAG-FM: A Foundation Model for Causal Discovery under Heterogeneous Causal Mechanisms13 Jul 2026
- 2026Order-based Causal Discovery for Multistage Processes4 Jul 2026
- 2026DCD-PFN: A Decoupling-Aware Foundation Model for Causal Discovery19 Jun 2026
- 2026Latent Confounded Causal Discovery via Lie Bracket Geometry17 Jun 2026
- 2026FoundCause: Causal Discovery with Latent Confounders from Observational Data16 Jun 2026
- 2026polyDAG: Polynomial Acyclicity Constraints for Efficient Continuous Causal Discovery in Visual Semantic Graphs5 Jun 2026
12 newest of 223 All 223 (JSON)