Epidemiology of Retinopathy of Prematurity in the US From 2003 to 2019.

PURPOSE: To present and validate deep learning model ensembles (VascX) for vessel, artery-vein, optic disc segmentation, and fovea localization for color fundus images (CFIs). VascX preprocessing and inference code and model weights were made publicly available to facilitate research on retinal vasculature.

METHODS

For model training, we combined over 15 published annotated datasets with CFIs from Dutch studies (mainly the Rotterdam Study). This resulted in diverse development sets with a variety of patient characteristics and imaging conditions. We trained UNet model ensembles using a new, more robust preprocessing algorithm and strong data augmentations. We compared VascX segmentation performance (Dice) to models with publicly available weights: AutoMorph and LittleWNet. We compared the quality of VascX (and previous models') features by measuring agreement (mean absolute error [MAE] and Pearson correlation) with features extracted from grader segmentations.

RESULTS

Dice scores revealed better performance from VascX across most datasets evaluated, especially for artery-vein and optic disc segmentation. VascX performed more consistently as the quality of the images decreased and for both disc and fovea-centered images. These improvements translated into higher-quality vascular features. Of 24 features evaluated, 14 showed a significant improvement in MAE when compared to AutoMorph and 23 when compared to LWNet. VascX had the highest correlations with ground-truth features in all but two cases.

CONCLUSIONS

VascX models perform well across a variety of conditions, likely due to the size and diversity of our development sets. VascX represents an important improvement in segmentation quality that translates into better vascular features to support more robust analyses of the retinal vasculature.

TRANSLATIONAL RELEVANCE: By making VascX public, we aim to facilitate and improve research linking retinal vascular biomarkers to ophthalmic and systemic conditions, relevant for the detection, prevention, and monitoring of disease.