A Computationally Efficient Approach to Segmentation of the Aorta and Coronary Arteries Using Deep Learning

Abstract

Early detection and diagnosis of coronary artery disease could reduce the risk of developing a heart attack. The coronary arteries are optimally visualised using computed tomography coronary angiography (CTCA) imaging. These images are reviewed by specialist radiologists who evaluate the coronary arteries for potential narrowing. A lack of radiologists in the UK is a constraint to timely diagnosis of coronary artery disease, particularly in the acute accident and emergency department setting. The development of automated methods by which coronary artery narrowing can be identified rapidly and accurately are therefore timely. Such complex computer based tools also need to be sufficiently computationally efficient that they can run on servers typically found in hospital settings, where graphical processing units for example are unavailable. We propose a fully automatic two-dimensional Unet model to segment the aorta and coronary arteries on CTCA images. Two models are trained to segment two regions of interest, (1) the aorta and the coronary arteries or (2) the coronary arteries alone. Our method achieves 91.20% and 88.80% dice similarity coefficient accuracy on regions of interest 1 and 2 respectively. Compared with a semi-automatic segmentation method, our model performs better when segmenting the coronary arteries alone. The performance of the proposed method is comparable to existing published two-dimensional or three-dimensional deep learning models. Importantly, the algorithmic and graphical processing unit memory efficiencies are maintained such that the model can be deployed without requiring graphical processing units, and therefore can be used in a hospital setting.