IDEAS home Printed from https://ideas.repec.org/a/bla/biomet/v79y2023i4p3023-3037.html
   My bibliography  Save this article

Nonparametric failure time: Time‐to‐event machine learning with heteroskedastic Bayesian additive regression trees and low information omnibus Dirichlet process mixtures

Author

Listed:
  • Rodney A. Sparapani
  • Brent R. Logan
  • Martin J. Maiers
  • Purushottam W. Laud
  • Robert E. McCulloch

Abstract

Many popular survival models rely on restrictive parametric, or semiparametric, assumptions that could provide erroneous predictions when the effects of covariates are complex. Modern advances in computational hardware have led to an increasing interest in flexible Bayesian nonparametric methods for time‐to‐event data such as Bayesian additive regression trees (BART). We propose a novel approach that we call nonparametric failure time (NFT) BART in order to increase the flexibility beyond accelerated failure time (AFT) and proportional hazard models. NFT BART has three key features: (1) a BART prior for the mean function of the event time logarithm; (2) a heteroskedastic BART prior to deduce a covariate‐dependent variance function; and (3) a flexible nonparametric error distribution using Dirichlet process mixtures (DPM). Our proposed approach widens the scope of hazard shapes including nonproportional hazards, can be scaled up to large sample sizes, naturally provides estimates of uncertainty via the posterior and can be seamlessly employed for variable selection. We provide convenient, user‐friendly, computer software that is freely available as a reference implementation. Simulations demonstrate that NFT BART maintains excellent performance for survival prediction especially when AFT assumptions are violated by heteroskedasticity. We illustrate the proposed approach on a study examining predictors for mortality risk in patients undergoing hematopoietic stem cell transplant (HSCT) for blood‐borne cancer, where heteroskedasticity and nonproportional hazards are likely present.

