A comparative study of heterogeneous machine learning algorithms for arrhythmia classification using feature selection technique and multi-dimensional datasets

被引:6
作者
Sharma, Abhinav [1 ]
Dhanka, Sanjay [1 ]
Kumar, Ankur [1 ]
Maini, Surita [1 ]
机构
[1] St Longowal Inst Engn & Technol, Dept Elect & Instrumentat Engn, Sangrur, Punjab, India
来源
ENGINEERING RESEARCH EXPRESS | 2024年 / 6卷 / 03期
关键词
artificial intelligence; logistic regression (LG); extreme gradient boosting (XGB); arrhythmia; random forest (RF); machine learning; DISEASE;
D O I
10.1088/2631-8695/ad5d51
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
Arrhythmia, a common cardiovascular disorder, refers to the abnormal electrical activity within the heart, leading to irregular heart rhythms. This condition affects millions of people worldwide, with severe implications on cardiac function and overall health. Arrhythmias can strike anyone at any age which is a significant cause of morbidity and mortality on a global scale. About 80% of deaths related to heart disease are caused by ventricular arrhythmias. This research investigated the application of an optimized multi-objectives supervised Machine Learning (ML) models for early arrhythmia diagnosis. The authors evaluated the model's performance on the arrhythmia dataset from the UCI ML repository with varying train-test splits (70:30, 80:20, and 90:10). Standard preprocessing techniques such as handling missing values, formatting, balancing, and directory analysis were applied along with Pearson correlation for feature selection, all aimed at enhancing model performance. The proposed optimized RF model achieved impressive performance metrics, including accuracy (95.24%), precision (100%), sensitivity (89.47%), and specificity (100%). Furthermore, the study compared the proposed approach to existing models, demonstrating significant improvements across various performance measures.
引用
收藏
页数:21
相关论文
共 50 条
[31]   Automatic Cataract Classification based on Ultrasound Technique using Machine Learning: A comparative Study [J].
Caixinha, Miguel ;
Velte, Elena ;
Santos, Mario ;
Perdigao, Fernando ;
Amaro, Joao ;
Gomes, Marco ;
Santos, Jaime .
PROCEEDINGS OF THE 2015 ICU INTERNATIONAL CONGRESS ON ULTRASONICS, 2015, 70 :1221-1224
[32]   An Approach to Feature Selection in Intrusion Detection Systems Using Machine Learning Algorithms [J].
Kavitha, G. ;
Elango, N. M. .
INTERNATIONAL JOURNAL OF E-COLLABORATION, 2020, 16 (04) :48-58
[33]   A Study on Facial Expression Change Detection Using Machine Learning Methods with Feature Selection Technique [J].
Sung, Sang-Ha ;
Kim, Sangjin ;
Park, Byung-Kwon ;
Kang, Do-Young ;
Sul, Sunhae ;
Jeong, Jaehyun ;
Kim, Sung-Phil .
MATHEMATICS, 2021, 9 (17)
[34]   A Comparative Study for the Selection of Machine Learning Algorithms based on Descriptive Parameters [J].
Kumar, Chettan ;
Kaeppel, Martin ;
Schuetzenmeier, Nicolai ;
Eisenhuth, Philipp ;
Jablonski, Stefan .
PROCEEDINGS OF THE 8TH INTERNATIONAL CONFERENCE ON DATA SCIENCE, TECHNOLOGY AND APPLICATIONS (DATA), 2019, :408-415
[35]   Comparative Study of Several Machine Learning Algorithms for Classification of Unifloral Honeys [J].
Mateo, Fernando ;
Tarazona, Andrea ;
Maria Mateo, Eva .
FOODS, 2021, 10 (07)
[36]   Improving the Classification of Airplane Accidents Severity using Feature Selection, Extraction and Machine Learning Models [J].
Kaidi, Rachid ;
AL Achhab, Mohammed ;
Lazaar, Mohamed ;
Omara, Hicham .
INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2023, 14 (12) :975-981
[37]   Feature selection with Fast Correlation-Based Filter for Breast cancer prediction and Classification using Machine Learning Algorithms [J].
Khourdifi, Youness ;
Bahaj, Mohamed .
2018 INTERNATIONAL SYMPOSIUM ON ADVANCED ELECTRICAL AND COMMUNICATION TECHNOLOGIES (ISAECT), 2018,
[38]   Machine learning with automatic feature selection for multi-class protein fold classification [J].
Huang, CD ;
Liang, SF ;
Lin, CT ;
Wu, RC .
JOURNAL OF INFORMATION SCIENCE AND ENGINEERING, 2005, 21 (04) :711-720
[39]   A machine learning approach for feature selection traffic classification using security analysis [J].
Shafiq, Muhammad ;
Yu, Xiangzhan ;
Bashir, Ali Kashif ;
Chaudhry, Hassan Nazeer ;
Wang, Dawei .
JOURNAL OF SUPERCOMPUTING, 2018, 74 (10) :4867-4892
[40]   A machine learning approach for feature selection traffic classification using security analysis [J].
Muhammad Shafiq ;
Xiangzhan Yu ;
Ali Kashif Bashir ;
Hassan Nazeer Chaudhry ;
Dawei Wang .
The Journal of Supercomputing, 2018, 74 :4867-4892