A Hybrid AI Model for Early Detection of Heart Disease Using Clinical and ECG Data
محتوى المقالة الرئيسي
الملخص
One of the main causes of morbidity and death is still cardiovascular diseases (CVDs) thereby stressing the need for developing effective early diagnostic tools. This paper proposes a hybrid approach using AI that combines deep ECG feature learning with clinical data to enhance early heart disease diagnosis. The ECG signals were preprocessed, divided into 2-second epochs, and then represented as 128-dimensional embedding using a 1D CNN-based feature extractor. At the same time, a set of 13 clinical risk factors was normalized and used to construct a lightweight clinical prediction branch. Both were then fused using a multimodal neural network that learns both ECG and clinical features. The findings of the experiment have demonstrated the effectiveness of the proposed hybrid framework, which has achieved 97.56% accuracy, 0.996 AUC, 99.0% sensitivity, and 96.0% specificity, significantly outperforming the individual models. Furthermore, the SHAP values-based explain ability method has assisted in revealing the most important clinical variables and ECG embedding features, thereby making the prediction results interpretable and meaningful from the clinical perspective. In summary, the proposed approach has shown how multimodal AI systems might help with early heart disease diagnosis and offers a promising solution for future real-time applications.
تفاصيل المقالة
القسم

هذا العمل مرخص بموجب Creative Commons Attribution 4.0 International License.
كيفية الاقتباس
المراجع
[1] Alhabib KF, Batais MA, Almigbal TH, Alshamiri MQ, Altaradi H, Rangarajan S, Yusuf S: Demographic, behavioral, and cardiovascular disease risk factors in the Saudi population: results from the Prospective Urban Rural Epidemiology study (PURE-Saudi). BMC Public Health. 2020, 20:1213.
[2] Schneeberger D, Stoeger K, Holzinger A: The European legal framework for medical AI. Machine Learning and Knowledge Extraction. Holzinger A, Kieseberg P, Tjoa A, Weippl E (ed): Springer, 2020. 209-26.
[3] Bhatt, C.M.; Patel, P.; Ghetia, T.; Mazzeo, P.L. Effective heart disease prediction using machine learning techniques. Algorithms 2023, 16, 88.
[4] Momin, M.A.; Bhagwat, N.S.; Dhiwar, A.V.; Chavhate, S.B.; Devekar, N.S. Smart body monitoring system using IoT and machine learning. engrXiv 2021.
[5] Yahyaie, M.; Tarokh, M.J.; Mahmoodyar, M.A. Use of internet of things to provide a new model for remote heart attack prediction. Telemed. J. e-Health 2019, 25, 499–510.
[6] Abba, S.; Garba, A.M. An IoT-based smart framework for a human heartbeat rate monitoring and control system. Proceedings 2020, 42, 36.
[7] Gupta, S.K.; Shrivastava, A.; Upadhyay, S.P.; Chaurasia, P.K. A machine learning approach for heart attack prediction. Int. J. Eng. Adv. Technol. 2021, 10, 124–134.
[8] Taylor, O.E.; Ezekiel, P.S.; Deedam-Okuchaba, F.B. A model to detect heart disease using machine learning algorithm. Int. J. Comput. Sci. Eng. 2019, 7, 1–5.
[9] Khan, M.A. An IoT framework for heart disease prediction based on MDCNN classifier. IEEE Access 2020, 8, 34717–34727.
[10] Lakshmanarao, A.; Swathi, Y.; Sundareswar, P.S. Machine learning techniques for heart disease prediction. Forest 2019, 95, 97.
[11] Ghosh, P.; Azam, S.; Jonkman, M.; Karim, A.; Shamrat, F.J.; Ignatious, E.; Shultana, S.; Beeravolu, A.R.; De Boer, F. Efficient prediction of cardiovascular disease using machine learning algorithms with relief and LASSO feature selection techniques. IEEE Access 2021, 9, 19304–19326.
[12] Benjamin EJ, Virani SS, Callaway CW, et al.: Heart disease and stroke statistics-2018 update: a report from the American Heart Association. Circulation. 2018, 137:e67-e492.
[13] Rao, G.M.; Ramesh, D.; Sharma, V.; Sinha, A.; Hassan, M.M.; Gandomi, A.H. AttGRU-HMSI: Enhancing heart disease diagnosis using hybrid deep learning approach. Sci. Rep. 2024, 14, 7833.
[14] Islam, M.N.; Raiyan, K.R.; Mitra, S.; Mannan, M.R.; Tasnim, T.; Putul, A.O.; Mandol, A.B. Predictis: An IoT and machine learn-ing-based system to predict risk level of cardio-vascular diseases. BMC Health Serv. Res. 2023, 23, 171.
[15] Sharma, V.; Yadav, S.; Gupta, M. Heart disease prediction using machine learning techniques. In Proceedings of the 2020 2nd International Conference on Advances in Computing, Communication Control and Networking (ICACCCN), Greater Noida, India, 18–19 December 2020; pp. 177–181.
[16] Available: https://physionet.org/content/mitdb/1.0.0
[17] Available: https://www.kaggle.com/datasets/johnsmith88/heart-disease-dataset