VascX Models: Model Ensembles for Retinal Vascular Analysis from Color Fundus Images

TL;DR

VascX models are an ensemble of retinal vasculature analysis models that outperform existing methods in segmentation accuracy across diverse datasets, improving disease feature extraction from color fundus images.

Contribution

The paper introduces VascX, a new set of model ensembles trained on a large, diverse dataset, significantly enhancing retinal vessel segmentation accuracy.

Findings

Superior segmentation performance across datasets and image qualities.

Improved artery-vein and disc segmentation accuracy.

More precise vascular feature extraction for disease analysis.

Abstract

We introduce VascX models, a comprehensive set of model ensembles for analyzing retinal vasculature from color fundus images (CFIs). Annotated CFIs were aggregated from public datasets . Additional CFIs, mainly from the population-based Rotterdam Study were annotated by graders for arteries and veins at pixel level, resulting in a dataset diverse in patient demographics and imaging conditions. VascX models demonstrated superior segmentation performance across datasets, image quality levels, and anatomic regions when compared to existing, publicly available models, likely due to the increased size and variety of our training set. Important improvements were observed in artery-vein and disc segmentation performance, particularly in segmentations of these structures on CFIs of intermediate quality, common in large cohorts and clinical datasets. Importantly, these improvements translated into significantly more accurate vascular features when we compared features extracted from VascX segmentation masks with features extracted from segmentation masks generated by previous models. With VascX models we provide a robust, ready-to-use set of model ensembles and inference code aimed at simplifying the implementation and enhancing the quality of automated retinal vasculature analyses. The precise vessel parameters generated by the model can serve as starting points for the identification of disease patterns in and outside of the eye.