Skip to content
heartKIT
Reference
HELIA

3-Class Beat Classification (BEAT-3-EFF-SM)

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

NAMEDATASETFSDURATION# CLASSESMODELPARAMSFLOPSMETRIC
BC-3-EFF-SMIcentia11k100Hz5s3EfficientNetV241K2.1M92.0% F1

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

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

The model is able to distinguish between normal sinus rhythm (NSR), premature/ectopic atrial contractions (PAC), and premature/ectopic ventricular contractions (PVC). The class mapping is as follows:

Base ClassTarget ClassLabel
0-NSR0NSR
1-PAC1PAC
2-PVC2PVC

The model is trained on the following datasets:


The confusion matrix for the model is depicted below.


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