DPWTE: A Deep Learning Approach to Time-to-Event Analysis using a Sparse Weibull Mixture Layer
Résumé
n this paper, we propose a deep-network-based approach that leverages a finite mixture of Weibull distributions to address a key challenge in time-to-event modeling: it comes to a parametric estimation of survival time distribution from censored data. The proposed model predicts the parameters of the mixture of Weibull distributions with respect to input variables. In addition, given that we set beforehand the upper bound of the mixture size, the model finds the requisite number of Weibull distributions to combine in order to model the event time distribution as accurately as possible. For this purpose, we introduce the Sparse Weibull Mixture layer that selects indirectly, through its weights, the Weibull dis-tributions composing the mixture, whose mixing parameters are significant. To stimulate this selection, we apply a sparse regularization on this layer by adding a penalty term to the loss function that takes into account both observed and censored observations. We validate our method on both simulated and real datasets, showing that the proposed methodology yields a performance improvement over the state-of-the-art models proposed in this work.
Fichier principal
DPWTE_arxiv (3).pdf (3.12 Mo)
Télécharger le fichier
ArchiOne_journal.png (47.1 Ko)
Télécharger le fichier
ArchiTwo_journal.png (61.91 Ko)
Télécharger le fichier
METABRIC.png (35.6 Ko)
Télécharger le fichier
Normalize_diag.png (34.86 Ko)
Télécharger le fichier
SEER_BC2.png (38.72 Ko)
Télécharger le fichier
SWM_layer.png (40.91 Ko)
Télécharger le fichier
alpha_layer.jpg (23.92 Ko)
Télécharger le fichier
archiOne_vf.png (58.21 Ko)
Télécharger le fichier
process_select.png (111.9 Ko)
Télécharger le fichier
selection_process.png (112.49 Ko)
Télécharger le fichier
threeWeibull_predicted.png (23.59 Ko)
Télécharger le fichier
| Origine | Fichiers produits par l'(les) auteur(s) |
|---|
| Origine | Fichiers produits par l'(les) auteur(s) |
|---|