There is a newer version of the record available.

Published January 22, 2020 | Version v1.0
Other Open

Pre-trained deep neural network models for ECG automatic abnormality detection

Description

# Pre-trained deep neural network models for ECG automatic abnormality detection

Contain the pre-trained models of the deep neural networks described on the paper:
"Automatic Diagnosis of the Short-Duration12-Lead ECG using a Deep Neural Network".

All files are in the format `.hdf5` and can be read using
```python
from keras.models import load_model
from keras.optimizers import Adam
model = load_model(args.model, compile=False)
model.compile(loss='binary_crossentropy', optimizer=Adam())
```
The model take as input tensors with dimension `(batch_size, 4096, 12)` where 4096 are samples of the ecg signal 
(sampled during, approximately, 10s at 400Hz) from 12 different leads. All signal are represented as
32 bits floating point numbers at the scale 1e-4V: so if the signal is in V it should be multiplied by 
1000 before feeding it to the neural network model. 
 
 
The neural network yield an output with dimension `(batch_size, 6)`. Each entry correspond to probability between 0 and 1
on the giving exam. It does that for 6 different ECGs abnormalities: (in that order)
- 1st degree AV block (1dAVb);
- right bundle branch block (RBBB);
- left bundle branch block (LBBB);
- sinus bradycardia (SB);
- atrial fibrillation (AF); and,
- sinus tachycardia (ST).
These abnormalities are not mutually exclusive and the probabilities outputs of the model does not necessarely sums to 1.

## Folder content

- The main model used along the paper is the one named `model.hdf5`. 

- In order to show the stability of the method we also train 10 different neural networks
with the same architecture and configuration but with different
initial seeds. These models are saved as `other_seeds/model_[1-10].hdf5`. The main 
model `model.hdf5` correspond to  `other_seeds/model_6.hdf5` which is the one with 
micro average precision imediatly above the median value.

- Finally, to assess the effect of how we structure our problem, we have considered alternative
scenarios where we use 90\%-5\%-5\% splits, stratified randomly, by patient or in chronological order. 
The models trained in these scenarios are saved in the folder `other_splits/`.

Files

model.zip

Files (335.9 MB)

Name Size
md5:d073ad8fd5f1e277940a35d830ee4174
335.9 MB Preview Download

Additional details

Related works

Is supplement to
Preprint: arXiv:1904.01949 (arXiv)