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 条
  • [21] Estimation of Battery Parameters of the Equivalent Circuit Model using Grey Wolf Optimization
    Sangwan, Venu
    Kumar, Rajesh
    Rathore, A. K.
    2016 IEEE 6TH INTERNATIONAL CONFERENCE ON POWER SYSTEMS (ICPS), 2016,
  • [22] Forecasting Seasonal Changes in Ocean Acidification Using a Novel Grey Seasonal Model with Grey Wolf Optimization
    Yin, Kedong
    Zhang, Kai
    Yang, Wendong
    JOURNAL OF GREY SYSTEM, 2023, 35 (01): : 20 - 38
  • [23] Demystifying Deep Learning Models for Retinal OCT Disease Classification using Explainable AI
    Apon, Tasnim Sakib
    Hasan, Mohammad Mahmudul
    Islam, Abrar
    Alam, Md Golam Rabiul
    2021 IEEE ASIA-PACIFIC CONFERENCE ON COMPUTER SCIENCE AND DATA ENGINEERING (CSDE), 2021,
  • [25] RETRACTED ARTICLE: The health of things for classification of protein structure using improved grey wolf optimization
    Prerna Sharma
    Apoorva Gupta
    Aastha Aggarwal
    Deepak Gupta
    Ashish Khanna
    Aboul Ella Hassanien
    Victor Hugo C. de Albuquerque
    The Journal of Supercomputing, 2020, 76 : 1226 - 1241
  • [26] Epileptic Seizure Classification and Feature Optimization Technique Using Grey Wolf Algorithm on Dynamic Datasets
    Thanuja K.
    Shoba M.
    Patil K.
    SN Computer Science, 4 (3)
  • [27] Test case optimization using grey wolf algorithm
    Srishti Kumari
    Shweta Jindal
    Arun Sharma
    Software Quality Journal, 2025, 33 (2)
  • [28] Fuzzy C-Means-Grey Wolf Optimization for Classification of Stroke
    Setiawan, Qisthina Syifa
    Rustam, Zuherman
    Sa'id, Alva Andhika
    Maulidina, Faisa
    Sadewo, Wismaji
    Novkaniza, Fevi
    2021 INTERNATIONAL CONFERENCE ON DECISION AID SCIENCES AND APPLICATION (DASA), 2021,
  • [29] Explainable AI-driven model for gastrointestinal cancer classification
    Binzagr, Faisal
    FRONTIERS IN MEDICINE, 2024, 11
  • [30] Photovoltaic Parameter Estimation Using Grey Wolf Optimization
    Darmansyah
    Robandi, Imam
    2017 3RD INTERNATIONAL CONFERENCE ON CONTROL, AUTOMATION AND ROBOTICS (ICCAR), 2017, : 593 - 597