ECGNet: An Efficient Network for Detecting Premature Ventricular Complexes Based on ECG Images
Zeyang Zhang, Ziheng Zhang, Cao Zou, Zhongcai Pei, Zheyuan Yang, Jing Wu, Shikun Sun, Fei Gu
Abstract
BACKGROUND: Preoperative prediction of the origin site of premature ventricular complexes (PVCs) is critical for the success of operations. However, current methods are not efficient or accurate enough. In addition, among the proposed strategies, there are few good prediction methods for electrocardiogram (ECG) images combined with deep learning aspects. METHODS: We propose ECGNet, a new neural network for the classification of 12-lead ECG images. In ECGNet, 609 ECG images from 310 patients who had undergone successful surgery in the Division of Cardiology, the First Affiliated Hospital of Soochow University, are utilized to construct the dataset. We adopt dense blocks, special convolution kernels and divergent paths to improve the performance of ECGNet. In addition, a new loss function is designed to address the sample imbalance situation, whose cause is the uneven distribution of cases themselves, which often occurs in the medical field. We also conduct extensive experiments in terms of network prediction accuracy to compare ECGNet with other networks, such as ResNet and DarkNet. RESULTS: Our ECGNet achieves extremely high prediction accuracy (91.74%) and efficiency with very small datasets. Our newly proposed loss function can solve the problem of sample imbalance during the training process. CONCLUSION: The proposed ECGNet can quickly and accurately realize the multiclassification of PVCs after training with little data. Our network has the potential to be helpful to doctors with a preoperative diagnosis of PVCs. We will continue to collect similar cases and perfect our network structure to further improve the accuracy of our network's prediction.