用自监督学习让少量眼底图也能学出好特征,减少标注依赖。
Domain-Specific Self-Supervised Representation Learning for Retinal Fundus Classification

- 用小批量训练+眼底图像特有增强,提升无监督表征质量。
- 在糖尿病视网膜病变分级任务中表现接近有监督方法。
- 适合数据少、算力有限的医疗图像场景,如基层医院应用。
尽管公开数据集增多,医学图像标注仍稀缺。监督学习虽性能强,但需大量标注数据,获取成本高。为此,对比自监督学习(SSL)成为从无标签数据中学习有效表征的可行方案。本文研究了SimSiam与SimCLR两种SSL框架在眼底图像疾病分类中的应用,重点分析增强策略和训练参数对资源受限条件下表征学习的影响。在数据和算力有限的情况下,探索使用小批量训练结合眼底图像特有增强技术的可行性。通过线性评估与微调,在多病种分类和糖尿病视网膜病变分级等下游任务上验证表征质量。结果表明,针对眼底图像特点设计增强策略显著提升性能。即使在受限条件下,轻量级SSL框架仍可学习可迁移表征,降低对大规模标注数据的依赖,并取得具有竞争力的结果。
原文摘要 · Abstract (English)
Despite the growing number of public datasets, annotated medical images remain scarce. Supervised learning methods achieve strong performance on many benchmarks, however require large amounts of labeled data, which are costly and time-consuming to obtain in the medical domain. To address this limitation, contrastive self-supervised learning (SSL) has emerged as a promising alternative for learning useful representations from unlabeled data. In this work, we investigate two SSL frameworks, SimSiam and SimCLR, for retinal disease classification from fundus images. We focus on understanding how augmentation strategies and training parameters influence representation learning under resource-constrained settings. Given limited data and computational capacity, we explore the feasibility of training SSL models with small batch sizes incorporated with retinal-specific augmentation techniques. Through a series of experiments, we assess the quality of learned representations via linear evaluation and fine-tuning across downstream tasks, including multi-disease classification and diabetic retinopathy grading. Our results show that tailoring augmentation strategies to the characteristics of retinal images plays a critical role in improving performance. Even under constrained settings, lightweight SSL frameworks can learn transferable representations that reduce dependence on large annotated datasets and achieve competitive results.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。