基于迁移学习的糖尿病视网膜病变辅助诊断算法

摘要：目的研究基于迁移学习的糖尿病视网膜病变(DR)诊断算法在小样本训练数据集中的应用。方法采用广东省肇庆市高要区人民医院拍摄的4465幅彩色眼底照片作为完整数据集。使用固定预训练参数和微调预训练参数的模型训练策略作为迁移学习组,将其与非迁移学习的随机初始化参数的策略对比,并将这3种策略应用在ResNet50、Inception V3和NASNet 3种深度学习网络的训练上。此外,从完整数据集中随机划分出小样本数据集,研究训练数据的减少对不同训练策略的影响。采用诊断模型的准确率和训练时间分析不同训练策略的效果。结果取不同网络架构中的最优结果。微调预训练参数策略取得的模型准确率为90.9%,高于固定预训练参数策略的88.1%及随机初始化参数策略的88.4%。固定预训练参数策略的训练所需时间为10min,少于微调预训练参数策略的16h及随机初始化参数策略的24h。在训练数据减少后,随机初始化参数策略得到的模型准确率平均下降8.6%,而迁移学习组准确率平均下降2.5%。结论结合迁移学习中的微调策略和NASNet架构的新型识别算法在小样本数据集下仍保持高准确率,具有高度的鲁棒性,可用于DR的有效筛查。