Skip to content

In this project, we will perform 12-lead ECG Multi-label Classification. Specifically, we will design a multi-model utilizing the characteristics of diagnoses from the Shaoxing and Ningbo databases.

License

Notifications You must be signed in to change notification settings

JaeBinCHA7/ECG-Multi-Label-Classification-Using-Multi-Model

Repository files navigation

ECG-Multi-Label-Classification-Using-Multi-Model

[paper] In this project, we will perform 12-lead ECG Multi-label Classification. Specifically, we will design a multi-model utilizing the characteristics of diagnoses from the Shaoxing and Ningbo databases.

Update

  • 2024.05.08

  • Requirements

This repo is implemented in Ubuntu 22.04, PyTorch 2.3.0, Python3.11, and CUDA12.0. For package dependencies, you can install them by:

pip install -r requirements.txt    

Dataset Installation

In this paper, we used a large-scale ECG database consisting of 45,152 12-lead electrocardiograms [18, 19]. The database is annotated with a total of 94 labels through the Systematized Nomenclature of Medicine Clinical Terms (SNOMED CT). For ease of classification, we selected 41 SNOMED CT codes, including those for arrhythmias and ischemic cardiovascular diseases, from the 94 codes based on the hierarchical structure of SNOMED CT codes and the Minnesota Code Manual. These were then consolidated into 20 groups.
https://physionet.org/content/ecg-arrhythmia/1.0.0/

Getting Started

  1. Install the necessary libraries.
  2. Set directory paths for your dataset. (options.py)
  3. Run train.py

Architecture [3]

Results

References

[1] Zheng, Jianwei, et al. "A 12-lead electrocardiogram database for arrhythmia research covering more than 10,000 patients." Scientific data 7.1 (2020): 48.
[2] Zheng, J. "Optimal multi-stage arrhythmia classification approach Sci." Reports 101 (2020): 1-17.
[3] Hwang, Seorim, et al. "Multi-label ECG Abnormality Classification Using A Combined ResNet-DenseNet Architecture with ResU Blocks." 2023 IEEE EMBS Special Topic Conference on Data Science and Engineering in Healthcare, Medicine and Biology. IEEE, 2023.** [paper] [code]

Contact

E-mail: [email protected]

About

In this project, we will perform 12-lead ECG Multi-label Classification. Specifically, we will design a multi-model utilizing the characteristics of diagnoses from the Shaoxing and Ningbo databases.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages