Skip to content
sleepKIT
Reference
HELIA

Sleep Detection Model

The following table provides the latest pre-trained models for sleep detection. Below we also provide additional details including training configuration, accuracy metrics, and hardware performance results for the models.

NAMELOCATION# CLASSESMODELPARAMSFLOPSACCURACYAP
SD-2-TCN-SMWrist2TCN9K1.5M/hr95.9%94.5%

The SD-2-TCN-SM model is a 2-stage sleep detection model that uses a Temporal Convolutional Network (TCN) architecture. The model is trained on accelerometer sensor data collected from the wrist and is able to classify sleep and wake stages.

  • Location: Wrist
  • Classes: Awake, Sleep
  • Frame Size: 2 hours
  • Datasets: MESA
  • Features: FS-W-A-5

The confusion matrix provides a detailed breakdown of the model’s performance on the test set. The rows represent the true labels, while the columns represent the predicted labels. The diagonal elements represent the number of correct predictions for each class, while the off-diagonal elements represent the number of incorrect predictions.

The following plot shows the model’s performance in detecting sleep efficieny- percentage of time spent asleep while in bed. The x-axis represents the true sleep efficiency, while the y-axis represents the predicted sleep efficiency.

The following plot shows the model’s performance in detecting total sleep time (TST). The x-axis represents the true TST, while the y-axis represents the predicted TST. The plot shows a strong correlation between the true and predicted TST values.


The following table provides the latest performance results when running on Apollo4 Plus EVB. These results are obtained using neuralSPOTs Autodeploy tool. From neuralSPOT repo, the following command can be used to capture EVB results via Autodeploy:

Terminal
python -m ns_autodeploy \
--tflite-filename model.tflite \
--model-name model \
--cpu-mode 192 \
--arena-size-scratch-buffer-padding 0 \
--max-arena-size 60 \
NameParamsFLOPSMetricTimeArenaEnergy
SD-2-TCN-SM9K1.5M/hr94.5% AP61ms/hr34KB586uJ/hr
AssetDescription
configuration.jsonConfiguration file
model.kerasKeras Model file
model.tfliteTFLite Model file
metrics.jsonMetrics file