Suggested Citation

  • Rodney A. Sparapani & Brent R. Logan & Martin J. Maiers & Purushottam W. Laud & Robert E. McCulloch, 2023. "Nonparametric failure time: Time‐to‐event machine learning with heteroskedastic Bayesian additive regression trees and low information omnibus Dirichlet process mixtures," Biometrics, The International Biometric Society, vol. 79(4), pages 3023-3037, December.
    • Handle: RePEc:bla:biomet:v:79:y:2023:i:4:p:3023-3037
    DOI: 10.1111/biom.13857
    as

    Download full text from publisher

    File URL: https://doi.org/10.1111/biom.13857
    Download Restriction: no
    File URL: https://libkey.io/10.1111/biom.13857?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    References listed on IDEAS

    as
    1. Yi Liu & Veronika Ročková, 2023. "Variable Selection Via Thompson Sampling," Journal of the American Statistical Association, Taylor & Francis Journals, vol. 118(541), pages 287-304, January.
    2. Antonio R. Linero, 2018. "Bayesian Regression Trees for High-Dimensional Prediction and Variable Selection," Journal of the American Statistical Association, Taylor & Francis Journals, vol. 113(522), pages 626-636, April.
    3. P. Richard Hahn & Carlos M. Carvalho, 2015. "Decoupling Shrinkage and Selection in Bayesian Linear Models: A Posterior Summary Perspective," Journal of the American Statistical Association, Taylor & Francis Journals, vol. 110(509), pages 435-448, March.
    4. Yang, Mingan & Dunson, David B. & Baird, Donna, 2010. "Semiparametric Bayes hierarchical models with mean and variance constraints," Computational Statistics & Data Analysis, Elsevier, vol. 54(9), pages 2172-2186, September.
    Full references (including those not matched with items on IDEAS)

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Sally Paganin & Christopher J. Paciorek & Claudia Wehrhahn & Abel Rodríguez & Sophia Rabe-Hesketh & Perry de Valpine, 2023. "Computational Strategies and Estimation Performance With Bayesian Semiparametric Item Response Theory Models," Journal of Educational and Behavioral Statistics, , vol. 48(2), pages 147-188, April.
    2. Donelli, Nicola & Peluso, Stefano & Mira, Antonietta, 2021. "A Bayesian semiparametric vector Multiplicative Error Model," Computational Statistics & Data Analysis, Elsevier, vol. 161(C).
    3. David Kohns & Tibor Szendrei, 2021. "Decoupling Shrinkage and Selection for the Bayesian Quantile Regression," Papers 2107.08498, arXiv.org.
    4. Sarah Brown & Pulak Ghosh & Bhuvanesh Pareek & Karl Taylor, 2017. "Financial Hardship and Saving Behaviour: Bayesian Analysis of British Panel Data," Working Papers 2017011, The University of Sheffield, Department of Economics.
    5. Hauzenberger Niko & Huber Florian & Koop Gary, 2024. "Dynamic Shrinkage Priors for Large Time-Varying Parameter Regressions Using Scalable Markov Chain Monte Carlo Methods," Studies in Nonlinear Dynamics & Econometrics, De Gruyter, vol. 28(2), pages 201-225, April.
    6. Cai, Jing-Heng & Song, Xin-Yuan & Lam, Kwok-Hap & Ip, Edward Hak-Sing, 2011. "A mixture of generalized latent variable models for mixed mode and heterogeneous data," Computational Statistics & Data Analysis, Elsevier, vol. 55(11), pages 2889-2907, November.
    7. Yang Li & Asim Ansari, 2014. "A Bayesian Semiparametric Approach for Endogeneity and Heterogeneity in Choice Models," Management Science, INFORMS, vol. 60(5), pages 1161-1179, May.
    8. Weixian Wang & Maozai Tian, 2024. "Variational Bayesian EM Algorithm for Quantile Regression in Linear Mixed Effects Models," Mathematics, MDPI, vol. 12(21), pages 1-16, October.
    9. Song, Xin-Yuan & Chen, Fei & Lu, Zhao-Hua, 2013. "A Bayesian semiparametric dynamic two-level structural equation model for analyzing non-normal longitudinal data," Journal of Multivariate Analysis, Elsevier, vol. 121(C), pages 87-108.
    10. Qifan Song & Guang Cheng, 2020. "Bayesian Fusion Estimation via t Shrinkage," Sankhya A: The Indian Journal of Statistics, Springer;Indian Statistical Institute, vol. 82(2), pages 353-385, August.
    11. Huber, Florian & Koop, Gary & Onorante, Luca & Pfarrhofer, Michael & Schreiner, Josef, 2023. "Nowcasting in a pandemic using non-parametric mixed frequency VARs," Journal of Econometrics, Elsevier, vol. 232(1), pages 52-69.
    12. Tang, Nian-Sheng & Tang, An-Min & Pan, Dong-Dong, 2014. "Semiparametric Bayesian joint models of multivariate longitudinal and survival data," Computational Statistics & Data Analysis, Elsevier, vol. 77(C), pages 113-129.
    13. Florian Huber & Luca Rossini, 2020. "Inference in Bayesian Additive Vector Autoregressive Tree Models," Papers 2006.16333, arXiv.org, revised Mar 2021.
    14. Jan Prüser & Florian Huber, 2024. "Nonlinearities in macroeconomic tail risk through the lens of big data quantile regressions," Journal of Applied Econometrics, John Wiley & Sons, Ltd., vol. 39(2), pages 269-291, March.
    15. Robert B. Gramacy, 2020. "Discussion," International Statistical Review, International Statistical Institute, vol. 88(2), pages 326-329, August.
    16. Niu, Zibo & Wang, Chenlu & Zhang, Hongwei, 2023. "Forecasting stock market volatility with various geopolitical risks categories: New evidence from machine learning models," International Review of Financial Analysis, Elsevier, vol. 89(C).
    17. Alberto Caron & Gianluca Baio & Ioanna Manolopoulou, 2022. "Estimating individual treatment effects using non‐parametric regression models: A review," Journal of the Royal Statistical Society Series A, Royal Statistical Society, vol. 185(3), pages 1115-1149, July.
    18. Daniel R. Kowal, 2023. "Subset selection for linear mixed models," Biometrics, The International Biometric Society, vol. 79(3), pages 1853-1867, September.
    19. Niko Hauzenberger & Michael Pfarrhofer & Luca Rossini, 2020. "Sparse time-varying parameter VECMs with an application to modeling electricity prices," Papers 2011.04577, arXiv.org, revised Apr 2023.
    20. Byron Botha & Rulof Burger & Kevin Kotzé & Neil Rankin & Daan Steenkamp, 2023. "Big data forecasting of South African inflation," Empirical Economics, Springer, vol. 65(1), pages 149-188, July.

    More about this item

    Statistics

    Access and download statistics

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:bla:biomet:v:79:y:2023:i:4:p:3023-3037. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: Wiley Content Delivery (email available below). General contact details of provider: http://www.blackwellpublishing.com/journal.asp?ref=0006-341X .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.