Decision Boundary-Based Anomaly Detection Model Using Improved AnoGAN From ECG Data

Abstract

Arrhythmia detection through deep learning is mainly classified through supervised learning. Supervised learning progresses through the labeled data. However, in the medical field, it is challenging to collect ECG data of patients with arrhythmia than ECG data of healthy people, and thus data bias occurs. Therefore, if you use a supervised learning model, there are problems with lack of data and imbalance between labels that arise during learning. Accordingly, this study proposes the decision boundary-based Anomaly detection model using improved AnoGAN from ECG data. In this study, at the time of learning, the loss of the Generator does not reduce, but the loss of a Discriminator lowers. Even if the Generator and Discriminator were designed to have the same learning count, the learning competency of Generator was judged to be lowered. In repeated experiments, it was found that the best loss balance was achieved when the learning count of Discriminator was 1 and that of Generator was 4. Another problem is that the decision boundary of AnoGAN is subjective. Accordingly, the repeated experiments based on F-measure are conducted to determine a decision boundary. For performance evaluation, the accuracy of the model is evaluated on the basis of Epoch, and the goodness-of-fit of the model is evaluated on the basis of AUC and F-measure. According to the evaluation of F-measure, the model has the best performance when the decision boundary is 200. In terms of Epoch, the model has the highest accuracy when the Epoch is 10. In addition, the proposed model has better goodness-of-fit than AnoGAN.

Keywords

• Heart disease
• arrhythmia
• health care
• deep learning
• electrocardiogram
• generative adversarial network