轻量级卷积网络融合提升糖尿病视网膜病变筛查准确率与效率
Balancing Accuracy and Efficiency: CNN Fusion Models for Diabetic Retinopathy Screening
- 通过融合ResNet50、EfficientNet-B0和DenseNet121的特征,提升模型泛化能力
- Eff+Den融合模型在11,156张图像上达82.89%准确率,正常与患病类F1分数均衡
- 该方法适合对速度与精度均有要求的全球性大规模糖尿病眼底筛查
糖尿病视网膜病变(DR)是可预防失明的主要原因,但大规模筛查受限于专科医生不足及设备与人群间图像质量差异。本文研究基于互补卷积神经网络(CNN)骨干的特征级融合,能否在来自五个公开数据集(APTOS、EyePACS、IDRiD、Messidor、ODIR)共11,156张眼底图像上实现高效且准确的二分类DR筛查。将ResNet50、EfficientNet-B0、DenseNet121三个预训练模型与两两及三重融合变体进行对比。五次独立实验表明,融合模型始终优于单个骨干。其中Eff+Den融合模型平均表现最佳(准确率82.89%),正常(83.60%)与糖尿病(82.60%)类别的F1分数均衡。尽管三重融合表现接近,但计算成本显著升高。推理分析显示:EfficientNet-B0最快(批量1000时约1.16毫秒/图像),而Eff+Den融合在准确率与延迟间取得良好平衡。结果表明,轻量级特征融合可增强跨异构数据集的泛化能力,支持对精度与吞吐量均敏感的大规模二分类筛查流程。
原文摘要 · Abstract (English)
Diabetic retinopathy (DR) remains a leading cause of preventable blindness, yet large-scale screening is constrained by limited specialist availability and variable image quality across devices and populations. This work investigates whether feature-level fusion of complementary convolutional neural network (CNN) backbones can deliver accurate and efficient binary DR screening on globally sourced fundus images. Using 11,156 images pooled from five public datasets (APTOS, EyePACS, IDRiD, Messidor, and ODIR), we frame DR detection as a binary classification task and compare three pretrained models (ResNet50, EfficientNet-B0, and DenseNet121) against pairwise and tri-fusion variants. Across five independent runs, fusion consistently outperforms single backbones. The EfficientNet-B0 + DenseNet121 (Eff+Den) fusion model achieves the best overall mean performance (accuracy: 82.89\%) with balanced class-wise F1-scores for normal (83.60\%) and diabetic (82.60\%) cases. While the tri-fusion is competitive, it incurs a substantially higher computational cost. Inference profiling highlights a practical trade-off: EfficientNet-B0 is the fastest (approximately 1.16 ms/image at batch size 1000), whereas the Eff+Den fusion offers a favorable accuracy--latency balance. These findings indicate that lightweight feature fusion can enhance generalization across heterogeneous datasets, supporting scalable binary DR screening workflows where both accuracy and throughput are critical.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。