Qin Jing, Gao Fujie, Wang Zumin, Wong David C, Zhao Zhibin, Relton Samuel D, Fang Hui
College of Software Engineering, Dalian University, Dalian, China.
College of Information Engineering, Dalian University, Dalian, China.
Artif Intell Med. 2023 Feb;136:102489. doi: 10.1016/j.artmed.2023.102489. Epub 2023 Jan 13.
Cardiac abnormality detection from Electrocardiogram (ECG) signals is a common task for cardiologists. To facilitate efficient and objective detection, automated ECG classification by using deep learning based methods have been developed in recent years. Despite their impressive performance, these methods perform poorly when presented with cardiac abnormalities that are not well represented, or absent, in the training data. To this end, we propose a novel one-class classification based ECG anomaly detection generative adversarial network (GAN). Specifically, we embedded a Bi-directional Long-Short Term Memory (Bi-LSTM) layer into a GAN architecture and used a mini-batch discrimination training strategy in the discriminator to synthesis ECG signals. Our method generates samples to match the data distribution from normal signals of healthy group so that a generalised anomaly detector can be built reliably. The experimental results demonstrate our method outperforms several state-of-the-art semi-supervised learning based ECG anomaly detection algorithms and robustly detects the unknown anomaly class in the MIT-BIH arrhythmia database. Experiments show that our method achieves the accuracy of 95.5% and AUC of 95.9% which outperforms the most competitive baseline by 0.7% and 1.7% respectively. Our method may prove to be a helpful diagnostic method for helping cardiologists identify arrhythmias.
从心电图(ECG)信号中检测心脏异常是心脏病专家的一项常见任务。为了实现高效、客观的检测,近年来已开发出基于深度学习方法的自动ECG分类技术。尽管这些方法表现出色,但当面对训练数据中未充分体现或不存在的心脏异常时,它们的性能会变差。为此,我们提出了一种基于一类分类的新型ECG异常检测生成对抗网络(GAN)。具体而言,我们将双向长短期记忆(Bi-LSTM)层嵌入到GAN架构中,并在判别器中使用小批量判别训练策略来合成ECG信号。我们的方法生成样本以匹配健康组正常信号的数据分布,从而能够可靠地构建一个广义异常检测器。实验结果表明,我们的方法优于几种基于半监督学习的先进ECG异常检测算法,并能在MIT-BIH心律失常数据库中稳健地检测出未知异常类别。实验表明,我们的方法准确率达到95.5%,AUC为95.9%,分别比最具竞争力的基线高出0.7%和1.7%。我们的方法可能被证明是一种有助于心脏病专家识别心律失常的有用诊断方法。