Tehran University of Medical Sciences

Science Communicator Platform

Stay connected! Follow us on X network (Twitter):
Share By
Comparison of Neural Network Training Algorithms for Classification of Heart Diseases Publisher



Karim H1 ; Niakan SR1 ; Safdari R1
Authors

Source: IAES International Journal of Artificial Intelligence Published:2018


Abstract

Heart disease is the first cause of death in different countries. Artificial neural network (ANN) technique can be used to predict or classification patients getting a heart disease. There are different training algorithms for ANN. We compared eight neural network training algorithms for classification of heart disease data from UCI repository containing 303 samples. Performance measures of each algorithm containing the speed of training, the number of epochs, accuracy, and mean square error (MSE) were obtained and analyzed. Our results showed that training time for gradient descent algorithms was longer than other training algorithms (8-10 seconds). In contrast, Quasi-Newton algorithms were faster than others (<=0 second). MSE for all algorithms was between 0.117 and 0.228. While there was a significant association between training algorithms and training time (p<0.05), the number of neurons in hidden layer had not any significant effect on the MSE and/or accuracy of the models (p>0.05). Based on our findings, for development an ANN classification model for heart diseases, it is best to use Quasi-Newton training algorithms because of the best speed and accuracy. © 2018 Institute of Advanced Engineering and Science. All rights reserved.
Related Docs
View other Related Docs
1. A Clustering-Based Differential Evolution Boosted by a Regularisation-Based Objective Function and a Local Refinement for Neural Network Training, 2022 IEEE Congress on Evolutionary Computation# CEC 2022 - Conference Proceedings (2022)
2. Developing an Intelligent System for Prediction of Optimal Dose of Warfarin in Iranian Adult Patients With Artificial Heart Valve, Frontiers in Health Informatics (2019)
3. Pain Estimation After Coronary Angiography Based on Vital Signs by Using Artificial Neural Networks, Anaesthesia# Pain and Intensive Care (2021)
Experts (# of related papers)
View all Related Experts
Reza Safdari (3)
Alireza Atashi (2)
Other Related Docs
4. Comparing Performance of Data Mining Algorithms in Prediction Heart Diseses, International Journal of Electrical and Computer Engineering (2015)
5. Predicting the Risk of Mortality and Rehospitalization in Heart Failure Patients: A Retrospective Cohort Study by Machine Learning Approach, Clinical Cardiology (2024)
6. Diagnosis of Breast Cancer Using Decision Tree, Artificial Neural Network and Naive Bayes to Provide a Native Model for Fars Province, Journal of Payavard Salamat (2019)
7. Optimizing Cardiovascular Disease Mortality Prediction: A Super Learner Approach in the Tehran Lipid and Glucose Study, BMC Medical Informatics and Decision Making (2024)
8. Improved Diagnostic Accuracy for Myocardial Perfusion Imaging Using Artificial Neural Networks on Different Input Variables Including Clinical and Quantification Data; [Precision Diagnostica Mejorada Para La Imagen De Perfusion Miocardica Usando Redes Neuronales Artificiales En Diferentes Variables De Entrada Incluyendo Datos Clinicos Y De Cuantificacion], Revista Espanola de Medicina Nuclear e Imagen Molecular (2019)
9. Presenting a Two-Stage Hybrid Model for Allocating Advanced Ventilators Using Machine Learning Methods: A Case Study, IEEE Access (2024)
10. Estimation of Myocardial Infarction Death in Iran: Artificial Neural Network, BMC Cardiovascular Disorders (2022)
11. Developing a Clinical Decision Support System for Prediction Postoperative Coronary Artery Bypass Grafting Infection in Diabetic Patients, Journal of Biomedical Physics and Engineering (2022)
12. Risk Predictions of Hospital-Acquired Pressure Injury in the Intensive Care Unit Based on a Machine Learning Algorithm, International Wound Journal (2023)
13. Comparing the Performance of Machine Learning Techniques for Low Back Pain Diagnosis, International Journal of Medical Engineering and Informatics (2023)
14. Applying Data Mining Approaches for Chronic Kidney Disease Diagnosis, International Journal of Intelligent Systems and Applications in Engineering (2021)
15. Development of a Decision Support System Tool to Predict the Pulmonary Function Using Artificial Neural Network Approach, Concurrency and Computation: Practice and Experience (2021)
16. A Systematic Literature Review and Classification of Knowledge Discovery in Traditional Medicine, Computer Methods and Programs in Biomedicine (2019)
17. Evaluation of Classification Algorithms Vs Knowledge-Based Methods for Differential Diagnosis of Asthma in Iranian Patients, International Journal of Information Systems in the Service Sector (2018)
18. A Covid-19 Forecasting System for Hospital Needs Using Anfis and Lstm Models: A Graphical User Interface Unit, Digital Health (2022)
19. Predicting the Need for Cardiovascular Surgery: A Comparative Study of Machine Learning Models, Journal of Electronics# Electromedical Engineering# and Medical Informatics (2024)
20. Machine Learning-Based Prediction of 1-Year Mortality in Hypertensive Patients Undergoing Coronary Revascularization Surgery, Clinical Cardiology (2023)