English

Deep Neural Networks for Semiparametric Frailty Models via H-likelihood

Machine Learning 2023-07-14 v1 Machine Learning Methodology

Abstract

For prediction of clustered time-to-event data, we propose a new deep neural network based gamma frailty model (DNN-FM). An advantage of the proposed model is that the joint maximization of the new h-likelihood provides maximum likelihood estimators for fixed parameters and best unbiased predictors for random frailties. Thus, the proposed DNN-FM is trained by using a negative profiled h-likelihood as a loss function, constructed by profiling out the non-parametric baseline hazard. Experimental studies show that the proposed method enhances the prediction performance of the existing methods. A real data analysis shows that the inclusion of subject-specific frailties helps to improve prediction of the DNN based Cox model (DNN-Cox).

Keywords

Cite

@article{arxiv.2307.06581,
  title  = {Deep Neural Networks for Semiparametric Frailty Models via H-likelihood},
  author = {Hangbin Lee and IL DO HA and Youngjo Lee},
  journal= {arXiv preprint arXiv:2307.06581},
  year   = {2023}
}