Coronary heart disease prediction based on hybrid deep learning

Abstract

Machine learning provides increasingly reliable assistance for medical experts in diagnosing coronary heart disease. This study proposes a deep learning hybrid model based coronary heart disease (CAD) prediction method, which can significantly improve the prediction accuracy compared to traditional solutions. This research scheme is based on the data of 7291 patients and proposes a hybrid model, which uses two different deep neural network models and a recurrent neural network model as the main model for training. The prediction results based on the main model training use a k-nearest neighbor model for secondary training so as to improve the accuracy of coronary heart disease prediction. The comparison between the model prediction results and the clinical diagnostic results shows that the prediction model has a prediction accuracy rate of 82.8%, a prediction precision rate of 87.08%, a prediction recall rate of 88.57%, a prediction F1-score of 87.82%, and an area under the curve value of 0.8 in the test set. Compared to single model machine learning predictions, the hybrid model has a significantly improved accuracy and has effectively solved the problem of overfitting. A deep learning based CAD prediction hybrid model that combines multiple weak models into a strong model can fully explore the complex inter-relationships between various features under limited feature values and sample size, improve the evaluation indicators of the prediction model, and provide effective auxiliary support for CAD diagnosis.