Abstract:Background Predictions of cardiovascular disease risks based on health records have long attracted broad research interests. Despite extensive efforts, the prediction accuracy has remained unsatisfactory. This raises the question as to whether the data insufficiency, statistical and machine-learning methods, or intrinsic noise have hindered the performance of previous approaches, and how these issues can be alleviated. Objective Based on a large population of patients with hypertension in Shenzhen, China, we aimed to establish a high-precision coronary heart disease (CHD) prediction model through big data and machine-learning Methods Data from a large cohort of 42,676 patients with hypertension, including 20,156 patients with CHD onset, were investigated from electronic health records (EHRs) 1-3 years prior to CHD onset (for CHD-positive cases) or during a disease-free follow-up period of more than 3 years (for CHD-negative cases). The population was divided evenly into independent training and test datasets. Various machine-learning methods were adopted on the training set to achieve high-accuracy prediction models and the results were compared with traditional statistical methods and well-known risk scales. Comparison analyses were performed to investigate the effects of training sample size, factor sets, and modeling approaches on the prediction performance. Results An ensemble method, XGBoost, achieved high accuracy in predicting 3-year CHD onset for the independent test dataset with an area under the receiver operating characteristic curve (AUC) value of 0.943. Comparison analysis showed that nonlinear models (K-nearest neighbor AUC 0.908, random forest AUC 0.938) outperform linear models (logistic regression AUC 0.865) on the same datasets, and machine-learning methods significantly surpassed traditional risk scales or fixed models (eg, Framingham cardiovascular disease risk models). Further analyses revealed that using time-dependent features obtained from multiple records, including both statistical variables and changing-trend variables, helped to improve the performance compared to using only static features. Subpopulation analysis showed that the impact of feature design had a more significant effect on model accuracy than the population size. Marginal effect analysis showed that both traditional and EHR factors exhibited highly nonlinear characteristics with respect to the risk scores. Conclusions We demonstrated that accurate risk prediction of CHD from EHRs is possible given a sufficiently large population of training data. Sophisticated machine-learning methods played an important role in tackling the heterogeneity and nonlinear nature of disease prediction. Moreover, accumulated EHR data over multiple time points provided additional features that were valuable for risk prediction. Our study highlights the importance of accumulating big data from EHRs for accurate disease predictions.

A Data-Driven Study of Prediction Methods for Coronary Heart Disease

Predicting Coronary Heart Disease Using a Suite of Machine Learning Models

Machine Learning-Based Risk Prediction for Coronary Heart Disease Using Clinical Data

Effective Heart Disease Prediction Using Machine Learning Techniques

Accurate Prediction of Coronary Heart Disease for Patients With Hypertension From Electronic Health Records With Big Data and Machine-Learning Methods: Model Development and Performance Evaluation

Predicting coronary heart disease in Chinese diabetics using machine learning

Heart disease prediction based on machine learning algorithms

A Comprehensive Analysis on Risk Prediction of Heart Disease using Machine Learning Models

Comparative analysis of supervised learning algorithms for prediction of cardiovascular diseases

Coronary Heart Disease Prediction On Small Datasets: A Comparative Analysis

Early prediction model for coronary heart disease using genetic algorithms, hyper-parameter optimization and machine learning techniques

An Efficient Heart Disease Prediction System Using Machine Learning

Heart Disease Prediction Based on MWMOTE and Res-BiGRU Models

Machine Learning-Based Classification Algorithms for the Prediction of Coronary Heart Diseases

Effectively Predicting the Presence of Coronary Heart Disease Using Machine Learning Classifiers

Prediction of cardiovascular and cerebrovascular diseases based on machine learning models

An accurate diagnosis of coronary heart disease by Catboost, with easily accessible data

Machine Learning Models for Cardiovascular Disease Prediction: A Comparative Study

[Study on Application of SVM in Prediction of Coronary Heart Disease].

Implementation of a Heart Disease Risk Prediction Model Using Machine Learning

Heart Disease Prediction using Machine Learning Techniques