轻量级卷积网络实现眼科疾病高精度检测,适合移动端部署。
Detection of retinal diseases using an accelerated reused convolutional network
- 设计新型ArConv卷积层,优化基础计算结构提升效率
- 仅130万参数,在RfMiD数据集上准确率达93.28%
- 比MobileNetV2更轻更快,适合手机端实时诊断
卷积神经网络持续演进,部分研究聚焦于提升准确率,部分致力于加速,还有些关注模型可及性。提升可及性有助于将神经网络应用于更广泛任务,包括眼病检测。早期诊断与眼科医生咨询可预防多种视力障碍。鉴于此问题的重要性,已从角膜收集多种数据集以支持神经网络建模。然而,过去多数方法计算复杂度高。本研究旨在提升深度神经网络的可及性,从最基础层面出发,重新设计并优化卷积层。为此,我们提出一种新型卷积层——ArConv层。得益于该层高效性能,所构建的新模型具备适配移动端的合适复杂度,能以高准确率完成疾病诊断任务。最终模型仅含130万参数。在相同条件下于RfMiD数据集训练评估,相较拥有220万参数的MobileNetV2模型,本模型在测试集上准确率达到0.9328,优于MobileNetV2的0.9266。
原文摘要 · Abstract (English)
Convolutional neural networks are continually evolving, with some efforts aimed at improving accuracy, others at increasing speed, and some at enhancing accessibility. Improving accessibility broadens the application of neural networks across a wider range of tasks, including the detection of eye diseases. Early diagnosis of eye diseases and consulting an ophthalmologist can prevent many vision disorders. Given the importance of this issue, various datasets have been collected from the cornea to facilitate the process of making neural network models. However, most of the methods introduced in the past are computationally complex. In this study, we tried to increase the accessibility of deep neural network models. We did this at the most fundamental level, specifically by redesigning and optimizing the convolutional layers. By doing so, we created a new general model that incorporates our novel convolutional layer named ArConv layers. Thanks to the efficient performance of this new layer, the model has suitable complexity for use in mobile phones and can perform the task of diagnosing the presence of disease with high accuracy. The final model we present contains only 1.3 million parameters. In comparison to the MobileNetV2 model, which has 2.2 million parameters, our model demonstrated better accuracy when trained and evaluated on the RfMiD dataset under identical conditions, achieving an accuracy of 0.9328 versus 0.9266 on the RfMiD test set.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。