Complementary Roles of Image Classification and Vessel Segmentation in AI-Based Screening for Retinopathy of Prematurity Plus Disease in a Kenyan Preterm Cohort

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision & Pattern Recognition, Medical Imaging AI · Depth: Expert, quick

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

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

Topics

Best for: AI Scientist, Research Scientist, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.