WaveAttention-ResNet: A Deep Learning-based Intelligent Diagnostic Model for the Auxiliary Diagnosis of Multiple Retinal Diseases

Biao Guo¹Daqing Wang¹Ruiqi Zhang²Jia Hou²Wenchao Liu³Yongfei Wu⁴Xudong Yang¹Lijuan Zhang^2*

• ¹Netchina Huaxin Technology Co., Ltd., Taiyuan, Shanxi, China
• ²Shanxi Eye Hospital, Taiyuan, China
• ³Shanxi Medical University, Taiyuan, Shanxi Province, China
• ⁴Taiyuan University of Technology, Taiyuan, Shanxi Province, China

Objective: This study constructs a deep learning-based combined algorithm named WaveAttention ResNet (WARN) to investigate the classification accuracy for seven common retinal diseases and the feasibility of AI-assisted diagnosis in this field. Methods:First, a deep learning-based classification network is constructed. The network is built upon ResNet18, integrated with the Convolutional Block Attention Module (CBAM) and wavelet convolution modules, forming the WARN method for retinal disease classification. Second, the public OCTDL dataset is used to train WARN, which contains classification data for seven retinal disease types: age-related macular degeneration (AMD), diabetic macular edema (DME), epiretinal membrane (ERM), normal (NO), retinal artery occlusion (RAO), retinal vein occlusion (RVO), and vitreomacular interface disease (VID). During this process, ablation experiments and significance tests are conducted on WARN, and comprehensive analyses of various indicators for WARN, ResNet-18, ResNet-50, Swin Transformer v2, EfficientNet, and Vision Transformer (ViT) are performed in retinal disease classification tasks. Finally, data provided by Shanxi Eye Hospital are used for testing, and classification results are analyzed. Results: WARN demonstrates excellent performance on the public OCTDL dataset. Ablation experiments and significance tests confirm the effectiveness of WARN, achieving an accuracy of 90.68%, F1-score of 91.29%, AUC of 97.50%, precision of 93.31%, and recall of 90.68% with relatively short training time. In the dataset from Shanxi Eye Hospital, WARN also performs well, with a recall of 90.85%, precision of 79.94%, and accuracy of 89.18%. Conclusion: This study fully confirms that the constructed WARN is efficient and feasible for classifying seven common retinal diseases. It further highlights the enormous potential and broad application prospects of AI technology in the field of auxiliary medical diagnosis.