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 条
  • [41] Building energy optimization using Grey Wolf Optimizer (GWO)
    Ghalambaz, Mehdi
    Yengejeh, Reza Jalilzadeh
    Davami, Amir Hossein
    CASE STUDIES IN THERMAL ENGINEERING, 2021, 27 (27)
  • [42] Allocation of Distributed Generators Using Parallel Grey Wolf Optimization
    Younesi, Soheil
    Ahmadi, Bahman
    Ceylan, Oguzhan
    Ozdemir, Aydogan
    PROCEEDINGS OF 9TH INTERNATIONAL CONFERENCE ON MODERN POWER SYSTEMS (MPS 2021), 2021,
  • [43] Design and Development of an Efficient Explainable AI Framework for Heart Disease Prediction
    Tenepalli, Deepika
    Navamani, T. M.
    INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2024, 15 (06) : 1494 - 1503
  • [44] Structural Damage Assessment via Model Updating Using Augmented Grey Wolf Optimization Algorithm
    Ghasemi, S.
    Amiri, G. Ghodrati
    Dehcheshmeh, M. Mohamadi
    INTERNATIONAL JOURNAL OF ENGINEERING, 2020, 33 (07): : 1173 - 1182
  • [45] Stochastic optimization using grey wolf optimization with optimal computing budget allocation
    Fu, Yaping
    Xiao, Hui
    Lee, Loo Hay
    Huang, Min
    APPLIED SOFT COMPUTING, 2021, 103
  • [46] Interpretable Retinal Disease Classification from OCT Images Using Deep Neural Network and Explainable AI
    Reza, Md Tanzim
    Ahmed, Farzad
    Sharar, Shihab
    Rasel, Annajiat Alim
    PROCEEDINGS OF INTERNATIONAL CONFERENCE ON ELECTRONICS, COMMUNICATIONS AND INFORMATION TECHNOLOGY 2021 (ICECIT 2021), 2021,
  • [47] Brain Tumor Detection and Classification Using Deep Learning and Sine-Cosine Fitness Grey Wolf Optimization
    ZainEldin, Hanaa
    Gamel, Samah A.
    El-Kenawy, El-Sayed M.
    Alharbi, Amal H.
    Khafaga, Doaa Sami
    Ibrahim, Abdelhameed
    Talaat, Fatma M.
    BIOENGINEERING-BASEL, 2023, 10 (01):
  • [48] Explainable AI based efficient ensemble model for breast cancer classification using optical coherence tomography
    Dhiman, Babita
    Kamboj, Sangeeta
    Srivastava, Vishal
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2024, 91
  • [49] Grey wolf optimization algorithm based on adaptive normal cloud model
    Zhang Z.
    Rao S.-H.
    Zhang S.-J.
    Kongzhi yu Juece/Control and Decision, 2021, 36 (10): : 2562 - 2568
  • [50] A Robust Feature Construction for Fish Classification Using Grey Wolf Optimizer
    Santosa, Paulus Insap
    Pramunendar, Ricardus Anggi
    CYBERNETICS AND INFORMATION TECHNOLOGIES, 2022, 22 (04) : 152 - 166