过度拟合
计算机科学
人工智能
机器学习
人工神经网络
预测建模
深度学习
试验装置
计算机辅助设计
交叉验证
特征(语言学)
数据挖掘
语言学
工程类
哲学
工程制图
作者
Feng Li,Yi Chen,Hongzeng Xu
摘要
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.
科研通智能强力驱动
Strongly Powered by AbleSci AI