Accurate ECG images classification using Vision Transformer
1-1- State of Art
DOI:
https://doi.org/10.56714/bjrs.50.1.26Keywords:
Keywords: Electrocardiogram, Classification, Vision transformer, Deep LearningAbstract
Electrocardiogram (ECG) classification plays a crucial role in the diagnosis and management of cardiovascular diseases. Deep learning-based approaches have shown promising results in automated ECG classification. However, the complexity of ECG signals, including variations in morphology, duration, and amplitude, poses significant challenges for existing deep learning models. In this regard, recent advancements in vision transformer models have shown remarkable performance in images processing and computer vision tasks. In this paper, we propose a deep vision transformer-based approach for ECG classification, which combines the power of convolutional neural networks and self-attention mechanisms. Our proposed model was tuned and enhanced by four hyper-parameters of the proposed model, it can effectively detect internally the main features of ECG images and achieve performance on benchmark ECG datasets. The proposed model can aid in the early detection and diagnosis of cardiovascular diseases, thus improving patient outcomes the final accuracy was 98.23% in dataset
Downloads
References
Kellett, J., & Rasool, S. (2011). The prediction of the in-hospital mortality of acutely ill medical patients by electrocardiogram (ECG) dispersion mapping compared with established risk factors and predictive scores—A pilot study. European Journal of Internal Medicine, 22(4), 394-398.
Safie, S. I., Soraghan, J. J., & Petropoulakis, L. (2011). Electrocardiogram (ECG) biometric authentication using pulse active ratio (PAR). IEEE Transactions on Information Forensics and Security, 6(4), 1315-1322.
Naseer, M. M., Ranasinghe, K., Khan, S. H., Hayat, M., Shahbaz Khan, F., & Yang, M. H. (2021). Intriguing properties of vision transformers. Advances in Neural Information Processing Systems, 34, 23296-23308.
Islam, K. (2022). Recent advances in vision transformer: A survey and outlook of recent work. arXiv preprint arXiv:2203.01536.
Parvaiz, A., Khalid, M. A., Zafar, R., Ameer, H., Ali, M., & Fraz, M. M. (2023). Vision transformers in medical computer vision—A contemplative retrospection. Engineering Applications of Artificial Intelligence, 122, 106126.
Maaz, M., Shaker, A., Cholakkal, H., Khan, S., Zamir, S. W., Anwer, R. M., & Shahbaz Khan, F. (2023, February). Edgenext: efficiently amalgamated cnn-transformer architecture for mobile vision applications. In Computer Vision–ECCV 2022 Workshops: Tel Aviv, Israel, October 23–27, 2022, Proceedings, Part VII (pp. 3-20). Cham: Springer Nature Switzerland.
Mudhafar Jalil Jassim et al.Greedy Learning of Deep Boltzmann Machine (GDBM)’s Variance and Search Algorithm for Efficient Image Retrieval. Publisher: IEEE
Mudhafar Jalil Jassim ,et al. An effective image retrieval based on optimized genetic algorithm utilized a novel SVM-based convolutional neural network classifier. Published: 28 August 2019
Dalmaz, O., Yurt, M., & Çukur, T. (2022). ResViT: residual vision transformers for multimodal medical image synthesis. IEEE Transactions on Medical Imaging, 41(10), 2598-2614.
Mustafa S. Khalefa, Zaid Amin Abduljabar. FINGERPRINT IMAGE ENHANCEMENT BY DEVELOP MEHTRE TECHNIQUE. Published 30 November 2011
Mudhafar Jalil Jassim Ghrabat, Zaid Alaa Hussien, Mustafa S. Khalefa,
Zaid Ameen Abduljabbar,et al.Fully automated model on breast cancer classification using deep learning classifiers. Vol. 28, No. 1, October 2022, pp. 183~191
ISSN:25024752,DOI:10.11591/ijeecs.v28.i1.pp183-191
Zhao, Y., Wang, G., Tang, C., Luo, C., Zeng, W., & Zha, Z. J. (2021). A battle of network structures: An empirical study of cnn, transformer, and mlp. arXiv preprint arXiv:2108.13002.
Liu, T., Si, Y., Yang, W., Huang, J., Yu, Y., Zhang, G., & Zhou, R. (2022). Inter-Patient Congestive Heart Failure Detection Using ECG-Convolution-Vision Transformer Network. Sensors, 22(9), 3283
Jamil, S., & Roy, A. M. (2023). An efficient and robust phonocardiography (pcg)-based valvular heart diseases (vhd) detection framework using vision transformer (vit). Computers in Biology and Medicine, 158, 106734.
Zhou, D., Kang, B., Jin, X., Yang, L., Lian, X., Jiang, Z., ... & Feng, J. (2021). Deepvit: Towards deeper vision transformer. arXiv preprint arXiv:2103.11886
Liu, T., Si, Y., Yang, W., Huang, J., Yu, Y., Zhang, G., & Zhou, R. (2022). Inter-Patient Congestive Heart Failure Detection Using ECG-Convolution-Vision Transformer Network. Sensors, 22(9), 3283
Chen, C. F. R., Fan, Q., & Panda, R. (2021). Crossvit: Cross-attention multi-scale vision transformer for image classification. In Proceedings of the IEEE/CVF international conference on computer vision (pp. 357-366).
Arnab, A., Dehghani, M., Heigold, G., Sun, C., Lučić, M., & Schmid, C. (2021). Vivit: A video vision transformer. In Proceedings of the IEEE/CVF international conference on computer vision (pp. 6836-6846)
Han, K., Wang, Y., Chen, H., Chen, X., Guo, J., Liu, Z., ... & Tao, D. (2022). A survey on vision transformer. IEEE transactions on pattern analysis and machine intelligence, 45(1), 87-110.
Ranftl, R., Bochkovskiy, A., & Koltun, V. (2021). Vision transformers for dense prediction. In Proceedings of the IEEE/CVF International Conference on Computer Vision (pp. 12179-12188).
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2024 Basrah Researches Sciences
This work is licensed under a Creative Commons Attribution 4.0 International License.