mlsanityData to 14 Oct 2026

Uncertainty-aware Causal Decision Making via Effect Bound Decomposition

arXiv 2601.22736 ↗First posted 30 Jan 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Causal inference from observational data can provide strong evidence for finding the best action in a decision-making scenario without having to perform expensive randomized trials. The causal effect of an action is often not pointwise identifiable even with infinite data due to unobserved confounding factors. Furthermore, having only finitely many samples adds another layer of uncertainty to causal effect estimation. Several existing methods can be used to obtain upper and lower bounds to the causal effect, ranging from symbolic methods to the more recent neural network-based approaches, which implicitly incorporate both sources of uncertainty. However, these methods do not inform whether collecting more samples may or may not help identify the best action from observational data, leaving experts in the dark about their data collection strategies. We address this problem with a novel framework that can distinguish the range of causal effect values that might be eliminated by collecting more samples from the range of values that, with high probability, cannot be eliminated with more observational samples. We show that this partitioning can be obtained by solving max-min and min-max optimization problems. We leverage neural causal models to approximately recover this decomposition in practice. We demonstrate via experiments on synthetic and real-world datasets that our algorithm can determine when collecting more samples will not help determine the best action. Our framework can help practitioners decide when to resort to non-observational studies or seek to measure some of the unmeasured confounders for optimal decision-making.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 33 of its 66 references that are in the map. First posted on arXiv 30 Jan 2026.

Builds on

33 of 66 references.

References · 33

  1. 2019Aleatoric and Epistemic Uncertainty in Machine Learning: An Introduction to Concepts and Methodscited by 913
  2. 2018Evidential Deep Learning to Quantify Classification Uncertaintycited by 732
  3. 2016Distributionally Robust Stochastic Optimization with Wasserstein Distancecited by 185
  4. 2017CausalGAN: Learning Causal Implicit Generative Models with Adversarial Trainingcited by 152
  5. 2021The Causal-Neural Connection: Expressiveness, Learnability, and Inferencecited by 98
  6. 2020DoWhy: An End-to-End Library for Causal Inferencecited by 95
  7. 2013Probabilities of Causation: Bounds and Identificationcited by 90
  8. 2018Confounding-Robust Policy Improvementcited by 76
  9. 2013Counterfactual Probabilities: Computational Methods, Bounds and Applicationscited by 67
  10. 2021Partial Counterfactual Identification from Observational and Experimental Datacited by 64
  11. 2020Identifying Causal-Effect Inference Failure with Uncertainty-Aware Modelscited by 57
  12. 2022Neural Causal Models for Counterfactual Identification and Estimationcited by 57
21 more
  1. 2021An Automated Approach to Causal Inference in Discrete Settingscited by 42
  2. 2022DoWhy-GCM: An extension of DoWhy for causal inference in graphical causal modelscited by 41
  3. 2022A Unified Wasserstein Distributional Robustness Framework for Adversarial Trainingcited by 36
  4. 2013On the Testability of Causal Models with Latent and Instrumental Variablescited by 26
  5. 2021Causal Decision Making and Causal Effect Estimation Are Not the Same... and Why It Matterscited by 25
  6. 2023Modeling Causal Mechanisms with Diffusion Models for Interventional and Counterfactual Queriescited by 24
  7. 2021Bounds on Causal Effects and Application to High Dimensional Datacited by 22
  8. 2020A General Method for Deriving Tight Symbolic Bounds on Causal Effectscited by 18
  9. 2022Partial Identification of Treatment Effects with Implicit Generative Modelscited by 17
  10. 2023A Neural Framework for Generalized Causal Sensitivity Analysiscited by 17
  11. 2025Multi-Armed Bandits With Machine Learning-Generated Surrogate Rewardscited by 14
  12. 2024Quantifying Aleatoric Uncertainty of the Treatment Effect: A Novel Orthogonal Learnercited by 10
  13. 2024Modular Learning of Deep Causal Generative Models for High-dimensional Causal Inferencecited by 9
  14. 2013Instrumentality Tests Revisitedcited by 8
  15. 2025Treatment Effect Estimation for Optimal Decision-Makingcited by 8
  16. 2023Ensembled Prediction Intervals for Causal Outcomes Under Hidden Confoundingcited by 7
  17. 2024Reward Maximization for Pure Exploration: Minimax Optimal Good Arm Identification for Nonparametric Multi-Armed Banditscited by 5
  18. 2023Epsilon-Identifiability of Causal Quantitiescited by 4
  19. 2024Conditional Generative Models are Sufficient to Sample from Any Causal Effect Estimandcited by 4
  20. 2023Approximate Causal Effect Identification under Weak Confoundingcited by 3
  21. 2023A Bandit Learning Method for Continuous Games under Feedback Delays with Residual Pseudo-Gradient Estimatecited by 1

Built on by

Not cited yet.