Papers
Topics
Authors
Recent
Gemini 2.5 Flash
Gemini 2.5 Flash
173 tokens/sec
GPT-4o
7 tokens/sec
Gemini 2.5 Pro Pro
46 tokens/sec
o3 Pro
4 tokens/sec
GPT-4.1 Pro
38 tokens/sec
DeepSeek R1 via Azure Pro
28 tokens/sec
2000 character limit reached

Advancing Counterfactual Inference through Nonlinear Quantile Regression (2306.05751v3)

Published 9 Jun 2023 in cs.LG and stat.ME

Abstract: The capacity to address counterfactual "what if" inquiries is crucial for understanding and making use of causal influences. Traditional counterfactual inference, under Pearls' counterfactual framework, typically depends on having access to or estimating a structural causal model. Yet, in practice, this causal model is often unknown and might be challenging to identify. Hence, this paper aims to perform reliable counterfactual inference based solely on observational data and the (learned) qualitative causal structure, without necessitating a predefined causal model or even direct estimations of conditional distributions. To this end, we establish a novel connection between counterfactual inference and quantile regression and show that counterfactual inference can be reframed as an extended quantile regression problem. Building on this insight, we propose a practical framework for efficient and effective counterfactual inference implemented with neural networks under a bi-level optimization scheme. The proposed approach enhances the capacity to generalize estimated counterfactual outcomes to unseen data, thereby providing an upper bound on the generalization error. Furthermore, empirical evidence demonstrates its superior statistical efficiency in comparison to existing methods. Empirical results conducted on multiple datasets offer compelling support for our theoretical assertions.

Definition Search Book Streamline Icon: https://streamlinehq.com
References (41)
  1. Rademacher and gaussian complexities: Risk bounds and structural results. Journal of Machine Learning Research, 3(Nov):463–482, 2002.
  2. Estimating the effects of continuous-valued interventions using generative adversarial networks. Advances in Neural Information Processing Systems, 33:16434–16445, 2020.
  3. Interventional and counterfactual inference with diffusion models. arXiv preprint arXiv:2302.00860, 2023.
  4. Making scalable meta learning practical. arXiv preprint arXiv:2310.05674, 2023a.
  5. Betty: An automatic differentiation library for multilevel optimization. In The Eleventh International Conference on Learning Representations, 2023b. URL https://openreview.net/forum?id=LV_MeMS38Q9.
  6. De Brouwer, E. Deep counterfactual estimation with categorical background variables. Advances in Neural Information Processing Systems, 35:35213–35225, 2022.
  7. Transport-based counterfactual models. arXiv preprint arXiv:2108.13025, 2021.
  8. Adversarial balancing-based representation learning for causal effect inference with observational data. Data Mining and Knowledge Discovery, 35(4):1713–1738, 2021.
  9. Hill, J. L. Bayesian nonparametric modeling for causal inference. Journal of Computational and Graphical Statistics, 20(1), 2011.
  10. Causal normalizing flows: from theory to practice. arXiv preprint arXiv:2306.05415, 2023.
  11. Learning representations for counterfactual inference. In International conference on machine learning, pp.  3020–3029. PMLR, 2016.
  12. Causal autoregressive flows. In International conference on artificial intelligence and statistics, pp.  3520–3528. PMLR, 2021.
  13. Quantile regression. Journal of economic perspectives, 15(4):143–156, 2001.
  14. The omniglot challenge: a 3-year progress report. Current Opinion in Behavioral Sciences, 29:97–104, 2019.
  15. Mnist handwritten digit database. ATT Labs [Online]. Available: http://yann.lecun.com/exdb/mnist, 2, 2010.
  16. Contrastive individual treatment effects estimation. In 2022 IEEE International Conference on Data Mining (ICDM), pp.  1053–1058. IEEE, 2022.
  17. Sample-efficient reinforcement learning via counterfactual-based data augmentation. arXiv preprint arXiv:2012.09092, 2020a.
  18. Reconsidering generative objectives for counterfactual reasoning. Advances in Neural Information Processing Systems, 33:21539–21553, 2020b.
  19. Causal transformer for estimating counterfactual outcomes. In International Conference on Machine Learning, pp.  15293–15329. PMLR, 2022.
  20. Foundations of machine learning. MIT press, 2018.
  21. Counterfactual (non-) identifiability of learned structural causal models. arXiv preprint arXiv:2301.09031, 2023.
  22. Counterfactual identifiability of bijective causal models. arXiv preprint arXiv:2302.02228, 2023.
  23. Deep structural causal models for tractable counterfactual inference. Advances in Neural Information Processing Systems, 33:857–869, 2020.
  24. Pearl, J. Causality: Models, Reasoning and Inference. Cambridge University Press, 2000.
  25. Causal inference in statistics: A primer. John Wiley Sons, 2016.
  26. Powell, D. Quantile treatment effects in the presence of covariates. Review of Economics and Statistics, 102(5):994–1005, 2020.
  27. High fidelity image counterfactuals with probabilistic causal models. arXiv preprint arXiv:2306.15764, 2023.
  28. Diffusion causal models for counterfactual estimation. arXiv preprint arXiv:2202.10166, 2022.
  29. Denoising diffusion implicit models. arXiv preprint arXiv:2010.02502, 2020.
  30. Single-model uncertainties for deep learning. Advances in Neural Information Processing Systems, 32, 2019.
  31. Nonparametric quantile estimation. 2006.
  32. Nonparametric conditional density estimation using piecewise-linear solution path of kernel quantile regression. Neural Computation, 21(2):533–559, 2009.
  33. The causal-neural connection: Expressiveness, learnability, and inference. Advances in Neural Information Processing Systems, 34:10823–10836, 2021.
  34. Neural causal models for counterfactual identification and estimation. arXiv preprint arXiv:2210.00035, 2022.
  35. Multiply robust estimation of causal quantile treatment effects. Statistics in Medicine, 39(28):4238–4251, 2020.
  36. Representation learning for treatment effect estimation from observational data. In Neural Information Processing Systems, 2018.
  37. Ganite: Estimation of individualized treatment effects using generative adversarial nets. In International conference on learning representations, 2018.
  38. On estimation of functional causal models: general results and application to the post-nonlinear causal model. ACM Transactions on Intelligent Systems and Technology (TIST), 7(2):1–22, 2015.
  39. Cycle-balanced representation learning for counterfactual inference. In Proceedings of the 2022 SIAM International Conference on Data Mining (SDM), pp.  442–450. SIAM, 2022.
  40. Estimating potential outcome distributions with collaborating causal networks. arXiv preprint arXiv:2110.01664, 2021a.
  41. Estimating uncertainty intervals from collaborating networks. The Journal of Machine Learning Research, 22(1):11645–11691, 2021b.

Summary

We haven't generated a summary for this paper yet.