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 |