Reliable and Energy-Efficient Diabetic Retinopathy Screening Using Memristor-Based Neural Networks

Diabetic retinopathy (DR) is a leading cause of permanent vision loss worldwide. It refers to irreversible retinal damage caused due to elevated glucose levels and blood pressure. Regular screening for DR can facilitate its early detection and timely treatment. Neural network-based DR classifiers ca...

Full description

Bibliographic Details
Main Authors: Sumit Diware, Koteswararao Chilakala, Rajiv V. Joshi, Said Hamdioui, Rajendra Bishnoi
Format: Article
Language:English
Published: IEEE 2024-01-01
Series:IEEE Access
Subjects:
Online Access:https://ieeexplore.ieee.org/document/10485620/
_version_ 1797221966809137152
author Sumit Diware
Koteswararao Chilakala
Rajiv V. Joshi
Said Hamdioui
Rajendra Bishnoi
author_facet Sumit Diware
Koteswararao Chilakala
Rajiv V. Joshi
Said Hamdioui
Rajendra Bishnoi
author_sort Sumit Diware
collection DOAJ
description Diabetic retinopathy (DR) is a leading cause of permanent vision loss worldwide. It refers to irreversible retinal damage caused due to elevated glucose levels and blood pressure. Regular screening for DR can facilitate its early detection and timely treatment. Neural network-based DR classifiers can be leveraged to achieve such screening in a convenient and automated manner. However, these classifiers suffer from reliability issue where they exhibit strong performance during development but degraded performance after deployment. Moreover, they do not provide supplementary information about the prediction outcome, which severely limits their widespread adoption. Furthermore, energy-efficient deployment of these classifiers on edge devices remains unaddressed, which is crucial to enhance their global accessibility. In this paper, we present a reliable and energy-efficient hardware for DR detection, suitable for deployment on edge devices. We first develop a DR classification model using custom training data that incorporates diverse image quality and image sources along with improved class balance. This enables our model to effectively handle both on-field variations in retinal images and minority DR classes, enhancing its post-deployment reliability. We then propose a pseudo-binary classification scheme to further improve the model performance and provide supplementary information about the model prediction. Additionally, we present an energy-efficient hardware design for our model using memristor-based computation-in-memory, to facilitate its deployment on edge devices. Our proposed approach achieves reliable DR classification with three orders of magnitude reduction in energy consumption over state-of-the-art hardware platforms.
first_indexed 2024-04-24T13:13:50Z
format Article
id doaj.art-5bf9028c8237420ea73a3f9332eff4e1
institution Directory Open Access Journal
issn 2169-3536
language English
last_indexed 2024-04-24T13:13:50Z
publishDate 2024-01-01
publisher IEEE
record_format Article
series IEEE Access
spelling doaj.art-5bf9028c8237420ea73a3f9332eff4e12024-04-04T23:00:34ZengIEEEIEEE Access2169-35362024-01-0112474694748210.1109/ACCESS.2024.338301410485620Reliable and Energy-Efficient Diabetic Retinopathy Screening Using Memristor-Based Neural NetworksSumit Diware0https://orcid.org/0000-0003-4461-1623Koteswararao Chilakala1Rajiv V. Joshi2https://orcid.org/0009-0007-7486-1531Said Hamdioui3https://orcid.org/0000-0002-8961-0387Rajendra Bishnoi4Computer Engineering Laboratory, Delft University of Technology, Delft, The NetherlandsCapgemini Engineering, Eindhoven, The NetherlandsIBM Thomas J. Watson Research Centre, Yorktown Heights, NY, USAComputer Engineering Laboratory, Delft University of Technology, Delft, The NetherlandsComputer Engineering Laboratory, Delft University of Technology, Delft, The NetherlandsDiabetic retinopathy (DR) is a leading cause of permanent vision loss worldwide. It refers to irreversible retinal damage caused due to elevated glucose levels and blood pressure. Regular screening for DR can facilitate its early detection and timely treatment. Neural network-based DR classifiers can be leveraged to achieve such screening in a convenient and automated manner. However, these classifiers suffer from reliability issue where they exhibit strong performance during development but degraded performance after deployment. Moreover, they do not provide supplementary information about the prediction outcome, which severely limits their widespread adoption. Furthermore, energy-efficient deployment of these classifiers on edge devices remains unaddressed, which is crucial to enhance their global accessibility. In this paper, we present a reliable and energy-efficient hardware for DR detection, suitable for deployment on edge devices. We first develop a DR classification model using custom training data that incorporates diverse image quality and image sources along with improved class balance. This enables our model to effectively handle both on-field variations in retinal images and minority DR classes, enhancing its post-deployment reliability. We then propose a pseudo-binary classification scheme to further improve the model performance and provide supplementary information about the model prediction. Additionally, we present an energy-efficient hardware design for our model using memristor-based computation-in-memory, to facilitate its deployment on edge devices. Our proposed approach achieves reliable DR classification with three orders of magnitude reduction in energy consumption over state-of-the-art hardware platforms.https://ieeexplore.ieee.org/document/10485620/Diabetic retinopathyneural networkscomputation-in-memoryresistive random access memoryRRAMmemristor
spellingShingle Sumit Diware
Koteswararao Chilakala
Rajiv V. Joshi
Said Hamdioui
Rajendra Bishnoi
Reliable and Energy-Efficient Diabetic Retinopathy Screening Using Memristor-Based Neural Networks
IEEE Access
Diabetic retinopathy
neural networks
computation-in-memory
resistive random access memory
RRAM
memristor
title Reliable and Energy-Efficient Diabetic Retinopathy Screening Using Memristor-Based Neural Networks
title_full Reliable and Energy-Efficient Diabetic Retinopathy Screening Using Memristor-Based Neural Networks
title_fullStr Reliable and Energy-Efficient Diabetic Retinopathy Screening Using Memristor-Based Neural Networks
title_full_unstemmed Reliable and Energy-Efficient Diabetic Retinopathy Screening Using Memristor-Based Neural Networks
title_short Reliable and Energy-Efficient Diabetic Retinopathy Screening Using Memristor-Based Neural Networks
title_sort reliable and energy efficient diabetic retinopathy screening using memristor based neural networks
topic Diabetic retinopathy
neural networks
computation-in-memory
resistive random access memory
RRAM
memristor
url https://ieeexplore.ieee.org/document/10485620/
work_keys_str_mv AT sumitdiware reliableandenergyefficientdiabeticretinopathyscreeningusingmemristorbasedneuralnetworks
AT koteswararaochilakala reliableandenergyefficientdiabeticretinopathyscreeningusingmemristorbasedneuralnetworks
AT rajivvjoshi reliableandenergyefficientdiabeticretinopathyscreeningusingmemristorbasedneuralnetworks
AT saidhamdioui reliableandenergyefficientdiabeticretinopathyscreeningusingmemristorbasedneuralnetworks
AT rajendrabishnoi reliableandenergyefficientdiabeticretinopathyscreeningusingmemristorbasedneuralnetworks