TripleSurv: Triplet Time-Adaptive Coordinate Learning Approach for Survival Analysis

被引:0
|
作者
Zhang, Liwen [1 ]
Zhong, Lianzhen [2 ,3 ,4 ]
Yang, Fan [1 ]
Tang, Linglong [5 ]
Dong, Di [1 ]
Hui, Hui [1 ]
Tian, Jie [6 ]
机构
[1] Chinese Acad Sci, Inst Automat, Key Lab Mol Imaging, Beijing 100190, Peoples R China
[2] China Mobile Hangzhou Informat Technol Co Ltd, Hangzhou 310000, Peoples R China
[3] Univ Chinese Acad Sci, Beijing 100049, Peoples R China
[4] Chinese Acad Sci, Inst Automat, Key Lab Mol Imaging, Beijing 100190, Peoples R China
[5] Sun Yat Sen Univ, Dept Radiat Oncol, Canc Ctr, Guangzhou 510060, Peoples R China
[6] Beihang Univ, Beijing Adv Innovat Ctr Big Data Based Precis Med, Sch Med, Beijing 100191, Peoples R China
基金
中国国家自然科学基金;
关键词
Calibration; likelihood estimation; neural network; survival analysis;
D O I
10.1109/TKDE.2024.3450910
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
A core challenge in survival analysis is to model the distribution of time-to-event data, where the event of interest may be a death, failure, or occurrence of a specific event. Previous studies have showed that ranking and maximum likelihood estimation loss functions are widely-used learning approaches for survival analysis. However, ranking loss only focus on the ranking of survival time and does not consider potential effect of samples' exact survival time values. Furthermore, the maximum likelihood estimation is unbounded and easily subject to outliers (e.g., censored data), which may cause poor performance of modeling. To handle the complexities of learning process and exploit valuable survival time values, we propose a time-adaptive coordinate loss function, TripleSurv, to achieve adaptive adjustments by introducing the differences in the survival time between sample pairs into the ranking, which can encourage the model to quantitatively rank relative risk of pairs, ultimately enhancing the accuracy of predictions. Most importantly, the TripleSurv is proficient in quantifying the relative risk between samples by ranking ordering of pairs, and consider the time interval as a trade-off to calibrate the robustness of model over sample distribution. Our TripleSurv is evaluated on three real-world survival datasets and a public synthetic dataset. The results show that our method outperforms the state-of-the-art methods and exhibits good model performance and robustness on modeling various sophisticated data distributions with different censor rates.
引用
收藏
页码:9464 / 9475
页数:12
相关论文
共 50 条
  • [31] A Machine Learning Approach for Real-Time Reachability Analysis
    Allen, Ross E.
    Clark, Ashley A.
    Starek, Joseph A.
    Pavone, Marco
    2014 IEEE/RSJ INTERNATIONAL CONFERENCE ON INTELLIGENT ROBOTS AND SYSTEMS (IROS 2014), 2014, : 2202 - 2208
  • [32] Machine learning pre-hospital real-time cardiac arrest outcome prediction (PReCAP) using time-adaptive cohort model based on the Pan-Asian Resuscitation Outcome Study
    Chang H.
    Kim J.W.
    Jung W.
    Heo S.
    Lee S.U.
    Kim T.
    Hwang S.Y.
    Do Shin S.
    Cha W.C.
    Ong M.
    Scientific Reports, 13 (1)
  • [33] CRESA: A Deep Learning Approach to Competing Risks, Recurrent Event Survival Analysis
    Gupta, Garima
    Sunder, Vishal
    Prasad, Ranjitha
    Shroff, Gautam
    ADVANCES IN KNOWLEDGE DISCOVERY AND DATA MINING, PAKDD 2019, PT II, 2019, 11440 : 108 - 122
  • [34] DeepSurv landmarking: a deep learning approach for dynamic survival analysis with longitudinal data
    Wang, Yixuan
    Xie, Jialiang
    Zhao, Xuejing
    JOURNAL OF STATISTICAL COMPUTATION AND SIMULATION, 2025, 95 (01) : 186 - 207
  • [35] An Adaptive Learning Rate for RBFNN Using Time-Domain Feedback Analysis
    Ali, Syed Saad Azhar
    Moinuddin, Muhammad
    Raza, Kamran
    Adil, Syed Hasan
    SCIENTIFIC WORLD JOURNAL, 2014,
  • [36] Adaptive importance sampling approach for structural time-variant reliability analysis
    Yuan, Xiukai
    Shu, Yunfei
    Qian, Yugeng
    Dong, Yiwei
    STRUCTURAL SAFETY, 2024, 111
  • [37] University dropout - A structural equation approach to discrete-time survival analysis
    Voelkle, Manuel C.
    Sander, Nicolas
    JOURNAL OF INDIVIDUAL DIFFERENCES, 2008, 29 (03) : 134 - 147
  • [38] Penalized Likelihood Approach for Simultaneous Analysis of Survival Time and Binary Longitudinal Outcome
    Choi J.
    Cai J.
    Zeng D.
    Sankhya B, 2017, 79 (2) : 190 - 216
  • [39] Integrated survival analysis using an event-time approach in a Bayesian framework
    Walsh, Daniel P.
    Dreitz, Victoria J.
    Heisey, Dennis M.
    ECOLOGY AND EVOLUTION, 2015, 5 (03): : 769 - 780
  • [40] An Alternative Approach for the Analysis of Time-to-Event and Survival Outcomes in Pulmonary Medicine
    Harhay, Michael O.
    Porcher, Raphael
    Cantu, Edward
    Crowther, Michael J.
    Christie, Jason D.
    Thabut, Gabriel
    Donaldson, Gavin C.
    AMERICAN JOURNAL OF RESPIRATORY AND CRITICAL CARE MEDICINE, 2018, 198 (05) : 684 - 687