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 条
  • [1] A time-adaptive FE2-approach within the method of vertical lines
    Hartmann, Stefan
    Dileep, Pranav Kumar
    Grafenhorst, Matthias
    COMPUTERS & MATHEMATICS WITH APPLICATIONS, 2023, 151 : 222 - 243
  • [2] Analysis of a Particle-In-Cell code based on a time-adaptive mesh
    Schnepp, S.
    Gionaj, E.
    Weiland, T.
    2007 IEEE PARTICLE ACCELERATOR CONFERENCE, VOLS 1-11, 2007, : 1490 - 1492
  • [3] Transient stability assessment of power system with time-adaptive method based on ensemble learning
    Wu S.
    Wang H.
    Dianli Xitong Baohu yu Kongzhi/Power System Protection and Control, 2022, 50 (24): : 112 - 119
  • [4] Time-Adaptive Machine Learning Models for Predicting the Severity of Heart Failure with Reduced Ejection Fraction
    Winger, Trevor
    Ozdemir, Cagri
    Narasimhan, Shanti L.
    Srivastava, Jaideep
    DIAGNOSTICS, 2025, 15 (06)
  • [5] Displacement control in time-adaptive non-linear finite-element analysis
    Hartmann, Stefan
    Quint, Karsten J.
    Hamkar, Ahmad-Wahadj
    ZAMM-ZEITSCHRIFT FUR ANGEWANDTE MATHEMATIK UND MECHANIK, 2008, 88 (05): : 342 - 364
  • [6] An optimized time-adaptive aerothermal coupling calculation method for aerothermal analysis of disc cavity system
    Bi, Shuai
    Mao, Junkui
    Wang, Lei
    Han, Feng
    CHINESE JOURNAL OF AERONAUTICS, 2024, 37 (09) : 164 - 177
  • [7] An optimized time-adaptive aerothermal coupling calculation method for aerothermal analysis of disc cavity system
    Shuai BI
    Junkui MAO
    Lei WANG
    Feng HAN
    Chinese Journal of Aeronautics, 2024, 37 (09) : 164 - 177
  • [8] Time-Adaptive Expectation Maximization Learning Framework for HMM Based Data-Driven Gas Sensor Calibration
    You, Yang
    Oechtering, Tobias J.
    IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2023, 19 (07) : 7986 - 7994
  • [9] Time Series Analysis for the Adaptive Prediction of Total Phosphorus in the Yangtze River: A Machine Learning Approach
    Ma, Tianqi
    Chen, Xing
    WATER, 2025, 17 (04)
  • [10] Time-adaptive loosely coupled analysis on fluid-thermal-structural behaviors of hypersonic wing structures under sustained aeroheating
    Chen, Fang
    Liu, Hong
    Zhang, Shengtao
    AEROSPACE SCIENCE AND TECHNOLOGY, 2018, 78 : 620 - 636