YOUR-Lead: YOLO and U-Net for Reconstruction of ECG Lead Signals - 3IA Côte d’Azur – Interdisciplinary Institute for Artificial Intelligence
Pré-Publication, Document De Travail Année : 2024

YOUR-Lead: YOLO and U-Net for Reconstruction of ECG Lead Signals

Résumé

Electrocardiograms are still vastly stored on paper, which restrains the development of data-driven algorithms for their analysis. We propose in this manuscript an efficient method to digitize Electrocardiogram (ECG) signals from paper images, as part of the "Digitization and Classification of ECG Images: The George B. Moody PhysioNet Challenge 2024". A total of 21,799 12-lead recordings from the PTB-XL public data-set were used to generate realistic images. The architecture of our method consists of three main steps. First, we generate ECG paper images with different characteristics such as resolutions, layouts, grid patterns, wrinkles, temperatures, text and noise levels to reproduce real-life challenges. Second, we perform a fully automated ECG image processing pipeline. To do so, a U-Net model is employed for image binarization where only the relevant part of the image is extracted. In addition, a pre-trained YOLO object detection model is fine-tuned to detect the signal box leads. Combining the obtained cleaned ECG image and detected lead boxes, we finally extract 1D ECG signal for each lead present in the ECG paper. By comparing original and estimated ECG signals, our model achieved a signal-to-noise ratio (SNR) score of 0.447 on the hidden validation set.
Fichier principal
Vignette du fichier
372_Preprint.pdf (3.03 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04684731 , version 1 (03-09-2024)

Identifiants

  • HAL Id : hal-04684731 , version 1

Citer

Rafael Silva, Yingyu Yang, Maëlis Morier, Safaa Al-Ali, Maxime Sermesant. YOUR-Lead: YOLO and U-Net for Reconstruction of ECG Lead Signals. 2024. ⟨hal-04684731⟩
18 Consultations
16 Téléchargements

Partager

More