Explorando Redes Neurais de Grafos para Classificação de Arritmias

• Rafael F. Oliveira UFOP
• Vander L. S. Freitas UFOP
• Gladston J. P. Moreira UFOP
• Eduardo J. S. Luz UFOP

DOI: https://doi.org/10.5753/sbcas.2022.222510

Abstract

Cardiac arrhythmia is a risky condition and its early diagnosis is of paramount importance. Therefore, the automation process is desirable. In this context, the Graph Neural Network (GNN) is explored to classify arrhythmias in electrocardiogram (ECG) signals in which each heartbeat is transformed in a graph via the Visibility Graph (VG) algorithm. Three models of GNN networks are evaluated: a Graph Convolutional Network with two layers (GCN-2L), another with three layers (GCN-3L), and GraphSAGE (GraphSAGE-4L). Following the inter-patient paradigm, the GraphSAGE-4L model presented the best performance, obtaining an average F1-score of 0.86 for the classes N, S and V.

References

ANSI/AAMI (2008). Testing and reporting performance results of cardiac rhythm and ST segment measurement algorithms. American National Standards Institute, Inc. (ANSI), Association for the Advancement of Medical Instrumentation (AAMI). ANSI/AAMI/ISO EC57, 1998-(R)2008.

Cohen, A. (1986). Biomedical Signal Processing: Compression and automatic recognition. CRC Press, 1st edition.

De Chazal, P., O'Dwyer, M., and Reilly, R. B. (2004). Automatic classification of heartbeats using ecg morphology and heartbeat interval features. IEEE transactions on biomedical engineering, 51(7):1196-1206.

Do, K., Tran, T., and Venkatesh, S. (2019). Graph transformation policy network for chemical reaction prediction. In Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining, pages 750-760.

Fout, A. M. (2017). Protein interface prediction using graph convolutional networks. PhD thesis, Colorado State University.

Hamilton, W. L., Ying, R., and Leskovec, J. (2017). Inductive representation learning on large graphs. In Proceedings of the 31st International Conference on Neural Information Processing Systems, pages 1025-1035.

Hannun, A. Y., Rajpurkar, P., Haghpanahi, M., Tison, G. H., Bourn, C., Turakhia, M. P., and Ng, A. Y. (2019). Cardiologist-level arrhythmia detection and classification in ambulatory electrocardiograms using a deep neural network. Nature medicine, 25(1):65.

Kipf, T. N. and Welling, M. (2016). Semi-supervised classification with graph convolutional networks. arXiv:1609.02907.

Kojima, R., Ishida, S., Ohta, M., Iwata, H., Honma, T., and Okuno, Y. (2020). kgcn: a graph-based deep learning framework for chemical structures. Journal of Cheminformatics, 12(1):1-10.

Lacasa, L., Luque, B., Ballesteros, F., Luque, J., and Nuno, J. C. (2008). From time series to complex networks: The visibility graph. Proceedings of the National Academy of Sciences, 105(13):4972-4975.

LeCun, Y., Bottou, L., Bengio, Y., and Haffner, P. (1998). Gradient-based learning applied to document recognition. Proceedings of the IEEE, 86(11):2278-2324.

Luz, E. J. d. S., Schwartz, W. R., Cámara-Chávez, G., and Menotti, D. (2016). Ecg-based heartbeat classification for arrhythmia detection: A survey. Computer methods and programs in biomedicine, 127:144-164.

Micheli, A. (2009). Neural network for graphs: A contextual constructive approach. IEEE Transactions on Neural Networks, 20(3):498-511.

Mondéjar-Guerra, V., Novo, J., Rouco, J., Penedo, M. G., and Ortega, M. (2019). Heartbeat classification fusing temporal and morphological information of ecgs via ensemble of classifiers. Biomedical Signal Processing and Control, 47:41-48.

Moody, G. B. and Mark, R. G. (1990). The mit-bih arrhythmia database on cd-rom and software for use with it. In [1990] Proceedings Computers in Cardiology, pages 185-188. IEEE.

Rajesh, K. N. and Dhuli, R. (2018). Classification of imbalanced ecg beats using resampling techniques and adaboost ensemble classifier. Biomedical Signal Processing and Control, 41:242-254.

Rajpurkar, P., Hannun, A. Y., Haghpanahi, M., Bourn, C., and Ng, A. Y. (2017). Cardiologist-level arrhythmia detection with convolutional neural networks. arXiv:1707.01836.

Scarselli, F., Gori, M., Tsoi, A. C., Hagenbuchner, M., and Monfardini, G. (2008). The graph neural network model. IEEE transactions on neural networks, 20(1):61-80.

Shi, H., Wang, H., Huang, Y., Zhao, L., Qin, C., and Liu, C. (2019). A hierarchical method based on weighted extreme gradient boosting in ecg heartbeat classification. Computer methods and programs in biomedicine, 171:1-10.

Sperduti, A. and Starita, A. (1997). Supervised neural networks for the classification of structures. IEEE Transactions on Neural Networks, 8(3):714-735.

Wang, L., Sun, W., Chen, Y., Li, P., and Zhao, L. (2018a). Wavelet transform based ecg denoising using adaptive thresholding. In Proceedings of the 2018 7th International Conference on Bioinformatics and Biomedical Science, pages 35-40.

Wang, M., Zheng, D., Ye, Z., Gan, Q., Li, M., Song, X., Zhou, J., Ma, C., Yu, L., Gai, Y., Xiao, T., He, T., Karypis, G., Li, J., and Zhang, Z. (2019). Deep graph library: A graph-centric, highly-performant package for graph neural networks. arXiv:1909.01315.

Wang, T., Lu, C., Sun, Y., Yang, M., Liu, C., and Ou, C. (2021). Automatic ecg classification using continuous wavelet transform and convolutional neural network. Entropy, 23(1):119.

Wang, Z., Chen, T., Ren, J., Yu, W., Cheng, H., and Lin, L. (2018b). Deep reasoning with knowledge graph for social relationship understanding. arXiv:1807.00504.

Wu, Z., Pan, S., Chen, F., Long, G., Zhang, C., and Philip, S. Y. (2020). A comprehensive survey on graph neural networks. IEEE transactions on neural networks and learning systems, 32(1):4-24.

Xu, K., Hu, W., Leskovec, J., and Jegelka, S. (2018). How powerful are graph neural networks? arXiv:1810.00826.

Yang, P., Wang, D., Zhao, W.-B., Fu, L.-H., Du, J.-L., and Su, H. (2021). Ensemble of kernel extreme learning machine based random forest classifiers for automatic heartbeat classification. Biomedical Signal Processing and Control, 63:102138.

Zheng, J., Zhang, J., Danioko, S., Yao, H., Guo, H., and Rakovski, C. (2020). A 12-lead electrocardiogram database for arrhythmia research covering more than 10,000 patients. Scientific Data, 7(1):1-8.

Zhou, J., Cui, G., Hu, S., Zhang, Z., Yang, C., Liu, Z., Wang, L., Li, C., and Sun, M. (2020). Graph neural networks: A review of methods and applications. AI Open, 1:57-81.