Recognition of diabetic retinopathy based on improved capsule network
10.3969/j.issn.1006-5725.2025.07.006
- VernacularTitle:基于改进胶囊网络的糖尿病性视网膜病变识别研究
- Author:
Zhouhua ZHU
1
;
Chengyuan TIAN
1
;
Zhijie HOU
1
;
Yi'na ZHOU
1
;
Bin WANG
1
Author Information
1. 西安科技大学通信与信息工程学院(陕西 西安 710000)
- Publication Type:Journal Article
- Keywords:
diabetic retinopathy recognition;
multi-scale;
small sample;
capsule network;
CBAM
- From:
The Journal of Practical Medicine
2025;41(7):968-975
- CountryChina
- Language:Chinese
-
Abstract:
Objective To address the challenges of accurately capturing critical features in small-sample diabetic retinopathy(DR)recognition models in real-world applications,and the overly smooth distribution of true and false feature coefficients,we propose an enhanced small-sample DR recognition method based on an improved capsule network.Methods Firstly,the method enhances image feature representation by removing redundant boundary information and employing discrete wavelet transform based on the Haar wavelet function,thereby high-lighting critical pathological features.Secondly,the convolutional layer of the capsule network is optimized through a multi-branch architecture to extract multi-scale features from retinal images,while incorporating a convolutional block attention module that is subsequently fed into the capsule layer.Finally,the sigmoid function replaces the soft-max function in dynamic routing,thereby improving the model's robustness.Result The enhanced neural network model achieved an accuracy of 98.62%on the Kaggle public dataset following a rigorous selection and preprocessing procedure.Conclusion The enhanced capsule network demonstrated superior precision in identifying diabetic reti-nopathy within small sample sizes compared to other state-of-the-art algorithms currently available.
