Multiclass Convolutional Neural Networks for Atrial Fibrillation Classification

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.