arXiv:2607.19864eess.IVcs.AI2026-07

针对糖尿病视网膜病变,为每类病灶单独训练检测模型,提升小病灶识别率。

PRISM-DR: Per-lesion Retinal Inference with Specialist Models for Diabetic Retinopathy

论文配图:PRISM-DR: Per-lesion Retinal Inference with Specialist Models for Diabetic Retinopathy
图 1 · 摘自论文原文
  • 每类病灶用独立单类别检测器,配置自适应优化
  • 在IDRiD数据集上测试mAP50达0.527,硬性渗出物达0.561
  • 适合医疗影像中多类差异大、难统一建模的场景

糖尿病视网膜病变是导致可预防失明的主要原因;其早期病灶微小、对比度低,人工筛查易漏检。现有自动检测多采用单一多类模型处理四类非增殖性病变(微动脉瘤、出血、硬性渗出、软性渗出),但这些病灶在大小、颜色、形态和出现频率上差异显著,共享模型偏向常见易检类别而忽略稀有难检类。本文提出PRISM-DR,一种病灶特异性检测流程:为每类病灶训练独立单类别检测器,各自配置最优参数。从原始眼底图像出发,流程包括感兴趣区域裁剪、眼底专用预处理、四路并行YOLO检测器、图像分块、每类病灶五折交叉验证集成、基于病灶物理尺寸与临床优先级的跨病灶抑制。每类选取最佳五个生成版本之一,数据增强通过贝叶斯优化调优。在IDRiD数据集上进行分层五折交叉验证训练,系统测试mAP50达0.527,F1为0.529,硬性渗出物最高AP50达0.561。未微调时模型在成像尺度接近IDRiD时迁移良好,随视野与分辨率偏离性能下降。该结果虽不高,反映单一来源训练集有限及任务难度,但将每类病灶视为独立检测问题,是替代单一多类模型的实用方案。

原文摘要 · Abstract (English)

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.

医学影像病灶检测YOLO糖尿病视网膜病变

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。