Deep Neural Network Based Accelerated Failure Time Models using Rank Loss
An accelerated failure time (AFT) model assumes a log-linear relationship
between failure times and a set of covariates. In contrast to other popular
survival models that work on hazard functions, the effects of covariates are
directly on failure times, whose interpretation is intuitive. The
semiparametric AFT model that does not specify the error distribution is
flexible and robust to departures from the distributional assumption. Owing to
the desirable features, this class of models has been considered as a promising
alternative to the popular Cox model in the analysis of censored failure time
data. However, in these AFT models, a linear predictor for the mean is
typically assumed. Little research has addressed the nonlinearity of predictors
when modeling the mean. Deep neural networks (DNNs) have received a focal
attention over the past decades and have achieved remarkable success in a
variety of fields. DNNs have a number of notable advantages and have been shown
to be particularly useful in addressing the nonlinearity. By taking advantage
of this, we propose to apply DNNs in fitting AFT models using a Gehan-type
loss, combined with a sub-sampling technique. Finite sample properties of the
proposed DNN and rank based AFT model (DeepR-AFT) are investigated via an
extensive stimulation study. DeepR-AFT shows a superior performance over its
parametric or semiparametric counterparts when the predictor is nonlinear. For
linear predictors, DeepR-AFT performs better when the dimensions of covariates
are large. The proposed DeepR-AFT is illustrated using two real datasets, which
demonstrates its superiority.