Promotion time cure rate model with a neural network estimated nonparametric component

Stat Med. 2021 Jul 10;40(15):3516-3532. doi: 10.1002/sim.8980. Epub 2021 Apr 29.

Abstract

Promotion time cure rate models (PCM) are often used to model the survival data with a cure fraction. Medical images or biomarkers derived from medical images can be the key predictors in survival models. However, incorporating images in the PCM is challenging using traditional nonparametric methods such as splines. We propose to use neural network to model the nonparametric or unstructured predictors' effect in the PCM context. Expectation-maximization algorithm with neural network for the M-step is used for parameter estimation. Asymptotic properties of the proposed estimates are derived. Simulation studies show good performance in terms of both prediction and estimation. We finally apply our methods to analyze the brain images from open access series of imaging studies data.

Keywords: EM algorithm; convergence rate; cure rate models; machine learning; survival analysis.

Publication types

  • Research Support, N.I.H., Extramural
  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms*
  • Computer Simulation
  • Humans
  • Neural Networks, Computer*