用对抗模糊提升糖尿病视网膜病变分类的泛化能力
AdvBlur: Adversarial Blur for Robust Diabetic Retinopathy Classification and Cross-Domain Generalization
- 在数据集加入对抗模糊图像,增强模型对设备差异的鲁棒性
- 在多个外部数据集上实现优于现有方法的跨域分类性能
- 适合关注医学图像泛化、真实场景部署的研究者
糖尿病视网膜病变(DR)是全球视力丧失的主要原因,早期准确检测可显著改善治疗效果。尽管已有大量深度学习模型用于从眼底图像中预测DR,但其性能常因成像设备、人群差异和拍摄条件不同导致的分布偏移而下降。本文提出一种名为AdvBlur的新方法,通过将对抗模糊图像引入训练集,并采用双损失函数框架,有效缓解未见分布变化的影响。在多个数据集上的综合评估表明,该方法在跨域泛化方面表现优异。我们还系统研究了相机类型、低质量图像及数据集规模的影响,并通过消融实验验证了模糊图像与损失函数设计的有效性。实验结果证明,该方法在未见外部数据集上达到与当前最优域泛化模型相当的性能。
原文摘要 · Abstract (English)
Diabetic retinopathy (DR) is a leading cause of vision loss worldwide, yet early and accurate detection can significantly improve treatment outcomes. While numerous Deep learning (DL) models have been developed to predict DR from fundus images, many face challenges in maintaining robustness due to distributional variations caused by differences in acquisition devices, demographic disparities, and imaging conditions. This paper addresses this critical limitation by proposing a novel DR classification approach, a method called AdvBlur. Our method integrates adversarial blurred images into the dataset and employs a dual-loss function framework to address domain generalization. This approach effectively mitigates the impact of unseen distributional variations, as evidenced by comprehensive evaluations across multiple datasets. Additionally, we conduct extensive experiments to explore the effects of factors such as camera type, low-quality images, and dataset size. Furthermore, we perform ablation studies on blurred images and the loss function to ensure the validity of our choices. The experimental results demonstrate the effectiveness of our proposed method, achieving competitive performance compared to state-of-the-art domain generalization DR models on unseen external datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。