Papers
Topics
Authors
Recent
Search
2000 character limit reached

A Novel real-time arrhythmia detection model using YOLOv8

Published 26 May 2023 in cs.CV and cs.AI | (2305.16727v3)

Abstract: In a landscape characterized by heightened connectivity and mobility, coupled with a surge in cardiovascular ailments, the imperative to curtail healthcare expenses through remote monitoring of cardiovascular health has become more pronounced. The accurate detection and classification of cardiac arrhythmias are pivotal for diagnosing individuals with heart irregularities. This study underscores the feasibility of employing electrocardiograms (ECG) measurements in the home environment for real-time arrhythmia detection. Presenting a fresh application for arrhythmia detection, this paper leverages the cutting-edge You-Only-Look-Once (YOLO)v8 algorithm to categorize single-lead ECG signals. We introduce a novel loss-modified YOLOv8 model, fine-tuned on the MIT-BIH arrhythmia dataset, enabling real-time continuous monitoring. The obtained results substantiate the efficacy of our approach, with the model attaining an average accuracy of 99.5% and 0.992 mAP@50, and a rapid detection time of 0.002 seconds on an NVIDIA Tesla V100. Our investigation exemplifies the potential of real-time arrhythmia detection, enabling users to visually interpret the model output within the comfort of their homes. Furthermore, this study lays the groundwork for an extension into a real-time explainable AI (XAI) model capable of deployment in the healthcare sector, thereby significantly advancing the realm of healthcare solutions.

