Survival prediction model for right-censored data based on improved composite quantile regression neural network

被引:3
作者
Qin, Xiwen [1 ]
Yin, Dongmei [1 ]
Dong, Xiaogang [1 ]
Chen, Dongxue [1 ]
Zhang, Shuang [1 ]
机构
[1] Changchun Univ Technol, Sch Math & Stat, Changchun 130012, Peoples R China
基金
中国国家自然科学基金;
关键词
right-censored data; composite quantile regression; neural network; whale optimization algorithm; feature selection; PROGNOSIS; SELECTION;
D O I
10.3934/mbe.2022354
中图分类号
Q [生物科学];
学科分类号
07 ; 0710 ; 09 ;
摘要
With the development of the field of survival analysis, statistical inference of right-censored data is of great importance for the study of medical diagnosis. In this study, a right-censored data survival prediction model based on an improved composite quantile regression neural network framework, called rcICQRNN, is proposed. It incorporates composite quantile regression with the loss function of a multi-hidden layer feedforward neural network, combined with an inverse probability weighting method for survival prediction. Meanwhile, the hyperparameters involved in the neural network are adjusted using the WOA algorithm, integer encoding and One-Hot encoding are implemented to encode the classification features, and the BWOA variable selection method for high dimensional data is proposed. The rcICQRNN algorithm was tested on a simulated dataset and two real breast cancer datasets, and the performance of the model was evaluated by three evaluation metrics. The results show that the rcICQRNN-5 model is more suitable for analyzing simulated datasets. The One-Hot encoding of the WOA-rcICQRNN-30 model is more applicable to the NKI70 data. The model results are optimal for ?????????????????? L 15 after feature selection for the METABRIC dataset. Finally, we implemented the method for cross-dataset validation. On the whole, the Cindex results using One-Hot encoding data are more stable, making the proposed rcICQRNN prediction model flexible enough to assist in medical decision making. It has practical applications in areas such as biomedicine, insurance actuarial and financial economics.
引用
收藏
页码:7521 / 7542
页数:22
相关论文
共 35 条
[1]   Multi-modal advanced deep learning architectures for breast cancer survival prediction [J].
Arya, Nikhilanand ;
Saha, Sriparna .
KNOWLEDGE-BASED SYSTEMS, 2021, 221
[2]   Generating survival times to simulate Cox proportional hazards models with time-varying covariates [J].
Austin, Peter C. .
STATISTICS IN MEDICINE, 2012, 31 (29) :3946-3958
[3]   Composite kernel quantile regression [J].
Bang, Sungwan ;
Eo, Soo-Heang ;
Jhun, Myoungshic ;
Cho, Hyung Jun .
COMMUNICATIONS IN STATISTICS-SIMULATION AND COMPUTATION, 2017, 46 (03) :2228-2240
[4]   Adaptive lasso penalised censored composite quantile regression [J].
Bang, Sungwan ;
Cho, Hyungjun ;
Jhun, Myoungshic .
INTERNATIONAL JOURNAL OF DATA MINING AND BIOINFORMATICS, 2016, 15 (01) :22-46
[5]  
Biganzoli E, 1998, STAT MED, V17, P1169, DOI 10.1002/(SICI)1097-0258(19980530)17:10<1169::AID-SIM796>3.3.CO
[6]  
2-4
[7]   Quantile regression neural networks: Implementation in R and application to precipitation downscaling [J].
Cannon, Alex J. .
COMPUTERS & GEOSCIENCES, 2011, 37 (09) :1277-1284
[8]   Deep learning with multimodal representation for pancancer prognosis prediction [J].
Cheerla, Anika ;
Gevaert, Olivier .
BIOINFORMATICS, 2019, 35 (14) :I446-I454
[9]   SURVIVAL ANALYSIS OF CENSORED-DATA - NEURAL-NETWORK ANALYSIS DETECTION OF COMPLEX INTERACTIONS BETWEEN VARIABLES [J].
DELAURENTIIS, M ;
RAVDIN, PM .
BREAST CANCER RESEARCH AND TREATMENT, 1994, 32 (01) :113-118
[10]   Marshall-Olkin frailty survival models for bivariate right-censored failure time data [J].
Giussani, A. ;
Bonetti, M. .
JOURNAL OF APPLIED STATISTICS, 2019, 46 (16) :2945-2961