Ensembled-SAMs for Enhanced Small Coronary Artery Segmentation in CCTA Images

Accurate coronary artery segmentation is crucial for quantitative analysis of coronary arteries in noninvasive coronary computed tomography angiography (CCTA) images. However, current segmentation algorithms often have unsatisfactory recall due to the small size and complex morphology of coronary arteries, particularly in the distal segments. To address this issue, we introduce a new fully automated method named Ensembled-SAMs, which harnesses the strengths of the Segment Anything Model (SAM) and the no-new-U-Net (nnU-Net). First, noisy bounding box prompts are automatically generated by a vesselness algorithm that highlights the tubular structures in the CCTA images. These noisy prompts are then used to fine-tune the SAM and its two variants separately. The SAM variants introduce a classification head in their mask decoder to alleviate the false positives. In addition, an nnU-Net segmentation network is trained from scratch. Finally, the outputs of the SAMs and the nnU-Net are strategically aggregated to obtain the final segmentation result. Experiments on both a self-built dataset and the public Automated Segmentation of Coronary Arteries (ASOCA) challenge dataset demonstrate that the proposed Ensembled-SAMs outperforms the state-of-the-arts, achieving precise segmentation of coronary arteries, with particular enhancement in delineating small coronary artery segments.