Automatic Location and Classification of Coronary Artery Stenosis Based on Deep Neural Network
Cong Chao1,2, Xiao Zhaohui3, Chen Wenjun2, Wang Yi1*
1(Army Medical Center of PLA, Army Medical University, Chongqing 400016, China) 2(School of Electrical and Electronic Engineering, Chongqing University of Technology, Chongqing 400054, China) 3(College of Computer Science and Engineering, Chongqing University of Technology, Chongqing 400054, China)
Abstract:In this paper, a deep neural network-based workflow was proposed to automatically detect and classify the stenosis features in coronary images. The algorithm mainly used quantitative coronary angiography (QCA) as a label for supervised learning and classifies the severity of coronary stenosis into normal (< 25% stenosis score) and stenosis (> 25% stenosis) categories and realized stenosis location detection in images. The algorithm used the inception model as the basic classifier to preliminarily classify the image level stenosis, and then combined with the multi-level pool structure to jointly predict the multi perspective angiography image to obtain the left-/right-artery/patient level stenosis prediction. On the basis of the classifier, the feature was further extracted, and the unsupervised learning model was used to realize the narrow location in the image.The training and cross validation were performed on a total of 10872 images in 235 clinical studies. The results showed that the algorithm achieved 85% accuracy and 0.91 AUC score in image level stenosis classification; in multi view joint prediction experiment, the sensitivity and AUC score of 0.94/0.90/0.96 and 0.87/0.88/0.86 respectively for left-/ right-/patient level stenosis classification prediction. In the stenosis localization experiment, the sensitivity of detection for left-/right-artery stenosis was 0.70/0.68, and the mean square error of 512 × 512 image was 37.6/39.3 pixels, respectively. In conclusion, the proposed method realized the potential of auxiliary diagnosis prediction from image to patient with high accuracy, which not only provided the preliminary screening ability in the process of coronary angiography, but also laid the foundation for more accurate and automatic computer-aided diagnosis.
丛超, 肖朝晖, 陈文俊, 王毅. 基于深度神经网络的冠脉造影图像的血管狭窄自动定位及分类预测[J]. 中国生物医学工程学报, 2021, 40(3): 301-309.
Cong Chao, Xiao Zhaohui, Chen Wenjun, Wang Yi. Automatic Location and Classification of Coronary Artery Stenosis Based on Deep Neural Network. Chinese Journal of Biomedical Engineering, 2021, 40(3): 301-309.
[1] Virani SS, Alonso A, Benjamin EJ, et al. Heart disease and stroke statistics—2020 update: A report from the American Heart Association[J]. Circulation, 2020, 141(9): e139-e596. [2] Grundeken MJ, Ishibashi Y, Généreux P, et al. Inter-core lab variability in analyzing quantitative coronary angiography for bifurcation lesions: A post-hoc analysis of a randomized trial [J]. JACC: Cardiovascular Interventions, 2015, 8(2): 305-314. [3] Delibasis KK, Kechriniotis AI, Tsonos C, et al. Automatic model-based tracing algorithm for vessel segmentation and diameter estimation[J]. Computer Methods and Programs in Biomedicine, 2010, 100(2): 108-122. [4] Compas CB, Syeda-Mahmood T, McNeillie P, et al. Automatic detection of coronary stenosis in x-ray angiography through spatio-temporal tracking[C]//2014 IEEE 11th international symposium on biomedical imaging (ISBI). Beijing: IEEE, 2014: 1299-1302. [5] Wan T, Feng H, Tong C, et al. Automated identification and grading of coronary artery stenoses with X-ray angiography[J]. Computer Methods and Programs in Biomedicine, 2018, 167: 13-22. [6] Fatemi MJR, Mirhassani SM, Ghasemi E. Detection of narrowed coronary arteries in X-ray angiographic images using contour processing of segmented heart vessels based on hessian vesselness filter and wavelet based image fusion [J]. International Journal of Computer Applications, 2011, 36(9): 27-33. [7] Hernandez-Vela A, Gatta C, Escalera S, et al. Accurate coronary centerline extraction, caliber estimation, and catheter detection in angiographies [J]. IEEE Transactions on Information Technology in Biomedicine, 2012, 16(6): 1332-1340. [8] Brieva J, Galvez M, Toumoulin C. Coronary extraction and stenosis quantification in X-ray angiographic imaging[C]// The 26th Annual International Conference of the IEEE Engineering in Medicine and Biology Society. San Francisco: IEEE, 2004, 1: 1714-1717. [9] Nasr-Esfahani E, Samavi S, Karimi N, et al. Vessel extraction in X-ray angiograms using deep learning[C]// 2016 38th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC). Orlando: IEEE, 2016: 643-646. [10] Nasr-Esfahani E, Karimi N, Jafari MH, et al. Segmentation of vessels in angiograms using convolutional neural networks[J]. Biomedical Signal Processing and Control, 2018, 40: 240-251. [11] Jo K, Kweon J, Kim Y H, et al. Segmentation of the main vessel of the left anterior descending artery using selective feature mapping in coronary angiography[J]. IEEE Access, 2018, 7: 919-930. [12] Zreik M, Van Hamersvelt RW, Wolterink JM, et al. A recurrent CNN for automatic detection and classification of coronary artery plaque and stenosis in coronary CT angiography[J]. IEEETransactions on Medical Imaging, 2018, 38(7): 1588-1598. [13] Wolterink JM, Leiner T, Viergever MA, et al. Automatic coronary calcium scoring in cardiac CT angiography using convolutional neural networks[C]// International Conference on Medical Image Computing and Computer-Assisted Intervention. Munich: Springer, 2015: 589-596. [14] Zreik M, Lessmann N, van Hamersvelt RW, et al. Deep learning analysis of the myocardium in coronary CT angiography for identification of patients with functionally significant coronary artery stenosis[J]. Medical Image Analysis, 2018, 44: 72-85. [15] Tesche C, De Cecco CN, Albrecht MH, et al. Coronary CT angiography-derived fractional flow reserve[J]. Radiology, 2017, 285(1): 17-33. [16] Bejnordi BE, Veta M, Van Diest PJ, et al. Diagnostic assessment of deep learning algorithms for detection of lymph node metastases in women with breast cancer[J]. JAMA, 2017, 318(22): 2199-2210. [17] Armato III SG, Drukker K, Li F, et al. LUNGx Challenge for computerized lung nodule classification[J]. Journal of Medical Imaging, 2016, 3(4): 044506. [18] Pratt H, Coenen F, Broadbent DM, et al. Convolutional neural networks for diabetic retinopathy[J]. Procedia Computer Science, 2016, 90: 200-205. [19] George RT, Arbab-Zadeh A, Cerci RJ, et al. Diagnostic performance of combined noninvasive coronary angiography and myocardial perfusion imaging using 320-MDCT: The CT angiography and perfusion methods of the CORE320 multicenter multinational diagnostic study[J]. American Journal of Roentgenology, 2011, 197(4): 829-837. [20] Miller JM, Rochitte CE, Dewey M, et al. Diagnostic performance of coronary angiography by 64-row CT [J]. New England Journal of Medicine, 2008, 359(22): 2324-2336. [21] Alderman EL, Stadius M. The angiographie definitions of the bypass angioplasty revascularization investigation[J]. Coronary Artery Disease, 1992, 3(12): 1189-1208. [22] Garcia JA, Movassaghi B, Casserly IP, et al. Determination of optimal viewing regions for X-ray coronary angiography based on a quantitative analysis of 3D reconstructed models[J]. The International Journal of Cardiovascular Imaging, 2009, 25(5): 455-462. [23] Szegedy C, Vanhoucke V, Ioffe S, et al. Rethinking the inception architecture for computer vision[C]// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas: IEEE, 2016: 2818-2826. [24] Goyal P, Kaiming H. Focal loss for dense object detection[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2018, 39: 2999-3007. [25] Ranschaert ER, Sergey M, Paul RA, et al. Artificial Intelligence in Medical Imaging: Opportunities, Applications and Risks[M]. Switzerland: Springer, 2019. [26] Litjens G, Kooi T, Bejnordi BE, et al. A survey on deep learning in medical image analysis[J]. Medical Image Analysis, 2017, 42: 60-88. [27] Zhou B, Khosla A, Lapedriza A, et al. Learning deep features for discriminative localization[C]// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas: IEEE, 2016: 2921-2929. [28] Lin TY, Maire M, Belongie S, et al. MicrosoftCOCO: Common objects in context[C]// European Conference on Computer Vision. Zurich: Springer, 2014: 740-755. [29] Au B, Shaham U, Dhruva S, et al. Automated characterization of stenosis in invasive coronary angiography images with convolutional neural networks[EL/OB]. https://arxiv.org/ftp/arxiv/papers/1807/1807.10597.pdf, 2018-06/2020-06-23. [30] Cong C, Kato Y, Vasconcellos HD, et al. Automated stenosis detection and classification in X-ray angiography using deep neural network[C]// 2019 IEEE International Conference on Bioinformatics and Biomedicine (BIBM). San Diego: IEEE, 2019: 1301-1308.