Combining Multiple Factors of LightGBM and XGBoost Algorithms to Predict the Morbidity of Double-High Disease

Abstract

In recent years, cardiovascular and cerebrovascular diseases have seriously jeopardized people’s health. Double-high (hyperlipidemia and hypertension) is one of the main causes of cardiovascular and cerebrovascular diseases in clinical practice. To diagnose cardiovascular and cerebrovascular diseases early, a reliable prediction system should be developed to assist doctors. In this paper, the different elements and evaluate the weight of these elements on double-high diseases are analyzed by machine learning method. The LightGBM algorithm and XGBoost algorithm were employed to construct the prediction models, respectively. Significantly, the proposed model was trained by real physical examination data and five meaningful and useful biochemical indicators were selected to encoding the raw physical examination data to numerical vector. The selected features are systolic blood pressure, diastolic blood pressure, serum triglyceride, serum high-density lipoprotein and serum low-density lipoprotein. The mean square error (MSE) after calculating the logarithm of the predicted value and the true value was introduced to assess the prediction model. Results show that this model can effectively predict cardiovascular and cerebrovascular diseases in advance.