Development and Application of an Intelligent Diagnosis System for Retinal Vein Occlusion Based on Deep Learning
Wei Xu, Zhipeng Yan, Nan Chen, Yuxin Luo, Yuke Ji, Minli Wang, Zhe Zhang
Abstract
This study is aimed at developing an intelligent algorithm based on deep learning and discussing its application for the classification and diagnosis of retinal vein occlusions (RVO) using fundus images. A total of 501 fundus images of healthy eyes and patients with RVO were used for model training and testing to investigate an intelligent diagnosis system. The images were first classified into four categories by fundus disease specialists: (i) healthy fundus (group 0), (ii) branch RVO (BRVO) (group 1), (iii) central RVO (CRVO) (group 2), and (iv) macular branch RVO (MBRVO) (group 3), before being diagnosed using the ResNet18 network model. Intelligent diagnoses were compared with clinical diagnoses. The specificity of the intelligent diagnosis system under each attention mechanism was 100% in group 0 and also revealed a high sensitivity of over 95%, <a:math xmlns:a="http://www.w3.org/1998/Math/MathML" id="M1"> <a:mi>F</a:mi> <a:mn>1</a:mn> </a:math> score of over 97%, and an accuracy of over 97% in this group. For the other three groups, the specificities of diagnosis ranged from 0.45 to 0.91 with different attention mechanisms, in which the ResNet18+coordinate attention (CA) model had the highest specificities of 0.91, 0.88, and 0.83 for groups 1, 2, and 3, respectively. It also provided a high accuracy of over 94% with a coordinate attention mechanism in all four groups. The intelligent diagnosis and classifier system developed herein based on deep learning can determine the presence of RVO and classify disease according to the site of occlusion. This proposed system is expected to provide a new tool for RVO diagnosis and screening and will help solve the current challenges due to the shortage of medical resources.