###
计算机系统应用英文版:2024,33(7):129-138
本文二维码信息
码上扫一扫!
基于改进U-Net的多流视网膜血管分割算法
(1.广西民族大学 人工智能学院, 南宁 530006;2.广西混杂计算与集成电路设计分析重点实验室, 南宁 530006)
Multi-flow Retinal Vessel Segmentation Algorithm Based on Improved U-Net
(1.School of Artificial Intelligence, Guangxi Minzu University, Nanning 530006, China;2.Guangxi Key Laboratory of Hybrid Computation and IC Design Analysis, Nanning 530006, China)
摘要
图/表
参考文献
相似文献
本文已被:浏览 241次   下载 615
Received:December 25, 2023    Revised:January 29, 2024
中文摘要: 针对传统U型网络特征冗余以及视网膜血管形态复杂、细血管分割困难等问题, 提出一种基于改进U-Net的多流视网膜血管分割算法. 算法包含两种特征流向, 分别是全局分割流与边缘特化流. 为了减少特征冗余, 全局分割流在部分卷积的基础上采用快速提取模块替代传统的U-Net卷积块, 构建了能够高效提取血管特征、加快算法推理速度的改进U-Net模型; 为了减少噪声干扰、提升细血管的分割精度, 边缘特化流利用形态学生成的边缘标注信息为指导, 采用多个边缘提取模块, 结合全局分割流的高级语义特征以及边缘注意力, 更具针对性的提取血管细节信息, 增强细血管的特征表达. 在DRIVE与STARE数据集上进行了算法的有效性测试, 敏感度分别为0.8415和0.8369, 准确率分别为0.9701和0.9718, AUC值分别为0.9877和0.9909, 整体性能优于现有算法.
Abstract:In addressing issues such as feature redundancy in traditional U-shaped networks and the complexity of retinal vascular morphology, as well as challenges in segmenting fine blood vessels, this study proposes a multi-flow retinal vascular segmentation algorithm based on improved U-Net. The algorithm incorporates two feature flows, a global segmentation flow and a boundary-specialized flow. To reduce feature redundancy, the global segmentation flow replaces the traditional U-Net convolution block with a fast extraction module based on partial convolution and constructs an improved U-Net model that can efficiently extract vascular features and accelerate algorithm inference speed. To minimize noise interference and enhance the segmentation accuracy of fine blood vessels, the boundary-specialized flow utilizes morphologically generated boundary annotations as guidance. Multiple boundary extraction modules, in combination with the high-level semantic features from the global segmentation flow and boundary attention, are employed to more selectively extract vascular details, thereby strengthening the feature representation of fine blood vessels. The effectiveness of the algorithm is evaluated on the DRIVE and STARE datasets, yielding sensitivity values of 0.841 5 and 0.836 9, accuracy values of 0.970 1 and 0.971 8, and AUC values of 0.987 7 and 0.990 9, respectively. The overall performance surpassed that of existing algorithms.
文章编号:     中图分类号:    文献标志码:
基金项目:国家自然科学基金(61866003)
引用文本:
陆锡恒,宣士斌.基于改进U-Net的多流视网膜血管分割算法.计算机系统应用,2024,33(7):129-138
LU Xi-Heng,XUAN Shi-Bin.Multi-flow Retinal Vessel Segmentation Algorithm Based on Improved U-Net.COMPUTER SYSTEMS APPLICATIONS,2024,33(7):129-138