ST-MEM

ViT-B/75

VUNO Inc.

12-lead ECG

Filter catalog by Modality:
ECG

General Purpose / Multi-task

Filter catalog by Disease / Trait:
General / Foundation

Embedding

Filter catalog by Task Type:
Representation Learning

Vision Transformer

Filter catalog by Architecture:
Transformer

PyTorch

Filter catalog by Framework:
PyTorch

CC BY-NC 4.0

Filter catalog by License:
Non-commercial / Research-only

file_copy

View model files code View code

Self-supervised ECG foundation model that adapts to varying lead combinations by patchifying 12-lead recordings across both space (leads) and time, then pretraining a ViT-B/75 encoder-decoder with a masked-autoencoder objective. Published at ICLR 2024 by VUNO Inc., and pretrained on the Chapman-Shaoxing-Ningbo dataset along with several other public 12-lead sources.

memory Specifications

gavel License

CC BY-NC 4.0

check_small Open source close_small No commercial use check_small Modifications allowed Attribution required No share-alike requirement

License for model weights only. Associated code may be licensed seperately, check code source for specific terms.

description Publication

Guiding Masked Representation Learning to Capture Spatio-Temporal Relationship of Electrocardiogram open_in_new

2024 · original paper

database Training & evaluation data

Chapman-Shaoxing open_in_new

train

public 45,152 subjects · China

45,152 12-lead, 10-second ECGs from Chapman University / Shaoxing People's Hospital / Ningbo First Hospital with arrhythmia diagnoses; ~56% male / 44% female.

science Capabilities & performance