基于改进胶囊网络的糖尿病性视网膜病变识别研究  

Recognition of diabetic retinopathy based on improved capsule network

作　　者：朱周华[1] 田成源 侯智杰 周怡纳 王斌 ZHU Zhouhua;TIAN Chengyuan;HOU Zhijie;ZHOU Yi′na;WANG Bin(School of Communication and Information Engineering,Xi′an University of Science and Technology,Xi′an 710000,Shaanxi,China)

机构地区：[1]西安科技大学通信与信息工程学院,陕西西安710000

出　　处：《实用医学杂志》2025年第7期968-975,共8页The Journal of Practical Medicine

基　　金：国家自然科学基金资助项目(编号:U19B2015)。

摘　　要：目的针对实际场景下小样本糖尿病性视网膜病变识别模型重要特征不易表达,真、假特征系数分布过于平稳的问题,提出了一种基于改进胶囊网络的小样本糖尿病性视网膜病变识别方法。方法首先采取删除图像不必要边界信息并使用基于Haar基函数的离散小波变换来处理图像的方法,来提高图像的特征表达能力,凸显重要病灶特征;其次,对胶囊网络的卷积层进行改进,采用多分支结构提取视网膜图像的多尺度特征,并添加卷积块注意力模块(convolutional block attention module,CBAM),封装并送入胶囊层中;最后,在动态路由中用sigmoid函数替代softmax函数,增强了模型的鲁棒性。结果所改进网络模型在节选并处理后的Kaggle公开数据集中测试准确率为98.62%。结论改进胶囊网络在小样本糖尿病性视网膜病变识别的任务中所取得的精确度高于当前其他先进算法。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 softmax 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 retinopathy within small sample sizes compared to other state⁃of⁃the⁃art algorithms currently available.

关 键 词：糖尿病性视网膜病变识别 多尺度 小样本 胶囊网络 CBAM 

分 类 号：R318[医药卫生—生物医学工程] TP18[医药卫生—基础医学] R445[自动化与计算机技术—控制理论与控制工程]