Online Program Home
  My Program

All Times EDT

Abstract Details

Activity Number: 349 - Lifetime Data Science Student Awards
Type: Topic-Contributed
Date/Time: Thursday, August 12, 2021 : 10:00 AM to 11:50 AM
Sponsor: Lifetime Data Science Section
Abstract #317197
Title: Deep Learning for Quantile Regression under Right Censoring: DeepQuantreg
Author(s): Yichen Jia* and Jong-Hyeon Jeong
Companies: University of Pittsburgh and University of Pittsburgh
Keywords: Huber Check Function; Inverse Probability Censoring Weights (IPCW); Neural Network; Right Censoring; Survival Analysis; Time to Event

The computational prediction algorithm of neural network has drawn much attention recently in statistics as well as in image recognition and natural language processing. Particularly in statistical application for censored survival data, the loss function used for optimization has been mainly based on the partial likelihood from Cox's model and its variations. This paper presents a novel application of the neural network to the quantile regression for survival data with right censoring, which is adjusted by the inverse of the estimated censoring distribution in the check function. The main purpose of this work is to show that the deep learning method could be flexible enough to predict nonlinear patterns more accurately compared to existing quantile regression methods, emphasizing on practicality of the method for censored survival data. Simulation studies were performed to generate nonlinear censored survival data and compare the deep learning method with the traditional quantile regression and nonparametric quantile regression methods in terms of prediction accuracy. The proposed method is illustrated with two publicly available breast cancer data sets with gene signatures.

Authors who are presenting talks have a * after their name.

Back to the full JSM 2021 program