论文标题
心律失常分类器使用具有自适应损失感知的多位网络量化的卷积神经网络量化
Arrhythmia Classifier Using Convolutional Neural Network with Adaptive Loss-aware Multi-bit Networks Quantization
论文作者
论文摘要
心血管疾病(CVD)是普遍致命疾病之一,在早期阶段发现它是一项具有挑战性的解决任务。最近,深度学习和卷积神经网络已被广泛用于对象的分类。此外,可以保证可以将许多网络部署在可穿戴设备上。越来越多的方法可用于实现心律失常检测的ECG信号分类。但是,由于有大量的参数导致内存和功耗,因此为心律失常检测提出的现有神经网络对硬件友好不够友好。 在本文中,我们提出了1-D自适应损失感知的量化,达到了高压率,可将记忆消耗降低23.36倍。为了适应我们的压缩方法,我们需要一个较小,更简单的网络。我们提出了一个17层的端到端神经网络分类器,以对MIT-BIH数据集进行培训的17种不同的节奏类别,以实现93.5%的分类精度,这比大多数现有方法高。由于使重要层变得更加关注并提供了修剪无用参数的机会,因此提出的量化方法避免了准确的降解。它甚至提高了准确率,比以前高95.84%,高2.34%。我们的研究实现了具有高性能和低资源消耗的1D卷积神经网络,该网络对硬件友好,并说明了在可穿戴设备上部署以实现实时心律失常诊断的可能性。
Cardiovascular disease (CVDs) is one of the universal deadly diseases, and the detection of it in the early stage is a challenging task to tackle. Recently, deep learning and convolutional neural networks have been employed widely for the classification of objects. Moreover, it is promising that lots of networks can be deployed on wearable devices. An increasing number of methods can be used to realize ECG signal classification for the sake of arrhythmia detection. However, the existing neural networks proposed for arrhythmia detection are not hardware-friendly enough due to a remarkable quantity of parameters resulting in memory and power consumption. In this paper, we present a 1-D adaptive loss-aware quantization, achieving a high compression rate that reduces memory consumption by 23.36 times. In order to adapt to our compression method, we need a smaller and simpler network. We propose a 17 layer end-to-end neural network classifier to classify 17 different rhythm classes trained on the MIT-BIH dataset, realizing a classification accuracy of 93.5%, which is higher than most existing methods. Due to the adaptive bitwidth method making important layers get more attention and offered a chance to prune useless parameters, the proposed quantization method avoids accuracy degradation. It even improves the accuracy rate, which is 95.84%, 2.34% higher than before. Our study achieves a 1-D convolutional neural network with high performance and low resources consumption, which is hardware-friendly and illustrates the possibility of deployment on wearable devices to realize a real-time arrhythmia diagnosis.