AN EXPLAINABLE AI MODEL IN HEART DISEASE CLASSIFICATION USING GREY WOLF OPTIMIZATION

被引:0
作者
Varun G. [1 ]
Jagadeeshwaran J. [1 ]
Nithish K. [1 ]
Sanjey D.S.A. [1 ]
Venkatesh V. [1 ]
Ashokkumar P. [2 ]
机构
[1] Department of Cybersecurity and Internet of Things, Sri Ramachandra Faculty of Engineering and Technology, Sri Ramachandra Institute of Higher Education and Research
[2] Department of Artificial Intelligence and Machine Learning, Sri Ramachandra Faculty of Engineering and Technology, Sri Ramachandra Institute of Higher Education and Research
来源
Scalable Computing | 2024年 / 25卷 / 04期
关键词
Explainable AI; feature selection; Grey wolf optimization; Heart Disease classification;
D O I
10.12694/scpe.v25i4.2858
中图分类号
学科分类号
摘要
Heart disease is one of the world’s leading causes of death. It is estimated that around one-third of all deaths are caused by heart disease in the entire world. Recently many research works have focused on using machine learning models to detect and warn patients about the occurrence of heart disease at the early stage. However, machine learning models provide promising results, and the performance of the classification is affected by various reasons which include imbalanced training, and missing values. There are three main contributions of this research work. Firstly, missing values are addressed by employing a grouping of instances. Secondly, a dual filter based feature selection is introduced to pick the most effective features and lastly, we make of Grey Wolf Optimization for optimizing the hyperparameters of the machine learning models. Together, these contributions aim to improve the robustness and efficiency of machine learning applications by addressing missing data, optimizing feature selection, and fine-tuning model parameters. The accuracy of 98.41% indicates the superiority of the proposed classification which is more than 17.15% than the existing machine learning models. On the other hand, we use Explainable AI (XAI) methods to make our proposed model interpretable. © 2024 SCPE.
引用
收藏
页码:3139 / 3151
页数:12
相关论文
共 50 条
[31]   Hybrid Paddy disease classification using optimized statistical feature based transformation technique with explainable AI [J].
Amudha, M. ;
Brindha, K. .
ENVIRONMENTAL RESEARCH COMMUNICATIONS, 2025, 7 (07)
[32]   Research on Mixed-Strategy Based Grey Wolf Optimization for Gene Expression Classification [J].
Wang, Yang .
IEEE ACCESS, 2023, 11 :103254-103273
[33]   Integrating Deep Learning, Grey Wolf Optimization, and SVM for Precise Plant Seedling Classification [J].
Atchogou, Anselme ;
Tepe, Cengiz .
BRAZILIAN ARCHIVES OF BIOLOGY AND TECHNOLOGY, 2024, 67 :1-15
[34]   Enhancement of Power System Operation using Grey Wolf Optimization Algorithm [J].
Hassan, Zeinab G. ;
Ezzat, Mohamed ;
Abdelaziz, Almoataz Y. .
2017 NINETEENTH INTERNATIONAL MIDDLE-EAST POWER SYSTEMS CONFERENCE (MEPCON), 2017, :397-402
[35]   Size Optimization of Truss Structures Using Improved Grey Wolf Optimizer [J].
Alkhraisat, Habes ;
Dalbah, Lamees Mohammad ;
Al-Betar, Mohammed Azmi ;
Awadallah, Mohammed A. A. ;
Assaleh, Khaled ;
Deriche, Mohamed .
IEEE ACCESS, 2023, 11 :13383-13397
[36]   Generalized fractional optimization-based explainable lightweight CNN model for malaria disease classification [J].
Khan, Zeshan Aslam ;
Waqar, Muhammad ;
Raja, Muhammad Junaid Ali Asif ;
Chaudhary, Naveed Ishtiaq ;
Khan, Abeer Tahir Mehmood Anwar ;
Raja, Muhammad Asif Zahoor .
Computers in Biology and Medicine, 2025, 185
[37]   Using Grey Wolf Hunting Mechanism to Improve Brain Storm Optimization [J].
Wang, Shi ;
Cai, Zonghui ;
Yu, Yang ;
Lei, Zhenyu ;
Gao, Shangce .
2018 5TH INTERNATIONAL CONFERENCE ON INFORMATION SCIENCE AND CONTROL ENGINEERING (ICISCE 2018), 2018, :602-606
[38]   Calibration of Linear Muskingum Model Coefficients and Coefficient Parameters Using Grey Wolf and Particle Swarm Optimization [J].
Saplioglu, Kemal .
WATER RESOURCES MANAGEMENT, 2025, 39 (03) :999-1014
[39]   Accurate parameter estimation of Au/GaN/GaAs schottky diode model using grey wolf optimization [J].
Rabehi, A. ;
Douara, A. ;
Helal, H. ;
Younsi, A. Memdouh ;
Amrani, M. ;
Abbas, I. E. ;
Comini, E. ;
Benamara, Z. .
REVISTA MEXICANA DE FISICA, 2024, 70 (02) :1-8
[40]   Grey Wolf Optimization using Improved mutation oppositional based learning for optimization problems [J].
Saitou, Hayata ;
Haraguch, Harumi .
2022 IEEE 27TH INTERNATIONAL CONFERENCE ON EMERGING TECHNOLOGIES AND FACTORY AUTOMATION (ETFA), 2022,