skip to main content
10.1145/3639592.3639606acmotherconferencesArticle/Chapter ViewAbstractPublication PagesaicccConference Proceedingsconference-collections
research-article

Multi-lead atrial fibrillation classification method using ConvNeXt and BiLSTM

Authors Info & Claims
Published:13 April 2024Publication History

ABSTRACT

Atrial fibrillation is one of the inducing factors of many cardiac abnormalities. Currently, most atrial fibrillation classification methods only use single-lead electrocardiogram data or validate the effectiveness of the algorithm on small-scale datasets. The classification method based on single-lead atrial fibrillation ignores the complementary characteristics of different lead disease information in multi-lead electrocardiograms. If only small-scale datasets are used, the generalization ability of the algorithm cannot be guaranteed. Therefore, a multi-lead classification model for atrial fibrillation is proposed in this paper, which combines ConvNeXt, a convolutional block attention mechanism and a bidirectional long short-term memory network. Based on the characteristics and positional representation of atrial fibrillation on the electrocardiogram, a convolutional block attention mechanism is embedded in the model, enabling the model to effectively extract temporal and spatial features of the electrocardiogram during training. The specificity, sensitivity, accuracy and F1 scores obtained by this method were 98.90%, 98.04%, 98.88% and 74.32%, respectively, through testing more than 150,000 ECG recordings on the Chinese cardiovascular disease database. The experimental results indicate that this method achieves good classification performance and can provide effective technical means for the prevention and auxiliary diagnosis of atrial fibrillation.

