PRISM-DR: Per-lesion Retinal Inference with Specialist Models for Diabetic Retinopathy
Not on the map. Why?
Abstract (arXiv)
Diabetic retinopathy is a leading cause of preventable blindness; its early lesions are small, low contrast, and easily missed in manual screening. Most automated detectors handle the four non-proliferative DR lesions: microaneurysms, hemorrhages, hard exudates, and soft exudates, with a single multi-class model, even though these lesions differ sharply in size, color, morphology, and prevalence, so a shared model favors common, easy classes over rare, difficult ones. We present PRISM-DR, a lesion-specific pipeline that trains one single-class detector per lesion, each with its own configuration. From a raw fundus image, the pipeline applies region of interest cropping, fundus-specific preprocessing, four parallel YOLO detectors, tiling, per-lesion ensembling of five cross-validation folds, and an inter-lesion suppression step that resolves overlaps by physical lesion size and clinical priority rather than confidence. Per lesion, the best of five YOLO generations is selected, and augmentation is tuned by Bayesian optimization. Trained on IDRiD with stratified five-fold cross-validation, the system reaches a test mAP50 of 0.527 and F1 of 0.529, highest AP50 on hard exudates with 0.561. Without fine-tuning, the models transfer well where the imaging scale is close to IDRiD and degrade as field of view and resolution depart. These modest absolute results reflect a small single-source training set and a difficult task; however, treating each lesion as a separate detection problem is a practical alternative to a single multi-class model.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 14 of its 71 references that are in the map. First posted on arXiv 22 Jul 2026.
Builds on
14 of 71 references.
References · 14
- 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,501
- 2022Slicing Aided Hyper Inference and Fine-tuning for Small Object Detectioncited by 105
- 2020A Benchmark for Studying Diabetic Retinopathy: Segmentation, Grading, and Transferabilitycited by 49
- 2017Grader variability and the importance of reference standards for evaluating machine learning models for diabetic retinopathycited by 44
- 2019CANet: Cross-disease Attention Network for Joint Diabetic Retinopathy and Diabetic Macular Edema Gradingcited by 35
- 2019DRvertGRADUATE: uncertainty-aware deep learning-based diabetic retinopathy grading in eye fundus imagescited by 25
- 2018Deep Learning based Computer-Aided Diagnosis Systems for Diabetic Retinopathy: A Surveycited by 19
- 2019Deep Learning Fundus Image Analysis for Diabetic Retinopathy and Macular Edema Gradingcited by 12
- 2017Microaneurysm Detection in Fundus Images Using a Two-step Convolutional Neural Networkscited by 5
- 2021BezierSeg: Parametric Shape Representation for Fast Object Segmentation in Medical Imagescited by 4
- 2021DRDrV3: Complete Lesion Detection in Fundus Images Using Mask R-CNN, Transfer Learning, and LSTMcited by 4
- 2023TJDR: A High-Quality Diabetic Retinopathy Pixel-Level Annotation Datasetcited by 3
Built on by
Not cited yet.