Comparative study of deep learning models for automatic coronary stenosis detection in x-ray angiography

Viacheslav Danilov, Olga Gerget, Kirill Klyshnikov, Evgeny Ovcharenko, Alejandro Frangi

Research output: Contribution to journalConference articlepeer-review

Abstract

The article explores the application of machine learning approach to detect both single-vessel and multivessel coronary artery disease from X-ray angiography. Since the interpretation of coronary angiography images requires interventional cardiologists to have considerable training, our study is aimed at analysing, training, and assessing the potential of the existing object detectors for classifying and detecting coronary artery stenosis using angiographic imaging series. 100 patients who underwent coronary angiography at the Research Institute for Complex Issues of Cardiovascular Diseases were retrospectively enrolled in the study. To automate the medical data analysis, we examined and compared three models (SSD MobileNet V1, Faster-RCNN ResNet-50 V1, Faster-RCNN NASNet) with various architecture, network complexity, and a number of weights. To compare developed deep learning models, we used the mean Average Precision (mAP) metric, training time, and inference time. Testing results show that the training/inference time is directly proportional to the model complexity. Thus, Faster-RCNN NASNet demonstrates the slowest inference time. Its mean inference time per one image made up 880 ms. In terms of accuracy, Faster-RCNN ResNet-50 V1 demonstrates the highest prediction accuracy. This model has reached the mAP metric of 0.92 on the validation dataset. SSD MobileNet V1 has demonstrated the best inference time with the inference rate of 23 frames per second.

Original languageEnglish
JournalCEUR Workshop Proceedings
Volume2744
Publication statusPublished - 2020
Event30th International Conference on Computer Graphics and Machine Vision, GraphiCon 2020 - Saint Petersburg, Russian Federation
Duration: 22 Sep 202025 Sep 2020

Keywords

  • Deep Learning
  • Stenosis Detection
  • Transfer Learning
  • X-ray Angiography

ASJC Scopus subject areas

  • Computer Science(all)

Fingerprint Dive into the research topics of 'Comparative study of deep learning models for automatic coronary stenosis detection in x-ray angiography'. Together they form a unique fingerprint.

Cite this