Masked Gradient-Based Causal Structure Learning
Not on the map. Why?
Abstract (arXiv)
This paper studies the problem of learning causal structures from observational data. We reformulate the Structural Equation Model (SEM) with additive noises in a form parameterized by binary graph adjacency matrix and show that, if the original SEM is identifiable, then the binary adjacency matrix can be identified up to super-graphs of the true causal graph under mild conditions. We then utilize the reformulated SEM to develop a causal structure learning method that can be efficiently trained using gradient-based optimization, by leveraging a smooth characterization on acyclicity and the Gumbel-Softmax approach to approximate the binary adjacency matrix. It is found that the obtained entries are typically near zero or one and can be easily thresholded to identify the edges. We conduct experiments on synthetic and real datasets to validate the effectiveness of the proposed method, and show that it readily includes different smooth model functions and achieves a much improved performance on most datasets considered.
Key facts. Cited by 89 papers in the map (data through 14 Oct 2026). First posted on arXiv 18 Oct 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 12; 2022: 19; 2023: 19; 2024: 18; 2025: 13; 2026: 8.
89 papers.
Sort:
By area:
- 2021D'ya like DAGs? A Survey on Structure Learning and Causal Discoverycited by 211
- 2021Disentanglement via Mechanism Sparsity Regularization: A New Principle for Nonlinear ICAcited by 160
- 2021DiBS: Differentiable Bayesian Structure Learningcited by 106
- 2022Diffusion Models for Causal Discovery via Topological Orderingcited by 66
- 2021gCastle: A Python Toolbox for Causal Discoverycited by 61
- 2021Ordering-Based Causal Discovery with Reinforcement Learningcited by 55
- 2021DAGs with No Curl: An Efficient DAG Structure Learning Approachcited by 53
- 2022Causal Representation Learning for Instantaneous and Temporal Effects in Interactive Systemscited by 51
- 2021Unsuitability of NOTEARS for Causal Graph Discoverycited by 47
- 2022Differentiable DAG Samplingcited by 47
- 2022Large-Scale Differentiable Causal Discovery of Factor Graphscited by 47
- 2023Causal Discovery from Temporal Data: An Overview and New Perspectivescited by 43
38 more
- 2023Structure Learning with Continuous Optimization: A Sober Look and Beyondcited by 36
- 2023A Survey on Causal Discovery Methods for I.I.D. and Time Series Datacited by 35
- 2021Towards Federated Bayesian Network Structure Learning with Continuous Optimizationcited by 31
- 2022Improving Multi-Task Generalization via Regularizing Spurious Correlationcited by 28
- 2022Truncated Matrix Power Iteration for Differentiable DAG Learningcited by 23
- 2022MissDAG: Causal Discovery in the Presence of Missing Data with Continuous Additive Noise Modelscited by 22
- 2023From Identifiable Causal Representations to Controllable Counterfactual Generation: A Survey on Causal Generative Modelingcited by 22
- 2022Differentiable Causal Discovery Under Latent Interventionscited by 21
- 2023NODAGS-Flow: Nonlinear Cyclic Causal Structure Learningcited by 20
- 2021FedDAG: Federated DAG Structure Learningcited by 17
- 2024DeepRicci: Self-supervised Graph Structure-Feature Co-Refinement for Alleviating Over-squashingcited by 16
- 2024Identifiable Exchangeable Mechanisms for Causal Structure and Representation Learningcited by 16
- 2023Learning Causally Disentangled Representations via the Principle of Independent Causal Mechanismscited by 14
- 2023RDGSL: Dynamic Graph Representation Learning with Structure Learningcited by 13
- 2024RealTCD: Temporal Causal Discovery from Interventional Data with Large Language Modelcited by 13
- 2022Guided Hybrid Quantization for Object detection in Multimodal Remote Sensing Imagery via One-to-one Self-teachingcited by 12
- 2021ML4C: Seeing Causality Through Latent Vicinitycited by 11
- 2022Towards Privacy-Aware Causal Structure Learning in Federated Settingcited by 11
- 2023Federated Causality Learning with Explainable Adaptive Optimizationcited by 11
- 2021Multi-task Learning of Order-Consistent Causal Graphscited by 10
- 2022CIPCaD-Bench: Continuous Industrial Process datasets for benchmarking Causal Discovery methodscited by 10
- 2022Novel Ordering-based Approaches for Causal Structure Learning in the Presence of Unobserved Variablescited by 10
- 2023Emerging Synergies in Causality and Deep Generative Models: A Surveycited by 10
- 2023Constraint-Free Structure Learning with Smooth Acyclic Orientationscited by 10
- 2023Effective Causal Discovery under Identifiable Heteroscedastic Noise Modelcited by 10
- 2023Evaluation of Induced Expert Knowledge in Causal Structure Learning by NOTEARScited by 8
- 2023Open problems in causal structure learning: A case study of COVID-19 in the UKcited by 7
- 2025Learning General Causal Structures with Hidden Dynamic Process for Climate Analysiscited by 7
- 2023MetaPhysiCa: OOD Robustness in Physics-informed Machine Learningcited by 6
- 2024Optimal Transport for Structure Learning Under Missing Datacited by 6
- 2025Rethinking Spatio-Temporal Anomaly Detection: A Vision for Causality-Driven Cybersecuritycited by 6
- 2021On the Role of Entropy-based Loss for Learning Causal Structures with Continuous Optimizationcited by 5
- 2022Learning Latent Structural Causal Modelscited by 5
- 2023Boosting Differentiable Causal Discovery via Adaptive Sample Reweightingcited by 5
- 2024Causal Representation Learning in Temporal Data via Single-Parent Decodingcited by 5
- 2024On Discovery of Local Independence over Continuous Variables via Neural Contextual Decompositioncited by 4
- 2024psiDAG: Projected Stochastic Approximation Iteration for DAG Structure Learningcited by 4
- 2025Differentiable Cyclic Causal Discovery Under Unmeasured Confounderscited by 4
50 most cited of 89 All 89 (JSON)
- 2026SVI-DAG: A Structured Variational Inference Approach to Bayesian Causal Discovery5 Aug 2026
- 2026Order-based Causal Discovery for Multistage Processes4 Jul 2026
- 2026A Framework for Directed Acyclic Hypergraph Learning19 Jun 2026
- 2026Score-Based Causal Discovery of Latent Variable Causal Models19 May 2026
- 2026Smoothing the Landscape: Causal Structure Learning via Diffusion Denoising Objectives2 Apr 2026
- 2026Adaptive-CaRe: Adaptive Causal Regularization for Robust Outcome Prediction6 Feb 2026
- 2026Causal Graph Learning via Distributional Invariance of Cause-Effect Relationship3 Feb 2026
- 2026Knowledge-Guided Time-Varying Causal Inference for Arctic Sea Ice Dynamics25 Jan 2026
- 2025Differentially Private and Federated Structure Learning in Bayesian Networks1 Dec 2025
- 2025Higher-Order Causal Structure Learning with Additive Models5 Nov 2025
- 2025CausalRec: A CausalBoost Attention Model for Sequential Recommendation24 Oct 2025
- 2025Shylock: Causal Discovery in Multivariate Time Series based on Hybrid Constraints24 Oct 2025
38 more
- 2025Score-based Greedy Search for Structure Identification of Partially Observed Linear Causal Models5 Oct 2025
- 2025Empowering Denoising Sequential Recommendation with Large Language Model Embeddings5 Oct 2025
- 2025A Novel Compression Framework for YOLOv8: Achieving Real-Time Aerial Object Detection on Edge Devices via Structured Pruning and Channel-Wise Distillation16 Sep 2025
- 2025Effects of Distributional Biases on Gradient-Based Causal Discovery in the Bivariate Categorical Case1 Sep 2025
- 2025Differentiable Cyclic Causal Discovery Under Unmeasured Confounders11 Aug 2025
- 2025A Gradient-based Causal Discovery Framework with Applications to Complex Industrial Processes15 Jul 2025
- 2025Rethinking Spatio-Temporal Anomaly Detection: A Vision for Causality-Driven Cybersecurity10 Jul 2025
- 2025Learning General Causal Structures with Hidden Dynamic Process for Climate Analysis21 Jan 2025
- 2025Causal Learning for Heterogeneous Subgroups Based on Nonlinear Causal Kernel Clustering20 Jan 2025
- 2024Differentiable Causal Discovery For Latent Hierarchical Causal Models29 Nov 2024
- 2024CGLearn: Consistent Gradient-Based Learning for Out-of-Distribution Generalization9 Nov 2024
- 2024psiDAG: Projected Stochastic Approximation Iteration for DAG Structure Learning31 Oct 2024
- 2024LOCAL: Learning with Orientation Matrix to Infer Causal Structure from Time Series Data25 Oct 2024
- 2024Causal Order Discovery based on Monotonic SCMs24 Oct 2024
- 2024MissNODAG: Differentiable Cyclic Causal Graph Learning from Incomplete Data24 Oct 2024
- 2024Revisiting Differentiable Structure Learning: Inconsistency of ℓ1 Penalty and Beyond24 Oct 2024
- 2024Causal Representation Learning in Temporal Data via Single-Parent Decoding9 Oct 2024
- 2024Interventional Causal Structure Discovery over Graphical Models with Convergence and Optimality Guarantees9 Aug 2024
- 2024Towards Generalizable Reinforcement Learning via Causality-Guided Self-Adaptive Representations30 Jul 2024
- 2024Scalable Variational Causal Discovery Unconstrained by Acyclicity6 Jul 2024
- 2024Identifiable Exchangeable Mechanisms for Causal Structure and Representation Learning20 Jun 2024
- 2024On Discovery of Local Independence over Continuous Variables via Neural Contextual Decomposition12 May 2024
- 2024RealTCD: Temporal Causal Discovery from Interventional Data with Large Language Model23 Apr 2024
- 2024Learning Cyclic Causal Models from Incomplete Data23 Feb 2024
- 2024Optimal Transport for Structure Learning Under Missing Data23 Feb 2024
- 2024DeepRicci: Self-supervised Graph Structure-Feature Co-Refinement for Alleviating Over-squashing23 Jan 2024
- 2024Boosting Causal Additive Models12 Jan 2024
- 2023Effective Causal Discovery under Identifiable Heteroscedastic Noise Model20 Dec 2023
- 2023Federated Causality Learning with Explainable Adaptive Optimization9 Dec 2023
- 2023Distributionally Robust Skeleton Learning of Discrete Bayesian Networks10 Nov 2023
- 2023Causal Structure Representation Learning of Confounders in Latent Space for Recommendation2 Nov 2023
- 2023Recovering Linear Causal Models with Latent Variables via Cholesky Factorization of Covariance Matrix1 Nov 2023
- 2023Joint Distributional Learning via Cramer-Wold Distance25 Oct 2023
- 2023From Identifiable Causal Representations to Controllable Counterfactual Generation: A Survey on Causal Generative Modeling17 Oct 2023
- 2023Constraint-Free Structure Learning with Smooth Acyclic Orientations15 Sep 2023
- 2023RDGSL: Dynamic Graph Representation Learning with Structure Learning5 Sep 2023
- 2023Learning Causally Disentangled Representations via the Principle of Independent Causal Mechanisms2 Jun 2023
- 2023Open problems in causal structure learning: A case study of COVID-19 in the UK5 May 2023
50 newest of 89 All 89 (JSON)
- 2021D'ya like DAGs? A Survey on Structure Learning and Causal Discoverycited by 211
- 2021Disentanglement via Mechanism Sparsity Regularization: A New Principle for Nonlinear ICAcited by 160
- 2021DiBS: Differentiable Bayesian Structure Learningcited by 106
- 2021gCastle: A Python Toolbox for Causal Discoverycited by 61
- 2021Ordering-Based Causal Discovery with Reinforcement Learningcited by 55
- 2021DAGs with No Curl: An Efficient DAG Structure Learning Approachcited by 53
- 2022Causal Representation Learning for Instantaneous and Temporal Effects in Interactive Systemscited by 51
- 2021Unsuitability of NOTEARS for Causal Graph Discoverycited by 47
- 2022Differentiable DAG Samplingcited by 47
- 2022Large-Scale Differentiable Causal Discovery of Factor Graphscited by 47
- 2023Causal Discovery from Temporal Data: An Overview and New Perspectivescited by 43
- 2023Structure Learning with Continuous Optimization: A Sober Look and Beyondcited by 36
12 most cited of 86 All 86 (JSON)
- 2026SVI-DAG: A Structured Variational Inference Approach to Bayesian Causal Discovery5 Aug 2026
- 2026Order-based Causal Discovery for Multistage Processes4 Jul 2026
- 2026A Framework for Directed Acyclic Hypergraph Learning19 Jun 2026
- 2026Score-Based Causal Discovery of Latent Variable Causal Models19 May 2026
- 2026Adaptive-CaRe: Adaptive Causal Regularization for Robust Outcome Prediction6 Feb 2026
- 2026Causal Graph Learning via Distributional Invariance of Cause-Effect Relationship3 Feb 2026
- 2026Knowledge-Guided Time-Varying Causal Inference for Arctic Sea Ice Dynamics25 Jan 2026
- 2025Differentially Private and Federated Structure Learning in Bayesian Networks1 Dec 2025
- 2025Higher-Order Causal Structure Learning with Additive Models5 Nov 2025
- 2025CausalRec: A CausalBoost Attention Model for Sequential Recommendation24 Oct 2025
- 2025Shylock: Causal Discovery in Multivariate Time Series based on Hybrid Constraints24 Oct 2025
- 2025Score-based Greedy Search for Structure Identification of Partially Observed Linear Causal Models5 Oct 2025
12 newest of 86 All 86 (JSON)