license: mit
ECGFounder: An Electrocardiogram Foundation Model Built on over 10 Million Recordings with External Evaluation across Multiple Domains
This is the official implementation of our paper "An Electrocardiogram Foundation Model Built on over 10 Million Recordings with External Evaluation across Multiple Domains".
Authors: Jun Li, Aaron Aguirre, Junior Moura, Jiarui Jin, Che Liu, Lanhai Zhong, Chenxi Sun, Gari Clifford, Brandon Westover, Shenda Hong.
🚀 Getting Started
🚩 News (Mar 2025): The pre-training checkpoint is now available on 🤗 Hugging Face!
Installation
To clone this repository:
git clone https://github.com/PKUDigitalHealth/ECGFounder.git
Environment Set Up
Install required packages:
conda create -n ECGFounder python=3.10
conda activate ECGFounder
pip install -r requirements.txt
Fine-tune on Downstream Tasks
In our paper, downstream datasets we used are as follows:
- MIMIC-ECG: Please download the MIMIC-ECG dataset from physionet.
Next, please download the model's checkpoint from the 🤗 Hugging Face. And place the model weights in path ./checkpoint
You can run the jupyter notebook to finetune the model by the example dataset.
References
If you found our work useful in your research, please consider citing our works at:
@article{li2024electrocardiogram, title={An Electrocardiogram Foundation Model Built on over 10 Million Recordings with External Evaluation across Multiple Domains}, author={Li, Jun and Aguirre, Aaron and Moura, Junior and Liu, Che and Zhong, Lanhai and Sun, Chenxi and Clifford, Gari and Westover, Brandon and Hong, Shenda}, journal={arXiv preprint arXiv:2410.04133}, year={2024} }