Skip to content
heartKIT
Reference
HELIA

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

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.

NAMEDATASETFSDURATION# CLASSESMODELPARAMSFLOPSMETRIC
SEG-4-TCN-LGLUDB, Synthetic100Hz2.5s4TCN10K3.9M89.4% F1

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

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 ClassTarget ClassLabel
0-NONE0NONE
1-PWAVE1PWAVE
2-QRS2QRS
3-TWAVE3TWAVE

The model is trained on the following datasets:


The confusion matrix for the segmentation model is depicted below.


AssetDescription
configuration.jsonConfiguration file
model.kerasKeras Model file
model.tfliteTFLite Model file
metrics.jsonMetrics file