File size: 3,248 Bytes
4eef0c2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
92539f7
 
 
 
 
 
 
4eef0c2
 
 
 
 
 
92539f7
4eef0c2
92539f7
 
 
4eef0c2
 
 
 
 
 
92539f7
 
 
 
 
4eef0c2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
---
language: en
license: apache-2.0
tags:
  - sleep-staging
  - eeg
  - pytorch
  - physioex
pretty_name: PhysioEx Pretrained Models
---

# PhysioEx — Pretrained Sleep Staging Models

Pretrained deep learning models for automatic sleep staging from [PhysioEx](https://github.com/guidogagl/physioex) (Gagliardi et al. 2025).

All models follow the AASM 5-class standard: **W, N1, N2, N3, REM**.

## Available Models

| Model | Architecture | Training Dataset | ACC | F1 | κ | Params | Reference |
|---|---|---|---|---|---|---|---|
| `sleeptransformer-phan` | Transformer | SHHS | **0.866** | **0.796** | **0.810** | 2.1M | Phan et al. 2022 |
| `tinysleepnet-supratak` | CNN + LSTM | Sleep-EDF | 0.842 | 0.790 | 0.784 | 83K | Supratak & Guo 2020 |
| `seqsleepnet-phan` | BiLSTM + Attention | Sleep-EDF | 0.810 | 0.758 | 0.742 | 659K | Phan et al. 2019 |
| `chambon2018` | Braindecode CNN | MASS SS3 | 0.826 | 0.731 | 0.736 | 29K | Chambon et al. 2018 |
| `tsinalis-2016` | 2-layer CNN | Sleep-EDF | 0.773 | 0.661 | 0.683 | 145K | Tsinalis et al. 2016 |

## Usage

```python
from physioex.models import load_from_pretrained

model = load_from_pretrained("sleeptransformer-phan", verbose=True)

# Extract or download embeddings
from physioex.models import load_embeddings
path = load_embeddings("sleeptransformer-phan", "hmc", verbose=True)
```

## Pre-extracted Embeddings

| Model | HuggingFace Repo | Datasets |
|---|---|---|
| SeqSleepNet-Phan | [`4rooms/seqsleepnet-phan-embeddings`](https://huggingface.co/datasets/4rooms/seqsleepnet-phan-embeddings) | 25 datasets |
| TinySleepNet-Supratak | [`4rooms/tinysleepnet-supratak-embeddings`](https://huggingface.co/datasets/4rooms/tinysleepnet-supratak-embeddings) | 25 datasets |
| SleepTransformer-Phan | [`4rooms/sleeptransformer-phan-embeddings`](https://huggingface.co/datasets/4rooms/sleeptransformer-phan-embeddings) | 8 datasets |
| Chambon2018 | [`4rooms/chambon2018-embeddings`](https://huggingface.co/datasets/4rooms/chambon2018-embeddings) | 8 datasets |
| Tsinalis-2016 | [`4rooms/tsinalis-2016-embeddings`](https://huggingface.co/datasets/4rooms/tsinalis-2016-embeddings) | 8 datasets |

## Datasets

| Dataset | Source | URL |
|---|---|---|
| Sleep-EDF | PhysioNet | https://physionet.org/content/sleep-edfx/1.0.0/ |
| HMC | PhysioNet | https://physionet.org/content/hmc-sleep-staging/1.1/ |
| DCSM | ERDA/KU | https://erda.ku.dk/public/archives/db553715ecbe1f3ac66c1dc569826eef/published-archive.html |
| SHHS | NSRR | https://sleepdata.org/datasets/shhs |
| MESA | NSRR | https://sleepdata.org/datasets/mesa |
| HomePAP | NSRR | https://sleepdata.org/datasets/homepap |
| STAGES | NSRR | https://sleepdata.org/datasets/stages |
| MASS | CEAMS | http://ceams-carsm.ca/mass/ |
| WSC | NSRR | https://sleepdata.org/datasets/wsc |
| MrOS | NSRR | https://sleepdata.org/datasets/mros |

## Citations

```bibtex
@article{gagliardi2025physioex,
    author={Gagliardi, Guido and Alfeo, Luca and Cimino, Mario G C A and Valenza, Gaetano and De Vos, Maarten},
    title={PhysioEx, a new Python library for explainable sleep staging through deep learning},
    journal={Physiological Measurement},
    url={http://iopscience.iop.org/article/10.1088/1361-6579/adaf73},
    year={2025},
}
```