轻量级网络精准分割视网膜血管,适合资源有限的临床场景。
LFRA-Net: A Lightweight Focal and Region-Aware Attention Network for Retinal Vessel Segmentatio
- 在编码器-解码器瓶颈和选择性跳跃连接中引入聚焦与区域感知注意力机制。
- 参数仅0.17万,内存0.66MB,GFLOPs为10.50,在三数据集上Dice超84%。
- 兼顾精度与效率,特别适合算力受限的实时医疗应用。
视网膜血管分割对视力威胁性及系统性疾病早期诊断至关重要,尤其在计算资源有限的真实临床环境中。尽管深度学习分割方法已取得显著进展,现有模型仍面临细小血管提取困难和计算成本高的挑战。本文提出LFRA-Net,通过在编码器-解码器瓶颈处引入聚焦调制注意力,并在选择性跳跃连接中加入区域感知注意力,构建轻量级网络以实现精确高效的视网膜血管分割。该网络通过高效捕捉局部与全局依赖关系,增强特征表达与区域聚焦能力。在三个公开数据集DRIVE、STARE和CHASE_DB上验证,其参数量仅为0.17百万,内存占用0.66MB,计算量10.50 GFLOPs,Dice分数分别为84.28%、88.44%和85.50%,Jaccard指数分别为72.86%、79.31%和74.70%。相比现有方法,LFRA-Net在分割精度与计算开销间实现了更优平衡,适用于资源受限环境下的实时临床应用。代码已开源于https://github.com/Mehwish4593/LFRA-Net。
原文摘要 · Abstract (English)
Retinal vessel segmentation is critical for the early diagnosis of vision-threatening and systemic diseases, especially in real-world clinical settings with limited computational resources. Although significant improvements have been made in deep learning-based segmentation methods, current models still face challenges in extracting tiny vessels and suffer from high computational costs. In this study, we present LFRA-Net by incorporating focal modulation attention at the encoder-decoder bottleneck and region-aware attention in the selective skip connections. LFRA-Net is a lightweight network optimized for precise and effective retinal vascular segmentation. It enhances feature representation and regional focus by efficiently capturing local and global dependencies. LFRA-Net outperformed many state-of-the-art models while maintaining lightweight characteristics with only 0.17 million parameters, 0.66 MB memory size, and 10.50 GFLOPs. We validated it on three publicly available datasets: DRIVE, STARE, and CHASE\_DB. It performed better in terms of Dice score (84.28\%, 88.44\%, and 85.50\%) and Jaccard index (72.86\%, 79.31\%, and 74.70\%) on the DRIVE, STARE, and CHASE\_DB datasets, respectively. LFRA-Net provides an ideal ratio between segmentation accuracy and computational cost compared to existing deep learning methods, which makes it suitable for real-time clinical applications in areas with limited resources. The code can be found at https://github.com/Mehwish4593/LFRA-Net.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。