Predicting the Risk of Depression Based on ECG Using RNN

Abstract

This paper presents a model to predict the risk of depression based on electrocardiogram (ECG). This proposed model uses a Recurrent Neural Network (RNN) and Long Short-Term Memory (LSTM) autoencoder to predict normal, abnormal, and PVC heartbeats. The RNN model is a deep learning-based model to classify normal, abnormal, and PVC heartbeats. We used the model as a classifier. The model uses a heart rates dataset to predict abnormal and PVC heartbeats. As for the dataset, we have used 5000 ECG samples. The model was trained on a training dataset and validation dataset. After that, it was tested on a test dataset. The model is trained on normal heartbeat rates, so the model can predict any heartbeat rates other than normal. Our contribution here is to build a model that can differentiate between “normal,” “abnormal,” and “risky” heartbeats. Our model predicts “normal” heartbeats with 97.24% accuracy and can predict “PVC” heartbeats with 100% accuracy. Other than the accuracy, we evaluated our model on the training loss graphs. These two types of training loss graphs were evaluated as “normal” versus “risky” and “abnormal” versus “risky.” We have seen great results there as well. The best losses for “normal,” “abnormal,” and “risky” are 5.71, 33.36, and 34.78. However, these results may improve if a larger dataset is used. In studies, it was found that patients suffering from depression may have a different kind of heartbeat than the normal ones. In most cases, it is PVC (Premature Ventricular Contraction) heartbeats. Therefore, the target is to predict abnormal heartbeats and PVC heartbeats.