Automatic Prediction of Atrial Fibrillation Based on Convolutional Neural Network Using a Short-term Normal Electrocardiogram Signal

摘要

Background In this study, we propose a method for automatically predicting atrial fibrillation (AF) based on convolutional neural network (CNN) using a short-term normal electrocardiogram (ECG) signal. Methods We designed a CNN model and optimized it by dropout and normalization. One-dimensional convolution, max-pooling, and fully-connected multiple perceptron were used to analyze the short-term normal ECG. The ECG signal was preprocessed and segmented to train and evaluate the proposed CNN model. The training and test sets consisted of the two AF and one normal dataset from the MIT-BIH database. Results The proposed CNN model for the automatic prediction of AF achieved a high performance with a sensitivity of 98.6%, a specificity of 98.7%, and an accuracy of 98.7%. Conclusion The results show the possibility of automatically predicting AF based on the CNN model using a short-term normal ECG signal. The proposed CNN model for the automatic prediction of AF can be a helpful tool for the early diagnosis of AF in healthcare fields.