Survival Prediction with Extreme Learning Machine, Supervised Principal Components and Regularized Cox Models in High-Dimensional Survival Data by Simulation

Abstract

Mortality risks of important diseases such as cancer can be estimated using gene profiles which are high-dimensional data obtained from gene expression sequences. However, it is impossible to analyze high-dimensional data with classical techniques due to multicollinearity, time-consuming processing load, and difficulty interpreting the results. For this purpose, extreme learning machine methods, which can solve regression and classification problems, have become one of the most preferred machine learning methods regarding fast data analysis and ease of application. The goal of this study is to compare estimation performance of risk score and short-term survival with survival extreme learning machine methods, L2-penalty Cox regression, and supervised principal components analysis in generated high-dimensional survival data. The survival models have been evaluated by Harrell’s concordance index, integrated Brier score, F1 score, kappa coefficient, the area under the curve, the area under precision-recall, accuracy, and Matthew’s correlation coefficient. All results showed that survival extreme learning machine methods that allow analyzing high-dimensional survival data without the necessity of dimension reduction perform very competitive with the other popular classical methods used in the study.