Augmentation Strategies for Self-Supervised Representation Learning from Electrocardiograms

Andersson, Matilda; Nilsson, Mattias; Flood, Gabrielle; Aström, Kalle (2023). Augmentation Strategies for Self-Supervised Representation Learning from Electrocardiograms 31st European Signal Processing Conference, EUSIPCO 2023 - Proceedings, 1075 - 1079. 31st European Signal Processing Conference, EUSIPCO 2023. Helsinki, Finland: European Signal Processing Conference, EUSIPCO
Download:
DOI:
Conference Proceeding/Paper | Published | English
Authors:
Andersson, Matilda ; Nilsson, Mattias ; Flood, Gabrielle ; Aström, Kalle
Department:
Mathematics (Faculty of Engineering)
eSSENCE: The e-Science Collaboration
ELLIIT: the Linköping-Lund initiative on IT and mobile communication
LTH Profile Area: AI and Digitalization
Mathematical Imaging Group
LTH Profile Area: Engineering Health
Research Group:
Mathematical Imaging Group
Abstract:

In this paper, we investigate the effects of different augmentation strategies in self-supervised representation learning from electrocardiograms. Our study examines the impact of random resized crop and time out on downstream performance. We also consider the importance of the signal length. Furthermore, instead of using two augmented copies of the sample as a positive pair, we suggest augmenting only one. The second signal is kept as the original signal. These different augmentation strategies are investigated in the context of pre-training and fine-tuning, following the different self-supervised learning frameworks BYOL, SimCLR, and VICReg. We formulate the downstream task as a multi-label classification task using a public dataset containing ECG recordings and annotations. In our experiments, we demonstrate that self-supervised learning can consistently outperform classical supervised learning when configured correctly. These findings are of particular importance in the medical domain, as the medical labeling process is particularly expensive, and clinical ground truth is often difficult to define. We are hopeful that our findings will be a catalyst for further research into augmentation strategies in self-supervised learning to improve performance in the detection of cardiovascular disease.

Keywords:
augmentation ; ECG ; electrocardiogram ; pre-processing ; representation learning ; self-supervised
ISBN:
9789464593600
ISSN:
2219-5491
LUP-ID:
2c07f416-037f-41cd-95a1-e4e6dc105fbf | Link: https://lup.lub.lu.se/record/2c07f416-037f-41cd-95a1-e4e6dc105fbf | Statistics

Cite this