Ensemble-Based Risk Scoring with Extreme Learning Machine for Prediction of Adverse Cardiac Events

• Main Authors: Liu, Nan, Sakamoto, Jeffrey Tadashi, Cao, Jiuwen, Koh, Zhi Xiong, Ho, Andrew Fu Wah, Lin, Zhiping, Ong, Marcus Eng Hock
• Other Authors: School of Electrical and Electronic Engineering
• Format: Article
• Language: English
• Published: 2018
• Subjects: Extreme Learning Machine; Ensemble Learning
• Online Access: https://hdl.handle.net/10356/86898; http://hdl.handle.net/10220/44244
• Institution: Nanyang Technological University

Accurate prediction of adverse cardiac events for the emergency department (ED) chest pain patients is essential in risk stratification due to the current ambiguity in diagnosing acute coronary syndrome. While most current practices rely on human decision by measuring clinical vital signs, computerized solutions are gaining popularity. We have previously proposed an ensemble-based scoring system (ESS). In this paper, we aim to extend the ESS system using extreme learning machine (ELM), a fast learning algorithm for neural networks. We recruited patients from the ED of Singapore General Hospital, and extracted features such as heart rate variability, 12-lead ECG parameters, and vital signs. We also proposed a novel algorithm called ESS-ELM to predict adverse cardiac events. Different from the original ESS algorithm, ESS-ELM uses the under-sampling technique only in model training. Our proposed method was compared to the original ESS algorithm and several clinical scores in predicting patient outcome. With a cohort of 797 recruited patients, we demonstrated that ESS-ELM outperformed the original ESS algorithm and three established clinical scores, namely HEART, TIMI, and GRACE, in terms of receiver operating characteristic analysis. Furthermore, we have investigated the impact of hidden node number and ensemble size on the predictive performance. ELM has demonstrated the flexibility in its integration with the ESS algorithm. Experiments showed the value of ESS-ELM in prediction of adverse cardiac events. Future works may include the use of new ELM-based learning methods and further validation with a new cohort of patients.