Published January 1, 2025 | Version v1
Conference paper Open

Optimizing Deep Learning Models for Ophthalmic Disease Detection on Resource-Constrained Devices

  • 1. Hacettepe Univ, Dept Comp Engn, Artificial Intelligence Engn, Ankara, Turkiye
  • 2. Hacettepe Univ, Dept Comp Engn, Ankara, Turkiye

Description

This paper explores the optimization of convolutional neural networks (CNNs) for ophthalmic disease detection via fundus image classification on resource-constrained edge devices. The study investigates the impact of pruning, quantization, and knowledge distillation on reducing model size and inference time while maintaining high classification accuracy. Using a dataset of retinal fundus images, the optimized models were deployed on a Jetson Nano microcontroller. Experimental results demonstrate that post-training quantization achieves the best trade-off between efficiency and accuracy, reducing model size to 11.22MB while maintaining a validation accuracy of 97.39%. Hybrid approaches combining knowledge distillation and quantization further reduced parameters while preserving performance, highlighting promising strategies for deploying deep learning in edge AI healthcare applications.

Files

bib-2d1573e4-f37a-4462-9220-904227858131.txt

Files (268 Bytes)

Name Size Download all
md5:d0378c0aef2386cce439a56f596276d3
268 Bytes Preview Download