A Metaheuristic Optimization Approach for Parameter Estimation in Arrhythmia Classification from Unbalanced Data

Sensors

Ignacio Algredo-Badillo¹  JuanCarlos Carrillo-Alarcón²  Antonio Muñoz³  LuisAlberto Morales-Rosales⁴  Mariana Lobato-Báez⁵  Héctor Rodríguez-Rángel⁶ 

[1] Department of Computer Science, Conacyt-Instituto Nacional de Astrofísica, Óptica y Electrónica (INAOE) , Tonantzintla, Puebla 72840, Mexico;Department of Computer Science, Instituto Nacional de Astrofísica, Óptica y Electrónica (INAOE) , Tonantzintla, Puebla 72840, Mexico;Engineering Department, University of Guadalajara, Av. Independencia Nacional 151, Autlán, Jalisco 48900, Mexico;Faculty of Civil Engineering, Conacyt-Universidad Michoacana de San Nicolás de Hidalgo, Morelia 58030, Michoacán, Mexico;Higher Technological Institute of Libres, Libres, Puebla 73780, Mexico;Technological Institute of Culiacan, Culiacan, Sinaloa 80220, Mexico;

关键词: electrocardiogram (ECG);    signal processing;    machine learning;    arrhythmia;    unbalanced;   

DOI  :  10.3390/s20113139

来源: DOAJ

【 摘 要 】

The electrocardiogram records the heart’s electrical activity and generates a significant amount of data. The analysis of these data helps us to detect diseases and disorders via heart bio-signal abnormality classification. In unbalanced-data contexts, where the classes are not equally represented, the optimization and configuration of the classification models are highly complex, reflecting on the use of computational resources. Moreover, the performance of electrocardiogram classification depends on the approach and parameter estimation to generate the model with high accuracy, sensitivity, and precision. Previous works have proposed hybrid approaches and only a few implemented parameter optimization. Instead, they generally applied an empirical tuning of parameters at a data level or an algorithm level. Hence, a scheme, including metrics of sensitivity in a higher precision and accuracy scale, deserves special attention. In this article, a metaheuristic optimization approach for parameter estimations in arrhythmia classification from unbalanced data is presented. We selected an unbalanced subset of those databases to classify eight types of arrhythmia. It is important to highlight that we combined undersampling based on the clustering method (data level) and feature selection method (algorithmic level) to tackle the unbalanced class problem. To explore parameter estimation and improve the classification for our model, we compared two metaheuristic approaches based on differential evolution and particle swarm optimization. The final results showed an accuracy of 99.95%, a F1 score of 99.88%, a sensitivity of 99.87%, a precision of 99.89%, and a specificity of 99.99%, which are high, even in the presence of unbalanced data.

【 授权许可】

Unknown