/MassMIB

paper

Primary LanguagePython

Masked Self-Supervised ECG Representation Learning via Multiview Information Bottleneck

This is the code for the paper "Masked Self-Supervised ECG Representation Learning via Multiview Information Bottleneck"

Dependency

  • python>=3.7
  • pytorch>=1.7.0
  • torchvision>=0.8.1
  • numpy>=1.19.5
  • tqdm>=4.62.0
  • scipy>=1.5.4
  • wfdb>=3.2.0
  • scikit-learn>=0.24.2

Dataset

PTB-XL dataset can be downloaded from PTB-XL, a large publicly available electrocardiography dataset v1.0.1 (physionet.org).

CPSC2018 dataset can be downloaded from The China Physiological Signal Challenge 2018 (icbeb.org)

Citation

If you find this idea useful in your research, please consider citing:

@article{
  title={Masked Self-Supervised ECG Representation Learning via Multiview Information Bottleneck},
}