Resumen
Atrial fibrillation (AF) is an arrhythmia that may cause blood clots and increase the risk of stroke and heart failure. Traditional 12-lead electrocardiogram (ECG) acquisition equipment is complex and difficult to carry. Short single-lead ECG recordings based on wearable devices can remedy these shortcomings. However, reliable and accurate atrial fibrillation detection is still an issue because of the limited information on the short single-lead ECG recordings. In this paper, we propose a novel multi-branch convolutional neural network and bidirectional long short-term memory network (MCNN-BLSTM) to deal with the reliability and accuracy of AF detection in short single-lead ECG recordings. Firstly, to fuller extract the feature information of short single-lead ECG recordings, the MCNN module is designed to dynamically set several corresponding branches according to the number of slices of short single-lead ECG recordings. Then, the BLSTM module is designed to further enhance the feature information learned from each branch. We validated the model on the PhysioNet/CinC Challenge 2017 (CinC2017) database and verified the generalization of the model on the China Physiological Signal Challenge 2018 (CPSC2018) database. The results show that the accuracy of the model on the CinC 2017 database reaches 87.57%, and the average F1 score reaches 84.56%. The accuracy of the model on the CPSC 2018 database reaches 87.50%, and the average F1 score reaches 82.01%. Compared with other advanced methods, our model shows better performance and can meet the daily needs of atrial fibrillation detection with short ECG wearable devices.