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 条
[51]   One-Step Multiview Subspace Segmentation via Joint Skinny Tensor Learning and Latent Clustering [J].
Tang, Yongqiang ;
Xie, Yuan ;
Zhang, Changqing ;
Zhang, Zhizhong ;
Zhang, Wensheng .
IEEE TRANSACTIONS ON CYBERNETICS, 2022, 52 (09) :9179-9193
[52]   A weighted random survival forest [J].
Utkin, Lev V. ;
Konstantinov, Andrei V. ;
Chukanov, Viacheslav S. ;
Kots, Mikhail V. ;
Ryabinin, Mikhail A. ;
Meldo, Anna A. .
KNOWLEDGE-BASED SYSTEMS, 2019, 177 :136-144
[53]  
van den Oord A, 2019, Arxiv, DOI arXiv:1807.03748
[54]   Machine Learning for Survival Analysis: A Survey [J].
Wang, Ping ;
Li, Yan ;
Reddy, Chandan K. .
ACM COMPUTING SURVEYS, 2019, 51 (06)
[55]   SurvTRACE: Transformers for Survival Analysis with Competing Events [J].
Wang, Zifeng ;
Sun, Jimeng .
13TH ACM INTERNATIONAL CONFERENCE ON BIOINFORMATICS, COMPUTATIONAL BIOLOGY AND HEALTH INFORMATICS, BCB 2022, 2022,
[56]   RISK-FACTORS FOR ACUTE RENAL-INSUFFICIENCY IN PATIENTS WITH SUSPECTED OR DOCUMENTED BACTERIAL PNEUMONIA [J].
WELAGE, LS ;
WALAWANDER, CA ;
TIMM, EG ;
GRASELA, TH .
ANNALS OF PHARMACOTHERAPY, 1994, 28 (04) :515-522
[57]  
Wu ZL, 2021, PR MACH LEARN RES, V149, P54
[58]  
Xiao Q., 2024, IEEE Trans. Pattern Anal. Mach. Intell.
[59]   Clustering Enhanced Multiplex Graph Contrastive Representation Learning [J].
Yuan, Ruiwen ;
Tang, Yongqiang ;
Wu, Yajing ;
Zhang, Wensheng .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2025, 36 (01) :1341-1355
[60]  
Zhong QX, 2021, ADV NEUR IN, V34