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...
Main Authors: | , , , , |
---|---|
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 |