Self-Consistent Equation-guided Neural Networks for Censored Time-to-Event Data

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Kim, Sehwan, Wang, Rui, Lu, Wenbin
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866929755666776064
author Kim, Sehwan
Wang, Rui
Lu, Wenbin
author_facet Kim, Sehwan
Wang, Rui
Lu, Wenbin
contents In survival analysis, estimating the conditional survival function given predictors is often of interest. There is a growing trend in the development of deep learning methods for analyzing censored time-to-event data, especially when dealing with high-dimensional predictors that are complexly interrelated. Many existing deep learning approaches for estimating the conditional survival functions extend the Cox regression models by replacing the linear function of predictor effects by a shallow feed-forward neural network while maintaining the proportional hazards assumption. Their implementation can be computationally intensive due to the use of the full dataset at each iteration because the use of batch data may distort the at-risk set of the partial likelihood function. To overcome these limitations, we propose a novel deep learning approach to non-parametric estimation of the conditional survival functions using the generative adversarial networks leveraging self-consistent equations. The proposed method is model-free and does not require any parametric assumptions on the structure of the conditional survival function. We establish the convergence rate of our proposed estimator of the conditional survival function. In addition, we evaluate the performance of the proposed method through simulation studies and demonstrate its application on a real-world dataset.
format Preprint
id arxiv_https___arxiv_org_abs_2503_09097
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Self-Consistent Equation-guided Neural Networks for Censored Time-to-Event Data
Kim, Sehwan
Wang, Rui
Lu, Wenbin
Machine Learning
Methodology
In survival analysis, estimating the conditional survival function given predictors is often of interest. There is a growing trend in the development of deep learning methods for analyzing censored time-to-event data, especially when dealing with high-dimensional predictors that are complexly interrelated. Many existing deep learning approaches for estimating the conditional survival functions extend the Cox regression models by replacing the linear function of predictor effects by a shallow feed-forward neural network while maintaining the proportional hazards assumption. Their implementation can be computationally intensive due to the use of the full dataset at each iteration because the use of batch data may distort the at-risk set of the partial likelihood function. To overcome these limitations, we propose a novel deep learning approach to non-parametric estimation of the conditional survival functions using the generative adversarial networks leveraging self-consistent equations. The proposed method is model-free and does not require any parametric assumptions on the structure of the conditional survival function. We establish the convergence rate of our proposed estimator of the conditional survival function. In addition, we evaluate the performance of the proposed method through simulation studies and demonstrate its application on a real-world dataset.
title Self-Consistent Equation-guided Neural Networks for Censored Time-to-Event Data
topic Machine Learning
Methodology
url https://arxiv.org/abs/2503.09097