Segmentation of Atrial Electrical Activity in Intracardiac Electrograms (IEGs) using Convolutional Neural Network (CNN) Trained on Small Imbalanced Dataset

Jakub Hejc1, David Pospisil2, Petra Novotna3, Martin Pesl4, Oto Janousek5, Marina Ronzhina5, Zdenek Starek6
1International Clinical Research Center, St. Anne’s University Hospital, Brno, Czech Republic, 2Department of Internal Medicine and Cardiology, University Hospital Brno, Brno, Czech Republic, 3Department of Biomedical Engineering, Brno University of Technology, 4Department of Biology, Faculty of Medicine, Masaryk University, Brno, Czech Republic 2 ICRC, St. Anne’s University Hospital, Brno, Czech Republic 3 1st Department of Internal Medicine, Cardio-Angiology, Faculty of Medicine, Masaryk University, Brno, Czech Republic, 5Department of Biomedical Engineering, Brno University of Technology, Brno, Czech Republic, 6ICRC, St. Anne’s University Hospital, Brno, Czech Republic; 1st Department of Internal Medicine, Cardio-Angiology, Faculty of Medicine, Masaryk University, Brno, Czech Republic


Abstract

Timing pattern of intracardiac atrial activity recorded by multipolar catheter in the coronary sinus (CS) provides insightful information about the type and approximate origin of common non-complex arrhythmias. Depending on the anatomy of the CS the atrial activity can be substantially disturbed by ventricular far field complex preventing accurate segmentation by convential methods. In this paper, we present small clinically validated database of 326 surface and intracardiac electrocardiograms (ECG and IECG) and a simple deep learning framework for semantic beat-to-beat segmentation of atrial activity in CS recordings. The model is based on a residual convolutional neural network (CNN) combined with pyramidal upsampling label decoder. It is capable to recognize well between of atrial and ventricular signals recorded by decapolar CS catheter in multiple arrhythmic scenarios reaching dice score of 0.875 on evaluation dataset. To address a dataset size and imbalance issues, we have adopted several preprocessing and learning techniques with adequate evaluation of its impact on the model performance.