Disentangled and reassociated deep representation for dynamic survival analysis with competing risks

被引:0
作者
Cui, Chang [1 ,2 ]
Tang, Yongqiang [1 ]
Zhang, Wensheng [1 ,2 ]
机构
[1] Chinese Acad Sci, Inst Automat, State Key Lab Multimodal Artificial Intelligence S, Beijing 100190, Peoples R China
[2] Univ Chinese Acad Sci, Sch Artificial Intelligence, Beijing 100049, Peoples R China
基金
中国国家自然科学基金;
关键词
Survival analysis; Deep learning; Longitudinal data; Competing risks; Contrastive learning; REGRESSION-MODELS; PNEUMONIA; FAILURE;
D O I
10.1016/j.knosys.2025.113295
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Survival analysis has been extensively utilized to analyze when the event of interest occurs. However, most of present studies merely focus on single risk and static data, while incapable of handling the scenario where competing risks and longitudinal observations are involved, which is prevalent in clinical practice, especially in the ICU. Although some impressive progress has been made in recent years, they generally utilize a single encoder to learn patient representations and input identical representations into each cause-specific subnetwork to learn the survival distribution of competing risks, thereby neglecting the specificity and association of each risk factor. In this study, we propose a novel model, namely competing risks disentangled and reassociated deep representation for dynamic survival analysis. On one hand, we propose risks-disentangled autoencoders to learn specific representations for each risk factor with contrastive learning. On the other hand, a risks-reassociated representation fusion module is proposed to explicitly learn the association relationships among competing risk representations with attention mechanism. Through extensive experiments on two popular clinical datasets, i.e., MIMIC-III and eICU, we demonstrate that our proposed model achieves advanced survival prediction performance. Visualization and interpretability analysis experiments are also provided to indicate the superior performance of our model.
引用
收藏
页数:13
相关论文
共 60 条
[1]   Machine learning model for predicting physical activity related bleeding risk in Chinese boys with haemophilia A [J].
Ai, Di ;
Cui, Chang ;
Tang, Yongqiang ;
Wang, Yan ;
Zhang, Ningning ;
Zhang, Chenyang ;
Zhen, Yingzi ;
Li, Gang ;
Huang, Kun ;
Liu, Guoqing ;
Chen, Zhenping ;
Zhang, Wensheng ;
Wu, Runhui .
THROMBOSIS RESEARCH, 2023, 232 :43-53
[2]   A time-dependent discrimination index for survival data [J].
Antolini, L ;
Boracchi, P ;
Biganzoli, E .
STATISTICS IN MEDICINE, 2005, 24 (24) :3927-3944
[3]   Multi-modal advanced deep learning architectures for breast cancer survival prediction [J].
Arya, Nikhilanand ;
Saha, Sriparna .
KNOWLEDGE-BASED SYSTEMS, 2021, 221
[4]  
Beaulieu-Jones BK, 2018, BIOCOMPUT-PAC SYM, P123
[5]  
Bender Andreas, 2020, JOINT EUR C MACH LEA, P158
[6]   Variational Bayes survival analysis for unemployment modelling [J].
Boskoski, Pavle ;
Perne, Matija ;
Ramesa, Martina ;
Boshkoska, Biljana Mileva .
KNOWLEDGE-BASED SYSTEMS, 2021, 229
[7]   Random forests [J].
Breiman, L .
MACHINE LEARNING, 2001, 45 (01) :5-32
[8]   Graph Structure Aware Contrastive Multi-View Clustering [J].
Chen, Rui ;
Tang, Yongqiang ;
Cai, Xiangrui ;
Yuan, Xiaojie ;
Feng, Wenlong ;
Zhang, Wensheng .
IEEE TRANSACTIONS ON BIG DATA, 2024, 10 (03) :260-274
[9]   Acute respiratory failure in patients with severe community-acquired pneumonia - A prospective randomized evaluation of noninvasive ventilation [J].
Confalonieri, M ;
Potena, A ;
Carbone, G ;
Della Porta, R ;
Tolley, EA ;
Meduri, GU .
AMERICAN JOURNAL OF RESPIRATORY AND CRITICAL CARE MEDICINE, 1999, 160 (05) :1585-1591
[10]  
COX DR, 1972, J R STAT SOC B, V34, P187