We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:


Current browse context:


Change to browse by:

References & Citations


(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Statistics > Applications

Title: Estimating Heterogeneous Survival Treatment Effect via Machine/Deep Learning Methods in Observational Studies

Abstract: The rise of personalized medicine necessitates improved causal inference methods for detecting treatment effect heterogeneity (TEH). Approaches for estimating TEH with observational data have largely focused on continuous outcomes. Methods for estimating TEH with right-censored survival outcomes are relatively limited and have been less vetted. Using flexible machine/deep learning (ML/DL) methods within the counterfactual framework is a promising approach to address challenges due to complex individual characteristics, to which treatments need to be tailored. We contribute a series of simulations representing a variety of confounded heterogenous survival treatment effect settings with varying degrees of covariate overlap, and compare the operating characteristics of three state-of-the-art survival ML/DL methods for the estimation of TEH. Our results show that the nonparametric Bayesian Additive Regression Trees within the framework of accelerated failure time model (AFT-BART-NP) consistently has the best performance, in terms of both bias and root-mean-squared-error. Additionally, AFT-BART-NP could provide nominal confidence interval coverage when covariate overlap is moderate or strong. Under lack of overlap where accurate estimation of the average causal effect is generally challenging, AFT-BART-NP still provides valid point and interval estimates for the treatment effect among units near the centroid of the propensity score distribution.
Comments: 24 pages, 3 figures, 2 tables
Subjects: Applications (stat.AP); Machine Learning (stat.ML)
Cite as: arXiv:2008.07044 [stat.AP]
  (or arXiv:2008.07044v1 [stat.AP] for this version)

Submission history

From: Liangyuan Hu [view email]
[v1] Mon, 17 Aug 2020 01:02:14 GMT (663kb)
[v2] Mon, 26 Oct 2020 14:09:58 GMT (5820kb,D)
[v3] Fri, 19 Feb 2021 19:38:10 GMT (4800kb,D)
[v4] Wed, 19 May 2021 15:54:08 GMT (7785kb,D)

Link back to: arXiv, form interface, contact.