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

被引:20
作者
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 条
[21]   Classification with machine learning algorithms after hybrid feature selection in imbalanced data sets [J].
Pulat, Meryem ;
Kocakoc, Ipek Deveci .
OPERATIONS RESEARCH AND DECISIONS, 2024, 34 (04) :157-183
[22]   Feature Selection and Comparison of Machine Learning Algorithms in Classification of Grazing and Rumination Behaviour in Sheep [J].
Mansbridge, Nicola ;
Mitsch, Jurgen ;
Bollard, Nicola ;
Ellis, Keith ;
Miguel-Pacheco, Giuliana G. ;
Dottorini, Tania ;
Kaler, Jasmeet .
SENSORS, 2018, 18 (10)
[23]   Classification of Heart Failure Using Machine Learning: A Comparative Study [J].
Chulde-Fernandez, Bryan ;
Enriquez-Ortega, Denisse ;
Guevara, Cesar ;
Navas, Paulo ;
Tirado-Espin, Andres ;
Vizcaino-Imacana, Paulina ;
Villalba-Meneses, Fernando ;
Cadena-Morejon, Carolina ;
Almeida-Galarraga, Diego ;
Acosta-Vargas, Patricia .
LIFE-BASEL, 2025, 15 (03)
[24]   Gearbox faults feature selection and severity classification using machine learning [J].
Zuber, Ninoslav ;
Bajric, Rusmir .
EKSPLOATACJA I NIEZAWODNOSC-MAINTENANCE AND RELIABILITY, 2020, 22 (04) :748-756
[25]   Data-Driven Diabetes Risk Factor Prediction Using Machine Learning Algorithms with Feature Selection Technique [J].
Kakoly, Israt Jahan ;
Hoque, Md. Rakibul ;
Hasan, Najmul .
SUSTAINABILITY, 2023, 15 (06)
[26]   A Comparative Flood Susceptibility Assessment in a Norwegian Coastal City Using Feature Selection Methods and Machine Learning Algorithms [J].
Lam Van Nguyen ;
Dieu Tien Bui ;
Seidu, Razak .
ADVANCES IN RESEARCH ON WATER RESOURCES AND ENVIRONMENTAL SYSTEMS, 2023, :591-618
[27]   Using Machine Learning to Score Multi-Dimensional Assessments of Chemistry and Physics [J].
Sarah Maestrales ;
Xiaoming Zhai ;
Israel Touitou ;
Quinton Baker ;
Barbara Schneider ;
Joseph Krajcik .
Journal of Science Education and Technology, 2021, 30 :239-254
[28]   Using Machine Learning to Score Multi-Dimensional Assessments of Chemistry and Physics [J].
Maestrales, Sarah ;
Zhai, Xiaoming ;
Touitou, Israel ;
Baker, Quinton ;
Schneider, Barbara ;
Krajcik, Joseph .
JOURNAL OF SCIENCE EDUCATION AND TECHNOLOGY, 2021, 30 (02) :239-254
[29]   Comparative Study of Optimum Medical Diagnosis of Human Heart Disease Using Machine Learning Technique With and Without Sequential Feature Selection [J].
Ahmad, Ghulab Nabi ;
Shafiullah ;
Algethami, Abdullah ;
Fatima, Hira ;
Akhter, Syed Md Humayun .
IEEE ACCESS, 2022, 10 :23808-23828
[30]   Attack classification using feature selection techniques: a comparative study [J].
Thakkar, Ankit ;
Lohiya, Ritika .
JOURNAL OF AMBIENT INTELLIGENCE AND HUMANIZED COMPUTING, 2021, 12 (01) :1249-1266