Cardiovascular disease detection from cardiac arrhythmia ECG signals using artificial intelligence models with hyperparameters tuning methodologies

被引:1
作者
Manivannan, Gowri Shankar [1 ]
Rajaguru, Harikumar [2 ]
Rajanna, S. [1 ]
V. Talawar, Satish [1 ]
机构
[1] Malnad Coll Engn, Hassan, Karnataka, India
[2] Bannari Amman Inst Technol, Sathyamangalam, Tamilnadu, India
关键词
Adam; Electrocardiogram; Grid search optimization; Hyperparameters; Cardio vascular diseases; DIMENSIONALITY REDUCTION; CLASSIFICATION; ALGORITHM; PCA; ICA;
D O I
10.1016/j.heliyon.2024.e36751
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
Cardiovascular disease (CVD) is connected with irregular cardiac electrical activity, which can be seen in ECG alterations. Due to its convenience and non-invasive aspect, the ECG is routinely exploited to identify different arrhythmias and automatic ECG recognition is needed immediately. In this paper, enhancement for the detection of CVDs such as Ventricular Tachycardia (VT), Premature Ventricular Contraction (PVC) and ST Change (ST) arrhythmia using different dimensionality reduction techniques and multiple classifiers are presented. Three-dimensionality reduction methods, such as Local Linear Embedding (LLE), Diffusion Maps (DM), and Laplacian Eigen (LE), are employed. The dimensionally reduced ECG samples are further feature selected with Cuckoo Search (CS) and Harmonic Search Optimization (HSO) algorithms. A publicly available MIT-BIH (Physionet) - VT database, PVC database, ST Change database and NSR database were used in this work. The cardiac vascular disturbances are classified by using seven classifiers such as Gaussian Mixture Model (GMM), Expectation Maximization (EM), Non-linear Regression (NLR), Logistic Regression (LR), Bayesian Linear Discriminant Analysis (BDLC), Detrended Fluctuation Analysis (Detrended FA), and Firefly. For different classes, the average overall accuracy of the classification techniques is 55.65 % when without CS and HSO feature selection, 64.36 % when CS feature selection is used, and 75.39 % when HSO feature selection is used. Also, to improve the performance of classifiers, the hyperparameters of four classifiers (GMM, EM, BDLC and Firefly) are tuned with the Adam and Grid Search Optimization (GSO) approaches. The average accuracy of classification for the CS feature-based classifiers that used GSO and Adam hyperparameter tuning was 79.92 % and 85.78 %, respectively. The average accuracy of classification for the HSO feature-based classifiers that used GSO and Adam hyperparameter tuning was 86.87 % and 93.77 %, respectively. The performance of the classifier is analyzed based on the accuracy parameter for both with and without feature selection methods and with hyperparameter tuning techniques. In the case of ST vs. NSR, a higher accuracy of 98.92 % is achieved for the LLE dimensionality reduction with HSO feature selection for the GMM classifier with Adam's hyperparameter tuning approach. The GMM classifier with the Adam hyperparameter tuning approach with 98.92 % accuracy in detecting ST vs. NSR cardiac disease is outperforming all other classifiers and methodologies.
引用
收藏
页数:35
相关论文
共 53 条
  • [1] BIOBEHAVIORAL VARIABLES AND MORTALITY OR CARDIAC-ARREST IN THE CARDIAC-ARRHYTHMIA PILOT-STUDY (CAPS)
    AHERN, DK
    GORKIN, L
    ANDERSON, JL
    TIERNEY, C
    HALLSTROM, A
    EWART, C
    CAPONE, RJ
    SCHRON, E
    KORNFELD, D
    HERD, JA
    RICHARDSON, DW
    FOLLICK, MJ
    [J]. AMERICAN JOURNAL OF CARDIOLOGY, 1990, 66 (01) : 59 - 62
  • [2] Efficient ECG classification based on Chi-square distance for arrhythmia detection
    Al-Shammary D.
    Noaman Kadhim M.
    Mahdi A.M.
    Ibaida A.
    Ahmed K.
    [J]. Journal of Electronic Science and Technology, 2024, 22 (02)
  • [3] A Fast Machine Learning Model for ECG-Based Heartbeat Classification and Arrhythmia Detection
    Alfaras, Miquel
    Soriano, Miguel C.
    Ortin, Silvia
    [J]. FRONTIERS IN PHYSICS, 2019, 7
  • [4] Reliability of old and new ventricular fibrillation detection algorithms for automated external defibrillators
    Amann, Anton
    Tratnig, Robert
    Unterkofler, Karl
    [J]. BIOMEDICAL ENGINEERING ONLINE, 2005, 4 (1)
  • [5] [Anonymous], 2008, J. Mach. Learn. Res.
  • [6] Classification of 12-lead ECGs Using Digital Biomarkers and Representation Learning
    Assaraf, David
    Levy, Jeremy
    Singh, Janmajay
    Chocron, Armand
    Behar, Joachim A.
    [J]. 2020 COMPUTING IN CARDIOLOGY, 2020,
  • [7] Laplacian eigenmaps for dimensionality reduction and data representation
    Belkin, M
    Niyogi, P
    [J]. NEURAL COMPUTATION, 2003, 15 (06) : 1373 - 1396
  • [8] ECG Signal Classification Using Various Machine Learning Techniques
    Celin, S.
    Vasanth, K.
    [J]. JOURNAL OF MEDICAL SYSTEMS, 2018, 42 (12)
  • [9] Detection and classification of microcalcification from digital mammograms with firefly algorithm, extreme learning machine and non-linear regression models: A comparison
    Chakravarthy, S. R. Sannasi
    Rajaguru, Harikumar
    [J]. INTERNATIONAL JOURNAL OF IMAGING SYSTEMS AND TECHNOLOGY, 2020, 30 (01) : 126 - 146
  • [10] The Matthews Correlation Coefficient (MCC) is More Informative Than Cohen's Kappa and Brier Score in Binary Classification Assessment
    Chicco, Davide
    Warrens, Matthijs J.
    Jurman, Giuseppe
    [J]. IEEE ACCESS, 2021, 9 : 78368 - 78381