Two-stream Network for ECG Signal Classification

Xinyao Hou,Shengmei Qin,Jianbo Su
DOI: https://doi.org/10.48550/arXiv.2210.06293
2022-10-05
Abstract:Electrocardiogram (ECG), a technique for medical monitoring of cardiac activity, is an important method for identifying cardiovascular disease. However, analyzing the increasing quantity of ECG data consumes a lot of medical resources. This paper explores an effective algorithm for automatic classifications of multi-classes of heartbeat types based on ECG. Most neural network based methods target the individual heartbeats, ignoring the secrets embedded in the temporal sequence. And the ECG signal has temporal variation and unique individual characteristics, which means that the same type of ECG signal varies among patients under different physical conditions. A two-stream architecture is used in this paper and presents an enhanced version of ECG recognition based on this. The architecture achieves classification of holistic ECG signal and individual heartbeat and incorporates identified and temporal stream networks. Identified networks are used to extract features of individual heartbeats, while temporal networks aim to extract temporal correlations between heartbeats. Results on the MIT-BIH Arrhythmia Database demonstrate that the proposed algorithm performs an accuracy of 99.38\%. In addition, the proposed algorithm reaches an 88.07\% positive accuracy on massive data in real life, showing that the proposed algorithm can efficiently categorize different classes of heartbeat with high diagnostic performance.
Signal Processing,Artificial Intelligence,Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?
The problem that this paper attempts to solve is to improve the accuracy and efficiency of electrocardiogram (ECG) signal classification, especially for the automatic classification of heartbeat types for heart diseases. With the continuous increase in the amount of ECG data in medical monitoring, manual analysis of these data not only consumes a large amount of medical resources but is also error - prone, resulting in unstable diagnosis results. Therefore, there is an urgent need for a method that can analyze ECG data quickly and accurately. This paper proposes a two - stream network architecture, aiming to overcome the challenges of existing methods in processing ECG signals, such as ignoring the secrets in the time series and the changes of the same type of ECG signals in different patients and under different physical conditions. Specifically, the main contributions of this paper include: - **Proposing a new two - stream network architecture**: This architecture combines the Identified Stream Network and the Temporal Stream Network, which are respectively used to extract the features of a single heartbeat and the time correlation between heartbeats. - **Improving the accuracy of ECG signal classification**: Verified by experiments, the accuracy of this algorithm on the MIT - BIH arrhythmia database reaches 99.38%, and the positive accuracy on large - scale data in real life reaches 88.07%. - **Being applicable to the classification of multiple heartbeat types**: This model can not only efficiently classify different heartbeat types but also has high diagnostic performance and can cope with the time - varying and individual differences of ECG signals. In conclusion, by introducing the two - stream network architecture, this paper solves the key problems in the automatic classification of ECG signals and provides strong support for the diagnosis and treatment of heart diseases.