A method for accurately analyzing electrocardiograms (ECGs), which are obtained from electrical signals generated by cardiac activity, is essential in heart disease diagnosis. However, rhythms are typically obtained with relatively few data samples and similar characteristics, making them difficult to classify. To solve these issues, we proposed a novel method that distinguishes a given ECG rhythm using a beat score map (BSM) image. Through the proposed method, the associations between beats and previously used features, such as the R-R interval, were considered. Rhythm classification was implemented by training a convolutional neural network model and using transfer learning with the created BSM image. As a result, the proposed method for ECG rhythms with small data samples showed significant results. It also showed good performance in differentiating atrial fibrillation (AFIB) and atrial flutter (AFL) rhythms, which are difficult to distinguish due to their similar characteristics. The performance for rhythms with a small number of samples of the proposed method is 20% better than an existing method. In addition, the performance based on the F-1 score for classifying AFIB and AFL of the proposed method is 30% better than the existing method. This study solved the previous limitations caused by small sample numbers and similar rhythms.
Keywords: ECG rhythms; convolutional neural network (CNN); deep learning; rhythm classification.