Retinal image quality assessment based on FA-Net

Authors:Wan Cheng,  You Qijing,  Sun Jing,  Shen Jianxin,  Yu Qiuli
DOI: 10.3760/cma.j.issn.2095-0160.2019.08.004
Published 2019-08-10
Cite as Chin J Exp Ophthalmol, 2019,37(8): 608-612.

Abstract

Objective

To propose a deep learning-based retinal image quality classification network, FA-Net, to make convolutional neural network (CNN) more suitable for image quality assessment in eye disease screening system.

Methods

The main network of FA-Net was composed of VGG-19.On this basis, attention mechanism was added to the CNN.By using transfer learning method in training, the weight of ImageNet was used to initialize the network.The attention net is based on foreground extraction by extracting the blood vessel and suspected regions of lesion and assigning higher weights to region of interest to enhance the learning of these important areas.

Results

Total of 2 894 fundus images were used for training FA-Net.FA-Net achieved 97.65% classification accuracy on a test set containing 2 170 fundus images, with the sensitivity and specificity of 0.978 and 0.960, respectively, and the area under curve(AUC) was 0.995.

Conclusions

Compared with other CNNs, the proposed FA-Net has better classification performance and can evaluate retinal fundus image quality more accurately and efficiently.The network takes into account the human visual system (HVS) and human attention mechanism.By adding attention module into the VGG-19 network structure, the classification results can be better interpreted as well as better classification performance.

Key words:

Retinal image quality assessment; Retinal image enhancement; Convolutional neural networks; Deep learning

Contributor Information

Wan Cheng
College of Electronic Information Engineering, Nanjing University of Aeronautics and Astronautics, Nanjing 210016, China
You Qijing
College of Electronic Information Engineering, Nanjing University of Aeronautics and Astronautics, Nanjing 210016, China
Sun Jing
College of Electronic Information Engineering, Nanjing University of Aeronautics and Astronautics, Nanjing 210016, China
Shen Jianxin
College of Electronic Information Engineering, Nanjing University of Aeronautics and Astronautics, Nanjing 210016, China
Yu Qiuli
Department of Ophthalmology, BenQ Hospital of Nanjing Medical University, Nanjing 210019, China
(Read 34 times, 1 visits today)
Updated: October 9, 2019 — 7:15 am