ECG Classification Pytorch

Application of deep learning and convolutional networks for ECG classification

The primary objective of this project is to use a 1D Convolutional Network paired with a Multilayer perceptron that finds unhealthy signal in a continous heart beat. All code is public and can be manipulated for use.

Instructions for set Up:

Make sure heartbeat.py and normalizer.py are in the same directory. heartbeat.py is a small written signal processing library and normalizer.py has several other util functions and is based on (requires) heartbeat.py
Folder mit_data (attached in data file) where all data is stored. This must also be in the same directory for heart beat to work.

What's in this Repo:

mit_data: data used in training as .txt & .csv files
heartbeat.py is a small written signal processing library and normalizer.py has several other util functions
torch_tools.py: all torch functions used (py file just holds the code and is not needed per say)
ECG_notebook.ipynb: Notebook with how to create classifier in Pytorch (requires heartbeat.py and normalizer.py)
PPT and PDF of associated written paper included

Description:

Electrocardiograms or ECG are signals that assess an individual’s cardiac rhythm as signal. This is a result of the depolarization and repolarization of the hearts’ chambers that can be interpreted voltage over time. The premise of this paper is the application of supervised deep learning to identify illustrations of labeled rhythmic aberrations. The proposed technique uses a series of single dimensional convolutions paired with a multilayered perceptron to classify five common arrhythmias. The model was trained with 75% of the data which was sampled with equal class counts per batch but tested on the data’s natural distribution. After implementation, the accuracy of the model proved to be 97.5 + .0044% over five iterations with per class metrics higher than 85% across all classes. Future improvements include different processing techniques as well as slight adjustment to model architecture.

Data:

The data (mit_data folder) used will be from the MIT-BIH Arrhythmia Database. The following database, collecting data from as far as 1975 has 48 instances of 30 minute (360 samples/sec) ECG records from 47 patients at Beth-Israel Hospital. “Twenty-three recordings were chosen at random from a set of 4000 24-hour ambulatory ECG recordings collected from a mixed population of inpatients (about 60%) and outpatients (about 40%). The remaining 25 recordings were selected from the same set to include less common but clinically significant arrhythmias that would not be well-represented in a small random sample.” All data exists per patients in the form of csv files and instances of annotated by associate txt files.

Database: https://physionet.org/physiobank/database/html/mitdbdir/intro.htm

dzr1026 / ecg_classification_pytorch Goto Github PK