Deploying and Evaluating Multiple Deep Learning Models on Edge Devices for Diabetic Retinopathy Detection

Asare, Akwasi, Gookyi, Dennis Agyemanh Nana, Boateng, Derrick, Wulnye, Fortunatus Aabangbio

arXiv.org Artificial Intelligence 

Abstract: Diabetic Retinopathy (DR), a leading cause of vision impairment in individuals with diabetes, affects approximately 34.6% of diabetes patients globally, with the number of cases projected to reach 242 million by 2045 . Traditional DR diagnosis relies on the manual examination of retinal fundus images, which is both time - consuming and resource intensive . This study presents a novel solution using Edge Impulse to deploy multiple deep learning models for real - time DR detection on edge devices . A robust dataset of over 3,662 retinal fundus images, sourced from the Kaggle EyePACS dataset, was curated, and enhanced through preprocessing techniques, including augmentation and normalization. Using TensorFlow, various Convolutional Neural Networks (CNNs), such as MobileNet, ShuffleNet, SqueezeNet, and a custom Deep Neural Network (DNN), were designed, trained, and optimized for edge deployment. The models were converted to TensorFlo w Lite and quantized to 8 - bit integers to reduce their size and enhance inference speed, with minimal trade - offs in accuracy. Performance evaluations across different edge hardware platforms, including smartphones and microcontrollers, highlighted key metrics such as inference speed, accuracy, precision, and resource utilization. MobileNet ach ieved an accuracy of 96.45%, while SqueezeNet demonstrated strong real - time performance with a small model size of 176 KB and latency of just 17 ms on GPU. ShuffleNet and the custom DNN achieved moderate accuracy but excelled in resource efficiency, making them suitable for lower - end devices. This integration of edge AI technology into healthcare presents a scalable, cost - effective solution for early DR detection, providing timely and accurate diagnosis, especially in resource - constrained and remote healthc are settings. Keywords: Diabetic Retinopathy, Edge Impulse, Deep Learning, Microcontroller Units, TensorFlow, Model Quantization, Edge AI 1. INTRODUCTION Diabetes is a significant global health challenge, with rates rising worldwide over the past two decades. One of the major complications of diabetes is Diabetic Retinopathy (DR), a severe eye condition that can lead to vision loss in adults (Maqsood and Gupta, 2022) . DR is caused by damage to the blood vessels in the retina, leading to swelling and leakage, which can impair vision (Saeed, Hussain and Aboalsamh, 2021) . Approximately 34.6% of individuals with diabetes develop DR, making it the leading cause of vision loss among working - age adults (Li et al., 2023) .