An Approach Based on Mutually Informed Neural Networks to Optimize the Generalization Capabilities of Decision Support Systems Developed for Heart Failure Prediction - 23/09/21
pages | 8 |
Iconographies | 4 |
Vidéos | 0 |
Autres | 0 |
Graphical abstract |
Highlights |
• | The problem of over-fitting in Heart Failure (HF) detection is highlighted. |
• | A hybrid decision support system (DSS) is developed to reduce the over-fitting and improve generalization. |
• | The proposed DSS shows better generalization and lower over-fitting than conventional neural networks. |
• | Simulation results validate effectiveness of the developed DSS over recently published work. |
Abstract |
Available clinical methods for heart failure (HF) diagnosis are expensive and require a high-level of experts intervention. Recently, various machine learning models have been developed for the prediction of HF where most of them have an issue of over-fitting. Over-fitting occurs when machine learning based predictive models show better performance on the training data yet demonstrate a poor performance on the testing data and the other way around. Developing a machine learning model which is able to produce generalization capabilities (such that the model exhibits better performance on both the training and the testing data sets) could overall minimize the prediction errors. Hence, such prediction models could potentially be helpful to cardiologists for the effective diagnose of HF. This paper proposes a two-stage decision support system to overcome the over-fitting issue and to optimize the generalization factor. The first stage uses a mutual information based statistical model while the second stage uses a neural network. We applied our approach to the HF subset of publicly available Cleveland heart disease database. Our experimental results show that the proposed decision support system has optimized the generalization capabilities and has reduced the mean percent error (MPE) to 8.8% which is significantly less than the recently published studies. In addition, our model exhibits a 93.33% accuracy rate which is higher than twenty eight recently developed HF risk prediction models that achieved accuracy in the range of 57.85% to 92.31%. We can hope that our decision support system will be helpful to cardiologists if deployed in clinical setup.
Le texte complet de cet article est disponible en PDF.Keywords : Deep neural network, Generalization, Heart failure diagnosis, Mutual information, Shallow neural network
Plan
Vol 42 - N° 5
P. 345-352 - octobre 2021 Retour au numéroBienvenue sur EM-consulte, la référence des professionnels de santé.
L’accès au texte intégral de cet article nécessite un abonnement.
Bienvenue sur EM-consulte, la référence des professionnels de santé.
L’achat d’article à l’unité est indisponible à l’heure actuelle.
Déjà abonné à cette revue ?