| --- |
| 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}, |
| } |
| ``` |
|
|
