2000 character limit reached
Bayesian Nonparametric Trees for Principal Causal Effects (2403.13256v1)
Published 20 Mar 2024 in stat.ME
Abstract: Principal stratification analysis evaluates how causal effects of a treatment on a primary outcome vary across strata of units defined by their treatment effect on some intermediate quantity. This endeavor is substantially challenged when the intermediate variable is continuously scaled and there are infinitely many basic principal strata. We employ a Bayesian nonparametric approach to flexibly evaluate treatment effects across flexibly-modeled principal strata. The approach uses Bayesian Causal Forests (BCF) to simultaneously specify two Bayesian Additive Regression Tree models; one for the principal stratum membership and one for the outcome, conditional on principal strata. We show how the capability of BCF for capturing treatment effect heterogeneity is particularly relevant for assessing how treatment effects vary across the surface defined by continuously-scaled principal strata, in addition to other benefits relating to targeted selection and regularization-induced confounding. The capabilities of the proposed approach are illustrated with a simulation study, and the methodology is deployed to investigate how causal effects of power plant emissions control technologies on ambient particulate pollution vary as a function of the technologies' impact on sulfur dioxide emissions.
- Chestnut, Lauraine G and Mills, David Mauthor (collaboration) (2005), “A fresh look at the benefits and costs of the US acid rain program,” Journal of environmental management, 77, 252–266.
- Chipman, Hugh A and George, Edward I and McCulloch, Robert E— (2010), “BART: Bayesian additive regression trees,” The Annals of Applied Statistics, 4, 266–298.
- Comment, Leah and Mealli, Fabrizia and Haneuse, Sebastien and Zigler, Corwin— (2019), “Survivor average causal effects for continuous time: a principal stratification approach to causal inference with semicompeting risks,” arXiv preprint arXiv:1902.09304.
- Ding, Peng and Li, Fan— (2018), “Causal inference: A Missing Data Perspective,” Statistical Science, 33, 214–237.
- Frangakis, Constantine E and Rubin, Donald B— (2002), “Principal stratification in causal inference,” Biometrics, 58, 21–29.
- Frumento, Paolo and Mealli, Fabrizia and Pacini, Barbara and Rubin, Donald B— (2012), “Evaluating the effect of training on wages in the presence of noncompliance, nonemployment, and missing outcome data,” Journal of the American Statistical Association, 107, 450–466.
- Gilbert, Peter B and Hudgens, Michael G— (2008), “Evaluating candidate principal surrogate endpoints,” Biometrics, 64, 1146–1154.
- Green, Donald P and Kern, Holger L— (2012), “Modeling heterogeneous treatment effects in survey experiments with Bayesian additive regression trees,” Public opinion quarterly, 76, 491–511.
- Hastie, Trevor and Tibshirani, Robert— (2000), “Bayesian backfitting (with comments and a rejoinder by the authors,” Statistical Science, 15, 196–223.
- Henneman, Lucas and Choirat, Christine and Dedoussi, Irene and Dominici, Francesca and Roberts, Jessica and Zigler, Corwin— (2023), “Mortality risk from United States coal electricity generation,” Science, 382, 941–946.
- Hill, Jennifer and Linero, Antonio and Murray, Jared— (2020), “Bayesian additive regression trees: A review and look forward,” Annual Review of Statistics and Its Application, 7.
- Hill, Jennifer L— (2011), “Bayesian nonparametric modeling for causal inference,” Journal of Computational and Graphical Statistics, 20, 217–240.
- Imai, Kosuke and Keele, Luke and Yamamoto, Teppei— (2010), “Identification, inference and sensitivity analysis for causal mediation effects,” Statistical science, 25, 51–71.
- Jin, Hui and Rubin, Donald B— (2008), “Principal stratification for causal inference with extended partial compliance,” Journal of the American Statistical Association, 103, 101–111.
- Kim, Chanmin and Daniels, Michael J and Hogan, Joseph W and Choirat, Christine and Zigler, Corwin M— (2019), “Bayesian methods for multiple mediators: Relating principal stratification and causal mediation in the analysis of power plant emission controls,” The annals of applied statistics, 13, 1927.
- Kim, Chanmin and Henneman, Lucas RF and Choirat, Christine and Zigler, Corwin M— (2020), “Health effects of power plant emissions through ambient air quality,” Journal of the Royal Statistical Society: Series A (Statistics in Society), 183, 1677–1703.
- Kim, Chanmin and Tec, Mauricio and Zigler, Corwin— (2023), “Bayesian nonparametric adjustment of confounding,” Biometrics, 79, 3252–3265.
- Li, Fan and Ding, Peng and Mealli, Fabrizia— (2023), “Bayesian causal inference: a critical review,” Philosophical Transactions of the Royal Society A, 381, 20220153.
- Lunn, David and Best, Nicky and Spiegelhalter, David and Graham, Gordon and Neuenschwander, Beat— (2009), “Combining MCMC with ‘sequential’PKPD modelling,” Journal of Pharmacokinetics and Pharmacodynamics, 36, 19–38.
- Lyu, Tianmeng and Bornkamp, Björn and Mueller-Velten, Guenther and Schmidli, Heinz— (2023), “Bayesian inference for a principal stratum estimand on recurrent events truncated by death,” Biometrics.
- Miratrix, Luke and Furey, Jane and Feller, Avi and Grindal, Todd and Page, Lindsay C— (2018), “Bounding, an accessible method for estimating principal causal effects, examined and explained,” Journal of Research on Educational Effectiveness, 11, 133–162.
- Nevo, Daniel and Gorfine, Malka— (2022), “Causal inference for semi-competing risks data,” Biostatistics, 23, 1115–1132.
- Page, Lindsay C— (2012), “Understanding the impact of career academy attendance: An application of the principal stratification framework for causal effects accounting for partial compliance,” Evaluation Review, 36, 99–132.
- Pearl, J.— (2001), “Direct and indirect effects,” in Proceedings of the seventeenth conference on uncertainty in artificial intelligence, Morgan Kaufmann Publishers Inc., pp. 411–420.
- Robins, J. M. and Greenland, S.— (1992), “Identifiability and exchangeability for direct and indirect effects,” Epidemiology, 143–155.
- Rosenbaum, Paul R and Rubin, Donald B— (1983), “The central role of the propensity score in observational studies for causal effects,” Biometrika, 70, 41–55.
- Rubin, D. B.— (1974), “Estimating causal effects of treatments in randomized and nonrandomized studies,” Journal of educational Psychology, 66, 688.
- Rubin, Donald B— (1980), “Randomization analysis of experimental data: The Fisher randomization test comment,” Journal of the American statistical association, 75, 591–593.
- Schwartz, Scott L and Li, Fan and Mealli, Fabrizia— (2011), “A Bayesian semiparametric approach to intermediate variables in causal inference,” Journal of the American Statistical Association, 106, 1331–1344.
- Tan, Yaoyuan Vincent and Roy, Jason— (2019), “Bayesian additive regression trees and the General BART model,” Statistics in medicine, 38, 5048–5069.
- Tec, Mauricio and Scott, James and Zigler, Corwin— (2022), “Weather2vec: Representation Learning for Causal Inference with Non-Local Confounding in Air Pollution and Climate Studies,” arXiv preprint arXiv:2209.12316.
- VanderWeele, Tyler J— (2008), “Simple relations between principal stratification and direct and indirect effects,” Statistics & Probability Letters, 78, 2957–2962.
- Wei, Y. and Xing, X. and Shtein, A. and Casto, E. and Hultquist, C. and Yazdi, M. D. and Li, L. and Schwartz, J.— (2022), “Daily and Annual PM2.5, O3, and NO2 Concentrations at ZIP Codes for the Contiguous U.S., 2000-2016, v1.0,” .
- Zeldow, Bret and Re III, Vincent Lo and Roy, Jason— (2019), “A semiparametric modeling approach using Bayesian Additive Regression Trees with an application to evaluate heterogeneous treatment effects,” The annals of applied statistics, 13, 1989.
- Zigler, Corwin Matthew— (2016), “The central role of Bayes’ theorem for joint estimation of causal effects and propensity scores,” The American Statistician, 70, 47–54.
- Zigler, Corwin M and Choirat, Christine and Dominici, Francesca— (2018), “Impact of National Ambient Air Quality Standards nonattainment designations on particulate pollution and health,” Epidemiology (Cambridge, Mass.), 29, 165–174.