EDR-Net: Lightweight deep neural network architecture for detecting referable diabetic retinopathy
Present architecture of convolution neural network for diabetic retinopathy (DR-Net) is based on normal convolution (NC). It incurs high computational cost as NC uses a multiplicative weight that measures a combined correlation in both cross-channel and spatial dimension of layer's inputs. This...
Saved in:
Main Authors: | , , , |
---|---|
Format: | Article |
Published: |
Institute of Electrical and Electronics Engineers Inc.
2022
|
Subjects: | |
Online Access: | http://eprints.utm.my/104439/ http://dx.doi.org/10.1109/TBCAS.2022.3182907 |
Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
id |
my.utm.104439 |
---|---|
record_format |
eprints |
spelling |
my.utm.1044392024-02-04T10:04:26Z http://eprints.utm.my/104439/ EDR-Net: Lightweight deep neural network architecture for detecting referable diabetic retinopathy Aujih, Ahmad Bukhari Shapiai, Mohd. Ibrahim Meriaudeau, Fabrice Tang, Tong Boon T Technology (General) Present architecture of convolution neural network for diabetic retinopathy (DR-Net) is based on normal convolution (NC). It incurs high computational cost as NC uses a multiplicative weight that measures a combined correlation in both cross-channel and spatial dimension of layer's inputs. This might cause the overall DR-Net architecture to be over-parameterised and computationally inefficient. This paper proposes EDR-Net-a new end-to-end, DR-Net architecture with depth-wise separable convolution module. The EDR-Net architecture was trained with DRKaggle-train dataset (35,126 images), and tested on two datasets, i.e. DRKaggle-test (53,576 images) and Messidor-2 (1,748 images). Results showed that the proposed EDR-Net achieved predictive performance comparable with current state-of-the-arts in detecting referable diabetic retinopathy (rDR) from fundus images and outperformed other light weight architectures, with at least two times less computation cost. This makes it more amenable for mobile device based computer-assisted rDR screening applications. Institute of Electrical and Electronics Engineers Inc. 2022 Article PeerReviewed Aujih, Ahmad Bukhari and Shapiai, Mohd. Ibrahim and Meriaudeau, Fabrice and Tang, Tong Boon (2022) EDR-Net: Lightweight deep neural network architecture for detecting referable diabetic retinopathy. IEEE Transactions on Biomedical Circuits and Systems, 16 (3). pp. 467-478. ISSN 1932-4545 http://dx.doi.org/10.1109/TBCAS.2022.3182907 DOI : 10.1109/TBCAS.2022.3182907 |
institution |
Universiti Teknologi Malaysia |
building |
UTM Library |
collection |
Institutional Repository |
continent |
Asia |
country |
Malaysia |
content_provider |
Universiti Teknologi Malaysia |
content_source |
UTM Institutional Repository |
url_provider |
http://eprints.utm.my/ |
topic |
T Technology (General) |
spellingShingle |
T Technology (General) Aujih, Ahmad Bukhari Shapiai, Mohd. Ibrahim Meriaudeau, Fabrice Tang, Tong Boon EDR-Net: Lightweight deep neural network architecture for detecting referable diabetic retinopathy |
description |
Present architecture of convolution neural network for diabetic retinopathy (DR-Net) is based on normal convolution (NC). It incurs high computational cost as NC uses a multiplicative weight that measures a combined correlation in both cross-channel and spatial dimension of layer's inputs. This might cause the overall DR-Net architecture to be over-parameterised and computationally inefficient. This paper proposes EDR-Net-a new end-to-end, DR-Net architecture with depth-wise separable convolution module. The EDR-Net architecture was trained with DRKaggle-train dataset (35,126 images), and tested on two datasets, i.e. DRKaggle-test (53,576 images) and Messidor-2 (1,748 images). Results showed that the proposed EDR-Net achieved predictive performance comparable with current state-of-the-arts in detecting referable diabetic retinopathy (rDR) from fundus images and outperformed other light weight architectures, with at least two times less computation cost. This makes it more amenable for mobile device based computer-assisted rDR screening applications. |
format |
Article |
author |
Aujih, Ahmad Bukhari Shapiai, Mohd. Ibrahim Meriaudeau, Fabrice Tang, Tong Boon |
author_facet |
Aujih, Ahmad Bukhari Shapiai, Mohd. Ibrahim Meriaudeau, Fabrice Tang, Tong Boon |
author_sort |
Aujih, Ahmad Bukhari |
title |
EDR-Net: Lightweight deep neural network architecture for detecting referable diabetic retinopathy |
title_short |
EDR-Net: Lightweight deep neural network architecture for detecting referable diabetic retinopathy |
title_full |
EDR-Net: Lightweight deep neural network architecture for detecting referable diabetic retinopathy |
title_fullStr |
EDR-Net: Lightweight deep neural network architecture for detecting referable diabetic retinopathy |
title_full_unstemmed |
EDR-Net: Lightweight deep neural network architecture for detecting referable diabetic retinopathy |
title_sort |
edr-net: lightweight deep neural network architecture for detecting referable diabetic retinopathy |
publisher |
Institute of Electrical and Electronics Engineers Inc. |
publishDate |
2022 |
url |
http://eprints.utm.my/104439/ http://dx.doi.org/10.1109/TBCAS.2022.3182907 |
_version_ |
1792147741281878016 |
score |
13.211869 |