Regularizing the Deepsurv Network Using Projection Loss for Medical Risk Assessment

Phawis Thammasorn, Stephanie K. Schaub, Daniel S. Hippe, Matthew B. Spraker, Jan C. Peeken, Landon S. Wootton, Paul E. Kinahan, Stephanie E. Combs, Wanpracha A. Chaovalitwongse, Matthew J. Nyflot

Research output: Contribution to journalArticlepeer-review

1 Scopus citations

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.

Original languageEnglish
Pages (from-to)8005-8020
Number of pages16
JournalIEEE Access
Volume10
DOIs
StatePublished - 2022

Keywords

  • Biomedical imaging
  • Data models
  • Hazards
  • Indexes
  • Oncology
  • Task analysis
  • Training

Fingerprint

Dive into the research topics of 'Regularizing the Deepsurv Network Using Projection Loss for Medical Risk Assessment'. Together they form a unique fingerprint.

Cite this