Coronary Artery Extraction from CT Coronary Angiography with Augmentation on Partially Labelled Data

Coronary artery disease (CAD) is an important cause of morbidity and mortality. CT coronary angiography is considered as first-line of investigation in patients suspected of having CAD. Coronary artery centerline extraction is a challenging prerequisite for coronary artery stenosis evaluation. These challenges include the small and complex structure, variation of plaques and imaging noise. Deep learning methods often require adequate annotated data to build a good model. This work aims to adopt a dataset that has partial annotation to augment the data to train a Coronary Neural Network (CorNN) to track the coronary artery centerline. We combined a small training dataset with densely labelled centerline and radius, augmented with a larger dataset with only the centerline sparsely labelled to train networks to track centerlines from 3D computed tomography coronary angiography. The vessel orientation estimation is patch based, with or without additional radius prediction. The patch data are carefully positioned and sampled, which are tagged with the orientations computed based on the centerlines. Our experiment results show that, with the augmentation of the new data, although partially annotated, nearly 10% or more improvement has been achieved for the coronary artery extraction by the proposed approach.