TLFT: Transfer Learning and Fourier Transform for ECG Classification

Abstract

Electrocardiogram (ECG) provides a non-invasive method for identifying cardiac issues, particularly arrhythmias or irregular heartbeats. In recent years, the fields of artificial intelligence and machine learning have made significant inroads into various healthcare applications, including the development of arrhythmia classifiers using deep learning techniques. However, a persistent challenge in this domain is the limited availability of large, well-annotated ECG datasets, which are crucial for building and evaluating robust machine learning models. To address this limitation, we propose a novel deep transfer learning framework designed to perform effectively on small training datasets. Our approach involves fine-tuning ResNet-18, a general-purpose image classifier, using the MIT-BIH arrhythmia dataset. This method aims to leverage the power of transfer learning to overcome the constraints of limited data availability. Furthermore, this paper conducts a critical examination of existing deep learning models in the field of ECG analysis. Our investigation reveals that many of these models suffer from methodological flaws, particularly in terms of data leakage. This issue potentially leads to overly optimistic performance estimates and raises concerns about the reliability and generalizability of these models in real-world clinical applications. By addressing these challenges, our work contributes to the advancement of more robust and reliable ECG analysis techniques, potentially improving the accuracy and applicability of automated arrhythmia detection in clinical settings.