Regularizing the Deepsurv Network Using Projection Loss for Medical Risk Assessment

W. Chaovalitwongse,M. Spraker,L. Wootton,Phawis Thammasorn,Paul Kinahan,D. Hippe,S. Combs,Matthew Nyflot,S. Schaub,J. Peeken
DOI: https://doi.org/10.1109/ACCESS.2022.3142032
IF: 3.9
IEEE Access
Abstract:State-of-the-art deep survival prediction approaches expand network parameters to accommodate performance over a fine discretization of output time. For medical applications where data are limited, the regression-based Deepsurv approach is more advantageous because its continuous output design limits unnecessary network parameters. Despite the practical advantage, the typical network lacks control over the feature distribution causing the network to be more prone to noisy information and occasional poor prediction performance. We propose a novel projection loss as a regularizing objective to improve the time-to-event Deepsurv model. The loss formulation maximizes the lower bound of the multiple-correlation coefficient between the network’s features and the desired hazard value. Reducing the loss also theoretically lowers the upper bound on the likelihood of discordant pair and improves C-index performance. We observe superior performances and robustness of regularized Deepsurv over many state-of-the-art approaches in our experiments with five public medical datasets and two cross-cohort validation tasks.
Medicine,Computer Science
What problem does this paper attempt to address?