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] A modified grey wolf optimization based feature selection method from EEG for silent speech classification
    Ghosh, Rajdeep
    Sinha, Nidul
    Biswas, Saroj Kumar
    Phadikar, Souvik
    [J]. JOURNAL OF INFORMATION & OPTIMIZATION SCIENCES, 2019, 40 (08) : 1639 - 1652
  • [22] Stochastic optimization using grey wolf optimization with optimal computing budget allocation
    Fu, Yaping
    Xiao, Hui
    Lee, Loo Hay
    Huang, Min
    [J]. APPLIED SOFT COMPUTING, 2021, 103
  • [23] Design and Development of an Efficient Explainable AI Framework for Heart Disease Prediction
    Tenepalli, Deepika
    Navamani, T. M.
    [J]. INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2024, 15 (06) : 1494 - 1503
  • [24] Precise Feature Selection in Predictive Genetic Models using Grey Wolf Optimization Algorithm
    Abbas, Mohamed
    [J]. PROCEEDINGS OF 2018 13TH INTERNATIONAL CONFERENCE ON COMPUTER ENGINEERING AND SYSTEMS (ICCES), 2018, : 63 - 68
  • [25] Automated face retrieval using bag-of-features and sigmoidal grey wolf optimization
    Shukla, Arun Kumar
    Kanungo, Suvendu
    [J]. EVOLUTIONARY INTELLIGENCE, 2021, 14 (03) : 1201 - 1212
  • [26] Interpretable Retinal Disease Classification from OCT Images Using Deep Neural Network and Explainable AI
    Reza, Md Tanzim
    Ahmed, Farzad
    Sharar, Shihab
    Rasel, Annajiat Alim
    [J]. PROCEEDINGS OF INTERNATIONAL CONFERENCE ON ELECTRONICS, COMMUNICATIONS AND INFORMATION TECHNOLOGY 2021 (ICECIT 2021), 2021,
  • [27] Grey wolf optimization algorithm based on adaptive normal cloud model
    Zhang Z.
    Rao S.-H.
    Zhang S.-J.
    [J]. Kongzhi yu Juece/Control and Decision, 2021, 36 (10): : 2562 - 2568
  • [28] Early-stage diagnosis of chronic kidney disease using majority vote - Grey Wolf optimization (MV-GWO)
    Siddhartha, Manu
    Kumar, Vaibhav
    Nath, Rajendra
    [J]. HEALTH AND TECHNOLOGY, 2022, 12 (01) : 117 - 136
  • [29] Early-stage diagnosis of chronic kidney disease using majority vote – Grey Wolf optimization (MV-GWO)
    Manu Siddhartha
    Vaibhav Kumar
    Rajendra Nath
    [J]. Health and Technology, 2022, 12 : 117 - 136
  • [30] Research on Mixed-Strategy Based Grey Wolf Optimization for Gene Expression Classification
    Wang, Yang
    [J]. IEEE ACCESS, 2023, 11 : 103254 - 103273