Open Access
Translator Disclaimer
2021 Bayesian Survival Tree Ensembles with Submodel Shrinkage
Antonio R. Linero, Piyali Basak, Yinpu Li, Debajyoti Sinha
Author Affiliations +
Bayesian Anal. Advance Publication 1-24 (2021). DOI: 10.1214/21-BA1285

Abstract

We consider Bayesian nonparametric estimation of a survival time subject to right-censoring in the presence of potentially high-dimensional predictors. We argue that several approaches, such as random survival forests and existing Bayesian nonparametric approaches, possess several drawbacks, including: computational difficulties; lack of known theoretical properties; and ineffectiveness at filtering out irrelevant predictors. We propose two models based on the Bayesian additive regression trees (BART) framework. The first, Modulated BART (MBART), is fully-nonparametric and models the failure time as the first occurrence of a non-homogeneous Poisson process. The second, CoxBART, uses a Bayesian implementation of Cox’s partial likelihood. These models are adapted to high-dimensional predictors, have default prior specifications, and require simple modifications of existing BART methods to implement. We show the effectiveness of these methods on simulated and benchmark datasets. We also establish that, for a simplified variant of MBART, the posterior distribution contracts at a near-minimax optimal rate in a high-dimensional sparse asymptotic regime.

Citation

Download Citation

Antonio R. Linero. Piyali Basak. Yinpu Li. Debajyoti Sinha. "Bayesian Survival Tree Ensembles with Submodel Shrinkage." Bayesian Anal. Advance Publication 1 - 24, 2021. https://doi.org/10.1214/21-BA1285

Information

Published: 2021
First available in Project Euclid: 9 August 2021

Digital Object Identifier: 10.1214/21-BA1285

JOURNAL ARTICLE
24 PAGES


SHARE
Advance Publication
Back to Top