Skip to content

4-Stage ECG Segmentation (SEG-4-TCN-LG)

Overview

The following table provides the latest pre-trained model for 4-class ECG segmentation. Below we also provide additional details including training configuration, performance metrics, and downloads.

NAME DATASET FS DURATION # CLASSES MODEL PARAMS FLOPS METRIC
SEG-4-TCN-LG LUDB, Synthetic 100Hz 2.5s 4 TCN 10K 3.9M 89.4% F1

Input

The model is trained on 2.5-second, raw ECG frames sampled at 100 Hz.

  • Sensor: ECG
  • Location: Wrist
  • Sampling Rate: 100 Hz
  • Frame Size: 2.5 seconds

Class Mapping

The model is able to segment ECG signals into four classes: P-wave, QRS complex, T-wave, and none. The class mapping is as follows:

Base Class Target Class Label
0-NONE 0 NONE
1-PWAVE 1 PWAVE
2-QRS 2 QRS
3-TWAVE 3 TWAVE

Datasets

The model is trained on the following datasets:


Model Performance

The confusion matrix for the segmentation model is depicted below.


Downloads

Asset Description
configuration.json Configuration file
model.keras Keras Model file
model.tflite TFLite Model file
metrics.json Metrics file