Complementary Roles of Image Classification and Vessel Segmentation in AI-Based Screening for Retinopathy of Prematurity Plus Disease in a Kenyan Preterm Cohort
Summary
An AI-based screening system for Retinopathy of Prematurity (ROP) Plus disease was developed and evaluated using a cohort of 121 Kenyan preterm infants, encompassing 237 eyes and 1,635 fundus images. This study investigated the complementary roles of image classification and vessel segmentation for detecting Plus disease, a subjective condition requiring treatment. Eleven distinct AI configurations, including image classifiers, multiple-instance learning, and multi-task segmentation-classification pipelines, were assessed. While RGB classifiers demonstrated high sensitivity, they also led to over-referral. Vessel segmentation proved feasible, achieving a Dice score of 0.533 and specificity of 0.979. Combining these approaches significantly enhanced performance, with a probability ensemble model yielding the best balanced results: 0.692 sensitivity, 0.914 specificity, and 0.803 balanced accuracy, surpassing standalone vision classifiers.
Key takeaway
For AI Scientists developing diagnostic systems for medical imaging in resource-limited regions, you should integrate both image classification and vessel segmentation into your workflows. This combined approach, particularly using a probability ensemble, offers superior balanced performance (0.692 sensitivity, 0.914 specificity) compared to single-method classifiers, effectively reducing over-referral while maintaining high sensitivity. Prioritize multi-site validation for robust deployment.
Key insights
Combining image classification and vessel segmentation improves AI-based ROP Plus detection by balancing sensitivity and specificity.
Principles
- Classifiers excel at sensitive case-finding.
- Segmentation boosts specificity, reducing over-referral.
- Combined AI workflows yield balanced performance.
Method
Evaluated 11 AI configurations for ROP Plus detection, including image classifiers, multiple-instance learning, multi-task segmentation-classification, and segment-then-classify pipelines, using patient-grouped nested cross-validation.
In practice
- Employ OR-based screening for maximum sensitivity.
- Apply AND-based confirmation for highest specificity.
- Use probability ensembles for balanced detection.
Topics
- Retinopathy of Prematurity
- Plus Disease
- Image Classification
- Vessel Segmentation
- AI Screening
- Medical Imaging
Best for: AI Scientist, Research Scientist, Computer Vision Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.