/MI_Detection_Transformer

Arrhythmia Detection with Vision Transformer using PTB-XL Dataset

Primary LanguageJupyter Notebook

LeeJeongHwi Master's Thesis

Title : Interpretable Arrhythmia Type Classification Using Vision Transformer Model

Environment

  • Python 3.10.13

  • Pytorch 2.1.0

  • Scikit-learn 1.3.2

  • numpy 1.26.0

  • pandas 2.1.3

  • ecg-plot 0.2.8

  • einops 0.7.0

  • pytorch-model-summary 0.1.2

  • scipy 1.11.4

  • seaborn 0.13.0

Database

PTB-XL, a large publicly available electrocardiography dataset 1.0.3

Model Summary

model

Filtering

Highpass Filter (for Powerline Noise Remove)

Before

freq_before

After

freq_af

Median Filter (for Baseline Wander Noise Remove)

Before

Data_wavelet_Bf

After

Data_wavelet_af

Train

Train Loss

Train_loss

Evaluation

Train_loss

Visualization

Conduction Disturbance (Type : LBBB)

head12

Reference

https://github.com/yonigottesman/ecg_vit