References

  1. Islam MS, Ammour N, Alajlan N, Rhythm based heartbeat duration normalization for atrial fibrillation detection[J]. Computers in Biology and Medicine, 2016, 72: 160-169.Google ScholarGoogle ScholarDigital LibraryDigital Library
  2. Moody GB, Mark RR.New method for detecting atrial fibrillation using RR intervals[J]. Computers in cardiology,1983, 10: 227-230.Google ScholarGoogle Scholar
  3. Lan T, Yang C. Prediction of recurrence of paroxysmal atrial fibrillation based on RR interval[J]. Journal of Biomedical Engineering, 2019, 36(4):521-530. (in Chese)Google ScholarGoogle Scholar
  4. Du XC, Rao NN, Qian MY, A novel method for real time atrial fibrillation detection in electrocardiograms using multiple parameters[J]. Ann Noninvasive Electrocardiology, 2014, 19(3): 217-225.Google ScholarGoogle ScholarCross RefCross Ref
  5. Athif M, Yasawardene PC, Daluwatte C. Detecting atrial fibrillation from short single lead ECGs using statistical and morphological features[J]. Physiological Measurement, 2018, 39(6): 064002.Google ScholarGoogle ScholarCross RefCross Ref
  6. Xu YF, Zhang S, Cao ZT, Extreme learning machine for heartbeat classification with hybrid time-domain and wavelet time-frequency features[J]. Journal of Healthcare Engineering, 2021, 2021(6):1-12.Google ScholarGoogle Scholar
  7. Zhou FY, Jin LP, Dong J. Review of convolutional neural network [J]. Chinese Journal of Computers, 2017, 40(06):1229-1251. (in Chese)Google ScholarGoogle Scholar
  8. Kek XY, Chin CS, Li Y. Intelligent low-complexity computing interleaving wavelet scattering based mobile shuffling network for acoustic scene classification[J]. IEEE Access, 2022,10: 82185-82201.Google ScholarGoogle ScholarCross RefCross Ref
  9. Chan TK, Chin CS. Multi branch convolutional macaron net for sound event detection[J]. IEEE/ACM Transactions on Audio, Speech and Language Processing, 2021, 29:2972-2985.Google ScholarGoogle ScholarDigital LibraryDigital Library
  10. Chan TK, Chin CS, Li Y. Semi-supervised NMF-CNN for sound event detection[J]. IEEE Access, 2021, 9:130529-130542.Google ScholarGoogle ScholarCross RefCross Ref
  11. Ma C, Wei S, Chen T, Integration of results from convolutional neural network in a support vector machine for the detection of atrial fibrillation [J]. IEEE Transactions on Instrumentation and Measurement, 2020, 70: 1-10.Google ScholarGoogle Scholar
  12. Clifford GD, Liu CY, Moody B, AF classification from a short single lead ECG recording: The physionet /computing in cardiology challenge 2017[C]. 2017 Computing in Cardiology (CinC) , IEEE, Rennes, France, 2017:1-4.Google ScholarGoogle ScholarCross RefCross Ref
  13. Liu FF, Liu CY, Zhao LN, An open access database for evaluating the algorithms of ECG rhythm and morphology abnormal detection[J]. Journal of Medical Imaging and Health Informatics, 2018, 8(7) :1368-1373.Google ScholarGoogle ScholarCross RefCross Ref
  14. Yu Z, Chen J, Liu Y, DDCNN: A Deep Learning Model for AF Detection From a Single-Lead Short ECG Signal [J]. IEEE Journal of Biomedical and Health Informatics, 2022, 26(10): 4987-4995.Google ScholarGoogle ScholarCross RefCross Ref
  15. Yang P, Wang D, Kang Z, Prediction model of paroxysmal atrial fibrillation based on pattern recognition and ensemble CNN-LSTM [J]. Journal of Zhejiang University(Engineering Science), 2020, 54(5):1039-1048.Google ScholarGoogle Scholar
  16. Zhao R, Xu P, Liu Y, ECG-based atrial fibrillation detection based on deep convolutional residual neural network [J]. Computer Science, 2022, 49(5):186-193.Google ScholarGoogle Scholar
  17. Wang L, Li X, Chen J, Study on atrial fibrillation prediction algorithm combining convolutional neural network and bi-directional long short-term memory network [J]. Journal of Jiangsu Practical Electrocardiology, 2022, 31(4): 256-261.Google ScholarGoogle Scholar
  18. Subramanyan L, Ganesan U. A novel deep neural network for detection of Atrial Fibrillation using ECG signals[J]. Knowledge-Based Systems, 2022, 258: 109926.Google ScholarGoogle ScholarDigital LibraryDigital Library
  19. Woo S, Park J, Lee JY, et al. CBAM: convolutional block attention module[J]. arXiv:1807.06521, 2018.Google ScholarGoogle Scholar
  20. Zhang J, Liu X, Dong J. CCDD: an enhanced standard ECG database with its management and annotation tools [J]. International Journal on Artificial Intelligence Tools, 2012, 21(5): 1240020.Google ScholarGoogle ScholarCross RefCross Ref
  21. Liu Z, Mao H, Wu C, A convnet for the 2020s[C]. Proceedings of the Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, New Orleans, USA, 2022, 11976-11986.Google ScholarGoogle ScholarCross RefCross Ref
  22. He K, Zhang X, Ren S, Deep residual learning for image recognition[C]. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, USA, 2016, 770-778.Google ScholarGoogle ScholarCross RefCross Ref
  23. Liu Z, Lin Y, Cao Y, Swin transformer: Hierarchical vision transformer using shifted windows[C]. Proceedings of the Proceedings of the IEEE/CVF international conference on computer vision, 2021. 10012-10022.Google ScholarGoogle Scholar
  24. Ba JL, Kiros JR, Hinton GE. Layer normalization[J]. arXiv preprint arXiv:160706450, 2016.Google ScholarGoogle Scholar
  25. Sifre L, Mallat S. Rigid-motion scattering for texture classification [J]. arXiv preprint arXiv:14031687, 2014.Google ScholarGoogle Scholar
  26. Imrana Y, Xiang YP, Ali L, A bidirectional LSTM deep learning approach for intrusion detection[J], Expert Systems with Applications, 2021, 185:115524.Google ScholarGoogle ScholarDigital LibraryDigital Library
  27. Gulcehre C, Moczulski M, Denil M, Noisy activation functions[C]. Proceedings of the International Conference on Machine Learning, New York, USA, 2016, 48: 3059-3068.Google ScholarGoogle Scholar
  28. Leng Z, Tan M, Liu C, Polyloss: A polynomial expansion perspective of classification loss functions [J]. arXiv preprint arXiv:220412511, 2022.Google ScholarGoogle Scholar
  29. Lin TY, Goyal P, Girshick R, Focal loss for dense object detection[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2020, 42(2):318-327.Google ScholarGoogle ScholarCross RefCross Ref
  30. Jin L, Dong J. Deep learning research on clinical electrocardiogram analysis [J]. SCIENTIA SINICA Informationis, 2015, 45(3): 398-416. . (in Chese)Google ScholarGoogle Scholar
  31. Hu J, Shen L, Albanie S, Squeeze-and-Excitation networks [J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2019, 42(8): 2011-2023.Google ScholarGoogle ScholarDigital LibraryDigital Library
  32. Jin LP, Dong J. Intelligent health vessel ABC-DE: an electrocardiogram cloud computing service [J]. IEEE transactions on cloud computing, 2020, 8(3): 861-874.Google ScholarGoogle Scholar
  33. Jo Y-Y, Cho Y, Lee SY, Explainable artificial intelligence to detect atrial fibrillation using electrocardiogram [J]. International Journal of Cardiology, 2021, 328: 104-110.Google ScholarGoogle ScholarCross RefCross Ref
  34. AI Understanding Electrocardiogram | Zhongke Maidi Artificial Intelligence Technology Exploration 12 Lead Electrocardiogram Intelligent Diagnosis [EB/OL]. http://www.medicalsystem.com.cn/news/258.html, [2023].(in Chese)Google ScholarGoogle Scholar

Index Terms

  1. Multi-lead atrial fibrillation classification method using ConvNeXt and BiLSTM

    Recommendations

    Comments

    Login options

    Check if you have access through your login credentials or your institution to get full access on this article.

    Sign in
    • Published in

      cover image ACM Other conferences
      AICCC '23: Proceedings of the 2023 6th Artificial Intelligence and Cloud Computing Conference
      December 2023
      280 pages
      ISBN:9798400716225
      DOI:10.1145/3639592

      Copyright © 2023 ACM

      Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than the author(s) must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected].

      Publisher

      Association for Computing Machinery

      New York, NY, United States

      Publication History

      • Published: 13 April 2024

      Permissions

      Request permissions about this article.

      Request Permissions

      Check for updates

      Qualifiers

      • research-article
      • Research
      • Refereed limited
    • Article Metrics

      • Downloads (Last 12 months)2
      • Downloads (Last 6 weeks)2

      Other Metrics

    PDF Format

    View or Download as a PDF file.

    PDF

    eReader

    View online with eReader.

    eReader

    HTML Format

    View this article in HTML Format .

    View HTML Format