Definition Search Book Streamline Icon: https://streamlinehq.com
References (56)
  1. Centers for Disease Control and Prevention, National Center for Health Statistics. About multiple cause of death, 1999–2019, 2021. data retrieved from CDC WONDER Online Database website, https://wonder.cdc.gov/mcd-icd10.html.
  2. Automated detection of shockable ecg signals: A review. Information Sciences, 571:580–604, 2021.
  3. Classification and self-supervised regression of arrhythmic ecg signals using convolutional neural networks. arXiv preprint arXiv:2210.14253, 2022.
  4. F.J. Dowd. Arrhythmias. In Reference Module in Biomedical Sciences. Elsevier, 2014.
  5. A novel machine learning framework for automated detection of arrhythmias in ecg segments. Journal of Ambient Intelligence and Humanized Computing, pages 1–18, 2021.
  6. Arrhythmia detection using deep convolutional neural network with long duration ecg signals. Computers in biology and medicine, 102:411–420, 2018.
  7. Application of higher order cumulant features for cardiac health diagnosis using ecg signals. International journal of neural systems, 23(04):1350014, 2013.
  8. A novel application of deep learning for single-lead ecg classification. Computers in biology and medicine, 99:53–62, 2018.
  9. Multiclass arrhythmia detection and classification from photoplethysmography signals using a deep convolutional neural network. Journal of the American Heart Association, 11(7):e023555, 2022.
  10. Ecg-based heartbeat classification for arrhythmia detection: A survey. Computer methods and programs in biomedicine, 127:144–164, 2016.
  11. Kumar and clark’s clinical medicine E-Book. Elsevier Health Sciences, 2020.
  12. Leonard S Lilly. Pathophysiology of heart disease: a collaborative project of medical students and faculty. Lippincott Williams & Wilkins, 2012.
  13. Accuracy of intracoronary ecg parameters for myocardial ischemia detection. Journal of electrocardiology, 64:50–57, 2021.
  14. Myocardial infarction severity stages classification from ecg signals using attentional recurrent neural network. IEEE Sensors Journal, 20(15):8711–8720, 2020.
  15. System for neural network determination of atrial fibrillation on ecg signals with wavelet-based preprocessing. Applied Sciences, 11(16):7213, 2021.
  16. Leon Glass. Cardiac oscillations and arrhythmia analysis. Complex Systems Science in Biomedicine, pages 409–422, 2006.
  17. Application of explainable artificial intelligence for healthcare: A systematic review of the last decade (2011–2022). Computer Methods and Programs in Biomedicine, page 107161, 2022.
  18. Clustering ecg complexes using hermite functions and self-organizing maps. IEEE Transactions on Biomedical Engineering, 47(7):838–848, 2000.
  19. Comparing wavelet transforms for recognizing cardiac patterns. IEEE Engineering in Medicine and Biology Magazine, 14(2):167–173, 1995.
  20. A patient-adaptable ecg beat classifier using a mixture of experts approach. IEEE transactions on biomedical engineering, 44(9):891–900, 1997.
  21. Support vector machine for arrhythmia discrimination with wavelet transform-based feature selection. In Computers in Cardiology 2000. Vol. 27 (Cat. 00CH37163), pages 407–410. IEEE, 2000.
  22. Classification of premature ventricular contraction in ecg. International Journal of Advanced Computer Science and Applications, 6(7), 2015.
  23. Premature ventricular contraction classification by the kth nearest-neighbours rule. Physiological measurement, 26(1):123, 2005.
  24. Automatic classification of cardiac arrhythmias based on hybrid features and decision tree algorithm. International Journal of Automation and Computing, 17(4):551–561, 2020.
  25. Deep learning. nature, 521(7553):436–444, 2015.
  26. Automated diagnosis of arrhythmia using combination of cnn and lstm techniques with variable length heart beats. Computers in biology and medicine, 102:278–287, 2018.
  27. A deep convolutional neural network model to classify heartbeats. Computers in biology and medicine, 89:389–396, 2017.
  28. A review on deep learning methods for ecg arrhythmia classification. Expert Systems with Applications: X, 7:100033, 2020.
  29. Electrocardiogram classification based on faster regions with convolutional neural network. Sensors, 19(11):2558, 2019.
  30. Automatic detection of arrhythmias using a yolo-based network with long-duration ecg signals. Engineering Proceedings, 2(1):84, 2020.
  31. A review of object detection based on deep learning. Multimedia Tools and Applications, 79:23729–23791, 2020.
  32. Physionet: a web-based resource for the study of physiologic signals. IEEE Engineering in Medicine and Biology Magazine, 20(3):70–75, 2001.
  33. A framework for cardiac arrhythmia detection from iot-based ecgs. World Wide Web, 23:2835–2850, 2020.
  34. AAMI ECAR. Recommended practice for testing and reporting performance results of ventricular arrhythmia detection algorithms. Association for the Advancement of Medical Instrumentation, 69, 1987.
  35. The impact of the mit-bih arrhythmia database. IEEE engineering in medicine and biology magazine, 20(3):45–50, 2001.
  36. Ecg heartbeat classification: A deep transferable representation. In 2018 IEEE international conference on healthcare informatics (ICHI), pages 443–444. IEEE, 2018.
  37. Learning data augmentation strategies for object detection. In Computer Vision–ECCV 2020: 16th European Conference, Glasgow, UK, August 23–28, 2020, Proceedings, Part XXVII 16, pages 566–583. Springer, 2020.
  38. You only look once: Unified, real-time object detection. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 779–788, 2016.
  39. A comprehensive review of yolo: From yolov1 to yolov8 and beyond. arXiv preprint arXiv:2304.00501, 2023.
  40. Object detection using yolo: Challenges, architectural successors, datasets and applications. Multimedia Tools and Applications, pages 1–33, 2022.
  41. YOLO by Ultralytics, January 2023.
  42. Fracture detection in pediatric wrist trauma x-ray images using yolov8 algorithm. arXiv preprint arXiv:2304.05071, 2023.
  43. YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors. arXiv preprint arXiv:2207.02696, 2022.
  44. Path aggregation network for instance segmentation. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 8759–8768, 2018.
  45. Feature pyramid networks for object detection. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 2117–2125, 2017.
  46. Generalized focal loss: Learning qualified and distributed bounding boxes for dense object detection. Advances in Neural Information Processing Systems, 33:21002–21012, 2020.
  47. Distance-iou loss: Faster and better learning for bounding box regression. In Proceedings of the AAAI conference on artificial intelligence, volume 34, pages 12993–13000, 2020.
  48. Enhancing geometric factors in model learning and inference for object detection and instance segmentation. IEEE transactions on cybernetics, 52(8):8574–8586, 2021.
  49. Wise-iou: Bounding box regression loss with dynamic focusing mechanism. arXiv preprint arXiv:2301.10051, 2023.
  50. Resolving class imbalance in object detection with weighted cross entropy losses. arXiv preprint arXiv:2006.01413, 2020.
  51. Real-time patient-specific ecg classification by 1-d convolutional neural networks. IEEE Transactions on Biomedical Engineering, 63(3):664–675, 2015.
  52. Automated ecg classification using dual heartbeat coupling based on convolutional neural network. IEEE Access, 6:27465–27472, 2018.
  53. Anitha S Prasad and N Kavanashree. Ecg monitoring system using ad8232 sensor. In 2019 International Conference on Communication and Electronics Systems (ICCES), pages 976–980. IEEE, 2019.
  54. Explainable artificial intelligence (xai): Concepts, taxonomies, opportunities and challenges toward responsible ai. Information fusion, 58:82–115, 2020.
  55. Heatmaps for visual explainability of cnn-based predictions for multivariate time series with application to healthcare. In 2020 IEEE International Conference on Healthcare Informatics (ICHI), pages 1–8. IEEE, 2020.
  56. Edge device detection of tea leaves with one bud and two leaves based on shufflenetv2-yolov5-lite-e. Agronomy, 13(2):577, 2023.
Citations (4)

Summary

No one has generated a summary of this paper yet.

Paper to Video (Beta)

No one has generated a video about this paper yet.

Whiteboard

No one has generated a whiteboard explanation for this paper yet.

Open Problems

We haven't generated a list of open problems mentioned in this paper yet.

Continue Learning

We haven't generated follow-up questions for this paper yet.

Collections

Sign up for free to add this paper to one or more collections.