Retinal image quality assessment based on FA-Net

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