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}
}