Atrial fibrillation (AF) is a common supraventricular arrhythmia. Its automatic identification by standard 12-lead electrocardiography (ECG) is still challenging. Recently, deep learning provided new instruments able to mimic the diagnostic ability of clinicians but only in case of binary classification (AF vs. normal sinus rhythm-NSR). However, binary classification is far from the real scenarios, where AF has to be discriminated also from several other physiological and pathological conditions. The aim of this work is to present a new AF multiclass classifier based on a convolutional neural network (CNN), able to discriminate AF from NSR, premature atrial contraction (PAC) and premature ventricular contraction (PVC). Overall, 2796 12-lead ECG recordings were selected from the open-source "PhysioNet/Computing in Cardiology Challenge 2021" database, to construct a dataset constituted by four balanced classes, namely AF class, PAC class, PVC class, and NSR class. Each lead of each ECG recording was decomposed into spectrogram by continuous wavelet transform and saved as 2D grayscale images, used to feed a 6-layers CNN. Considering the same CNN architecture, a multiclass classifiers (all classes) and three binary classifiers (AF class, PAC class, and PVC class vs. NSR class) were created and validated by a stratified shuffle split cross-validation of 10 splits. Performance was quantified in terms of area under the curve (AUC) of the receiver operating characteristic. Multiclass classifier performance was high (AF class: 96.6%; PAC class: 95.3%; PVC class: 92.8%; NSR class: 97.4%) and preferable to binary classifiers. Thus, our CNN AF multiclass classifier proved to be an efficient tool for AF discrimination from physiological and pathological confounders. Clinical Relevance-Our CNN AF multiclass classifier proved to be suitable for AF discrimination in real scenarios.

Multiclass Convolutional Neural Networks for Atrial Fibrillation Classification / Sbrollini, Agnese; Tomassini, Selene; Emaldi, Enrico; Marcantoni, Ilaria; Morettini, Micaela; Dragoni, Aldo F; Burattini, Laura. - ELETTRONICO. - 2022:(2022), pp. 1288-1291. (Intervento presentato al convegno 44th Annual International Conference of the IEEE Engineering in Medicine & Biology Society (EMBC) tenutosi a Glasgow, Scotland, UK nel 11-15 Luglio 2022) [10.1109/EMBC48229.2022.9871124].

Multiclass Convolutional Neural Networks for Atrial Fibrillation Classification

Sbrollini, Agnese;Tomassini, Selene;Marcantoni, Ilaria;Morettini, Micaela;Dragoni, Aldo F;Burattini, Laura
2022-01-01

Abstract

Atrial fibrillation (AF) is a common supraventricular arrhythmia. Its automatic identification by standard 12-lead electrocardiography (ECG) is still challenging. Recently, deep learning provided new instruments able to mimic the diagnostic ability of clinicians but only in case of binary classification (AF vs. normal sinus rhythm-NSR). However, binary classification is far from the real scenarios, where AF has to be discriminated also from several other physiological and pathological conditions. The aim of this work is to present a new AF multiclass classifier based on a convolutional neural network (CNN), able to discriminate AF from NSR, premature atrial contraction (PAC) and premature ventricular contraction (PVC). Overall, 2796 12-lead ECG recordings were selected from the open-source "PhysioNet/Computing in Cardiology Challenge 2021" database, to construct a dataset constituted by four balanced classes, namely AF class, PAC class, PVC class, and NSR class. Each lead of each ECG recording was decomposed into spectrogram by continuous wavelet transform and saved as 2D grayscale images, used to feed a 6-layers CNN. Considering the same CNN architecture, a multiclass classifiers (all classes) and three binary classifiers (AF class, PAC class, and PVC class vs. NSR class) were created and validated by a stratified shuffle split cross-validation of 10 splits. Performance was quantified in terms of area under the curve (AUC) of the receiver operating characteristic. Multiclass classifier performance was high (AF class: 96.6%; PAC class: 95.3%; PVC class: 92.8%; NSR class: 97.4%) and preferable to binary classifiers. Thus, our CNN AF multiclass classifier proved to be an efficient tool for AF discrimination from physiological and pathological confounders. Clinical Relevance-Our CNN AF multiclass classifier proved to be suitable for AF discrimination in real scenarios.
2022
978-1-7281-2782-8
File in questo prodotto:
Non ci sono file associati a questo prodotto.

I documenti in IRIS sono protetti da copyright e tutti i diritti sono riservati, salvo diversa indicazione.

Utilizza questo identificativo per citare o creare un link a questo documento: https://hdl.handle.net/11566/306481
 Attenzione

Attenzione! I dati visualizzati non sono stati sottoposti a validazione da parte dell'ateneo

Citazioni
  • ???jsp.display-item.citation.pmc??? ND
  • Scopus 0
  • ???jsp.display-item.citation.isi??? ND
social impact