轻量网络FSG-Net提升视网膜血管分割精度,兼顾性能与可复现性。
Full-scale Representation Guided Network for Retinal Vessel Segmentation
- 用全尺度特征模块捕捉血管结构,再通过注意力引导卷积精细优化
- 在多个公开数据集上达到与顶尖方法相当的分割准确率
- 模块可替换,适合各类医学图像分割任务,代码开源可复现
过去十年中,U-Net及其变体始终是视网膜血管分割的主流架构。本文提出全尺度引导网络(FSG-Net),采用现代化卷积块构建特征表示模块,有效捕获全尺度结构信息,并通过引导卷积块进一步细化。具体地,在引导卷积中引入注意力引导滤波器,利用其与非锐化掩模的相似性增强细小血管结构。将全尺度信息传递至注意力模块,生成更具上下文相关性的注意力图,再输入至注意力引导滤波器实现进一步优化。前导结构可替换为任意U-Net变体,保证灵活性与可扩展性。为公平对比,我们复现了多个公开研究以评估其可扩展性与可复现性。实验表明,尽管架构紧凑,FSG-Net在多个公共数据集上性能仍媲美当前最优方法。消融实验验证了各组件对性能提升的实质性贡献。代码已开源:https://github.com/ZombaSY/FSG-Net-pytorch。
原文摘要 · Abstract (English)
The U-Net architecture and its variants have remained state-of-the-art (SOTA) for retinal vessel segmentation over the past decade. In this study, we introduce a Full-Scale Guided Network (FSG-Net), where a novel feature representation module using modernized convolution blocks effectively captures full-scale structural information, while a guided convolution block subsequently refines this information. Specifically, we introduce an attention-guided filter within the guided convolution block, leveraging its similarity to unsharp masking to enhance fine vascular structures. Passing full-scale information to the attention block facilitates the generation of more contextually relevant attention maps, which are then passed to the attention-guided filter, providing further refinement to the segmentation performance. The structure preceding the guided convolution block can be replaced by any U-Net variant, ensuring flexibility and scalability across various segmentation tasks. For a fair comparison, we re-implemented recent studies available in public repositories to evaluate their scalability and reproducibility. Our experiments demonstrate that, despite its compact architecture, FSG-Net delivers performance competitive with SOTA methods across multiple public datasets. Ablation studies further demonstrate that each proposed component meaningfully contributes to this competitive performance. Our code is available on https://github.com/ZombaSY/FSG-Net-pytorch.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。