Automated retinopathy of prematurity screening using deep neural network with attention mechanism

Yuanyuan Peng; Weifang Zhu; Feng Chen; Daoman Xiang; Xinjian Chen

doi:10.1117/12.2548290

10 March 2020 Automated retinopathy of prematurity screening using deep neural network with attention mechanism

Yuanyuan Peng, Weifang Zhu, Feng Chen, Daoman Xiang, Xinjian Chen

Author Affiliations +

Proceedings Volume 11313, Medical Imaging 2020: Image Processing; 1131321 (2020) https://doi.org/10.1117/12.2548290
Event: SPIE Medical Imaging, 2020, Houston, Texas, United States

Abstract

Retinopathy of prematurity (ROP) is an ocular disease which occurs in premature babies and is considered as one of the largest preventable causes of childhood blindness. However, insufficient ophthalmologists are qualified for ROP screening, especially in developing countries. Therefore, automated screening of ROP is particularly important. In this paper, we propose a new ROP screening network, in which pre-trained ResNet18 is taken as backbone and a proposed attention block named Complementary Residual Attention Block (CRAB) and Squeeze-and-Excitation (SE) block as channel attention module are introduced. Our main contributions are: (1) Demonstrating the 2D convolutional neural network model pre-trained on natural images can be fine-tuned for ROP screening. (2) Based on the pre-trained ResNet18, we propose an improved scheme combining which that effectively integrates attention mechanism for ROP screening. The proposed classification network was evaluated on 9794 fundus images from 650 subjects, in which 8351 are randomly selected as training set according to subjects and others are selected as testing set. The results showed that the performance of the proposed ROP screening network achieved 99.17% for accuracy, 98.65% for precision, 98.31% for recall, 98.48% for F1 score and 99.84% for AUC. The preliminary experimental results show the effectiveness of the proposed method.

Citation Download Citation

Yuanyuan Peng, Weifang Zhu, Feng Chen, Daoman Xiang, and Xinjian Chen "Automated retinopathy of prematurity screening using deep neural network with attention mechanism", Proc. SPIE 11313, Medical Imaging 2020: Image Processing, 1131321 (10 March 2020); https://doi.org/10.1117/12.2548290

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

;

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE