Improving the efficacy of deep-learning models for heart beat detection on heterogeneous datasets

• Main Authors: Bizzego, Andrea, Gabrieli, Giulio, Neoh, Michelle Jin-Yee, Esposito, Gianluca
• Other Authors: School of Social Sciences
• Format: Article
• Language: English
• Published: 2021
• Subjects: Science::Biological sciences::Human anatomy and physiology; Engineering::Computer science and engineering::Computing methodologies::Artificial intelligence; Electrocardiogram Signals; Deep Neural Networks
• Online Access: https://hdl.handle.net/10356/153414
• Institution: Nanyang Technological University

Deep learning (DL) has greatly contributed to bioelectric signal processing, in particular to extract physiological markers. However, the efficacy and applicability of the results proposed in the literature is often constrained to the population represented by the data used to train the models. In this study, we investigate the issues related to applying a DL model on heterogeneous datasets. In particular, by focusing on heart beat detection from electrocardiogram signals (ECG), we show that the performance of a model trained on data from healthy subjects decreases when applied to patients with cardiac conditions and to signals collected with different devices. We then evaluate the use of transfer learning (TL) to adapt the model to the different datasets. In particular, we show that the classification performance is improved, even with datasets with a small sample size. These results suggest that a greater effort should be made towards the generalizability of DL models applied on bioelectric signals, in particular, by retrieving more representative datasets.