Score: 1

Quantifying uncertainty of individualized treatment effects in right-censored survival data: A comparison of Bayesian additive regression trees and causal survival forest

Published: April 6, 2025 | arXiv ID: 2504.04571v1

By: Daijiro Kabata, Nicholas C. Henderson, Ravi Varadhan

BigTech Affiliations: Johns Hopkins University

Potential Business Impact:

Helps doctors know if treatments will work.

Business Areas:
A/B Testing Data and Analytics

Estimation of individualized treatment effects (ITE), also known as conditional average treatment effects (CATE), is an active area of methodology development. However, much less attention has been paid to the quantification of uncertainty of ITE/CATE estimates in right-censored survival data. Here we undertake an extensive simulation study to examine the coverage of interval estimates from two popular estimation algorithms, Bayesian additive regression trees (BART) and causal survival forest (CSF). We conducted simulation designs from 3 different settings: first, in a setting where BART was developed for an accelerated failure time model; second, where CSF was developed; and finally, a ``neutral'' simulation taken from a setting where neither BART nor CSF was developed. BART outperformed CSF in all three simulation settings. Both the BART and CSF algorithms involve multiple hyperparameters, and BART credible intervals had better coverage than the CSF confidence intervals under the default values, as well as under optimized values, of these hyperparameters.

Country of Origin
🇺🇸 United States

Page Count
22 pages

Category
Statistics:
Methodology