Moment Matters: Mean and Variance Causal Graph Discovery from Heteroscedastic Observational Data
Not on the map. Why?
Abstract (arXiv)
Heteroscedasticity -- where the variance of a variable changes with other variables -- is pervasive in real data, and elucidating why it arises from the perspective of statistical moments is crucial in scientific knowledge discovery and decision-making. However, standard causal discovery does not reveal which causes act on the mean versus the variance, as it returns a single moment-agnostic graph, limiting interpretability and downstream intervention design. We propose a Bayesian, moment-driven causal discovery framework that infers separate textitmean and textitvariance causal graphs from observational heteroscedastic data. We first derive the identification results by establishing sufficient conditions under which these two graphs are separately identifiable. Building on this theory, we develop a variational inference method that learns a posterior distribution over both graphs, enabling principled uncertainty quantification of structural features (e.g., edges, paths, and subgraphs). To address the challenges of parameter optimization in heteroscedastic models with two graph structures, we take a curvature-aware optimization approach and develop a prior incorporation technique that leverages domain knowledge on node orderings, improving sample efficiency. Experiments on synthetic, semi-synthetic, and real data show that our approach accurately recovers mean and variance structures and outperforms state-of-the-art baselines.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 32 of its 88 references that are in the map. First posted on arXiv 27 Feb 2026.
Builds on
32 of 88 references.
References · 32
- 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,501
- 2013Causal Discovery with Continuous Additive Noise Modelscited by 255
- 2021Beware of the Simulated DAG! Causal Discovery Benchmarks May Be Easy To Gamecited by 161
- 2022Bayesian Structure Learning with Generative Flow Networkscited by 153
- 2021DiBS: Differentiable Bayesian Structure Learningcited by 106
- 2020Causal Autoregressive Flowscited by 99
- 2020Generalization Bounds and Representation Learning for Estimation of Potential Outcomes and Causal Effectscited by 72
- 2021BCD Nets: Scalable Variational Approaches for Bayesian Causal Discoverycited by 68
- 2019PC-Fairness: A Unified Framework for Measuring Causality-based Fairnesscited by 67
- 2020SoftSort: A Continuous Relaxation for the argsort Operatorcited by 66
- 2019Reliable training and estimation of variance networkscited by 63
- 2013Learning Gaussian Networkscited by 62
20 more
- 2022On the Identifiability and Estimation of Causal Location-Scale Noise Modelscited by 56
- 2019ABCD-Strategy: Budgeted Experimental Design for Targeted Causal Structure Discoverycited by 52
- 2023Joint Bayesian Inference of Graphical Structure and Parameters with a Single Generative Flow Networkcited by 50
- 2022Interventions, Where and How? Experimental Design for Causal Models at Scalecited by 49
- 2022Identifying Patient-Specific Root Causes with the Heteroscedastic Noise Modelcited by 38
- 2022Faithful Heteroscedastic Regression with Neural Networkscited by 37
- 2023BayesDAG: Gradient-Based Posterior Inference for Causal Discoverycited by 33
- 2020Causal Rule Ensemble: Interpretable Discovery and Inference of Heterogeneous Treatment Effectscited by 13
- 2017Selective inference for effect modification via the lassocited by 12
- 2023Understanding Pathologies of Deep Heteroskedastic Regressioncited by 10
- 2023Effective Causal Discovery under Identifiable Heteroscedastic Noise Modelcited by 10
- 2022Feature Selection for Discovering Distributional Treatment Effect Modifierscited by 7
- 2023Heteroscedastic Causal Structure Learningcited by 7
- 2024A Skewness-Based Criterion for Addressing Heteroscedastic Noise in Causal Discoverycited by 7
- 2022Causal inference in drug discovery and developmentcited by 6
- 2023Robust Estimation of Causal Heteroscedastic Noise Modelscited by 6
- 2024Effective Bayesian Causal Inference via Structural Marginalisation and Autoregressive Orderscited by 6
- 2023Benchmarking Bayesian Causal Discovery Methods for Downstream Treatment Effect Estimationcited by 3
- 2023Meta-learning for heterogeneous treatment effect estimation with closed-form solverscited by 2
- 2024Differentiable Pareto-Smoothed Weighting for High-Dimensional Heterogeneous Treatment Effect Estimationcited by 2
Built on by
Not cited yet.