Abstract
While deep learning approaches to survival data have demonstrated empirical success in applications, most of these methods are difficult to interpret and mathematical understanding of them is lacking. This paper studies the partially linear Cox model, where the nonlinear component of the model is implemented using a deep neural network. The proposed approach is flexible and able to circumvent the curse of dimensionality, yet it facilitates interpretability of the effects of treatment covariates on survival. We establish asymptotic theories of maximum partial likelihood estimators and show that our nonparametric deep neural network estimator achieves the minimax optimal rate of convergence (up to a polylogarithmic factor). Moreover, we prove that the corresponding finite-dimensional estimator for treatment covariate effects is -consistent, asymptotically normal and attains semiparametric efficiency. Extensive simulation studies and analyses of two real survival data sets show the proposed estimator produces confidence intervals with superior coverage as well as survival time predictions with superior concordance to actual survival times.
Funding Statement
The first author was supported by National Science Foundation of China Grant NSFC-11931001 and Key Laboratory of Econometrics (Xiamen University), Ministry of Education.
The third author was supported by NSF Grant DMS-1914917 and NIH Grant UG3-0D023313 (ECHO Program).
Acknowledgments
The authors are grateful to the Editor, Associate Editor and referees for their helpful comments that led to numerous improvements of the paper. The first author also thanks Professor Ying Yang at Department of Mathematical Sciences, Tsinghua University for kind support.
Citation
Qixian Zhong. Jonas Mueller. Jane-Ling Wang. "Deep learning for the partially linear Cox model." Ann. Statist. 50 (3) 1348 - 1375, June 2022. https://doi.org/10.1214/21-AOS2153
Information