75K参数的轻量模型,实现高精度视网膜血管分割,适合移动端部署。
LightVesselNet: An Ultra-Lightweight Sub-100K Parameter Network for Retinal Blood Vessel Segmentation

- 采用紧凑编码器-解码器结构,融合通道与空间注意力机制。
- 在5个数据集上敏感度超0.8,Dice系数最高达0.8649。
- 参数少、计算量低,适合医疗边缘设备和移动筛查应用。
视网膜血管分割对糖尿病视网膜病变和青光眼的早期检测至关重要。尽管深度学习模型已取得优异分割效果,但通常需大量计算资源,难以在边缘设备部署。本文提出LightVesselNet,一种专为资源受限环境设计的高效神经网络。该模型仅含75K参数,性能却媲美更大模型。其采用紧凑编码器-解码器架构,结合通道与空间注意力机制、瓶颈处多尺度特征聚合模块,以及解码器中的子像素上采样策略;并引入专用边缘残差连接以保留细小血管细节。在五个公开数据集(DRIVE、STARE、CHASEDB1、FIVES、HRF)上的实验显示,敏感度分别为0.8189、0.8499、0.8640、0.8634、0.8096,Dice系数分别为0.8070、0.8072、0.8181、0.8649、0.7686。相比现有最优模型,LightVesselNet在性能与参数量或GFlops比上表现更优。跨数据集评估验证了其泛化能力。整体而言,LightVesselNet是低资源临床场景与移动筛查工具的理想选择。
原文摘要 · Abstract (English)
Retinal blood vessel segmentation plays a vital role in the early detection of diabetic retinopathy and glaucoma. While recent deep learning models have achieved great segmentation accuracy, they typically require heavy computational resources, making real-world deployment on edge devices difficult. In this paper, we propose LightVesselNet, an efficient neural network designed for retinal vessel segmentation in a resource-constrained environment. Despite containing only 75K parameters, LightVesselNet performs competitively with much larger models. The network employs a compact encoder decoder architecture enhanced with channel and spatial attention mechanisms, a multi-scale feature aggregation module at the bottleneck, and a subpixel upsampling strategy in the decoder. A dedicated edge residual connection preserves fine vessel detail throughout decoding. Extensive experiments on five publicly available datasets: DRIVE, STARE, CHASEDB1, FIVES, and HRF, yield sensitivity scores of 0.8189, 0.8499, 0.8640, 0.8634, 0.8096, and Dice coefficients of 0.8070, 0.8072, 0.8181, 0.8649, and 0.7686, respectively. LightVesselNet shows improved efficiency (Performance vs Parameter or GFlops) compared to State-of-the-Art models. Cross-dataset evaluation confirms the model's generalisation capability. Overall, LightVesselNet is a strong candidate for deployment in low-resource clinical settings and mobile screening tools.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。