Multi-Scale Deep Residual Shrinkage Network for Atrial Fibrillation Recognition

Abstract

In this paper, a novel multi-scale deep residual shrinkage network (MS-DRSN) is proposed for signal denoising and atrial fibrillation (AF) recognition. Signal denoising is done by multi-scale threshold denoising module (MS-TDM), which consists of two parts: threshold acquisition and threshold denoising. The thresholds are automatically obtained through the global attention module constructed by the neural network. Threshold denoising chooses Garrote as the threshold function, which combines the advantages of soft and hard thresholding. The multi-scale features consist of global attention module and local attention module, and then the multi-scale features are denoised using the acquired thresholds and threshold functions, and the AF recognition task is finally completed in the Softmax layer after the superposition of multiple MS-TDMs. An adaptive synthetic sampling (ADASYN) algorithm is also used to oversample the dataset and achieve data category balancing by generating new samples, which improves the accuracy of AF recognition and alleviates the overfitting of the neural network. This method was experimented and validated on the PhysioNet2017 dataset. The experimental results show that the approach achieves an accuracy of 0.894 and an [Formula: see text] score of 0.881, which is better than current machine learning and deep learning models.