改进U型网络结构,提升眼底血管分割精度
TransUNext: towards a more advanced U-shaped framework for automatic vessel segmentation in the fundus image

- 融合Transformer与CNN,用轻量自注意力捕捉全局与局部特征
- 在4个数据集上AUC均超0.98,最高达0.9910
- 适合眼科影像分析、医学图像分割研究者参考
自动精准分割眼底血管图像已成为糖尿病等眼病计算机辅助诊断的重要前提。由于血管末端与背景对比度低、血管细长且形态多变,高精度视网膜血管分割仍具挑战。本文提出一种更先进的混合式Transformer与CNN的U型架构TransUNext,将高效自注意力机制嵌入U-Net的编码器和解码器,以最小计算开销捕获局部特征与全局依赖。同时引入全局多尺度融合(GMSF)模块升级跳跃连接,融合高层语义与低层细节信息,消除高低层语义差异。受ConvNeXt启发,设计TransNeXt Block优化U-Net各基础模块的计算复杂度,避免跨维度特征空间转换中的信息损失。在DRIVE、STARE、CHASE-DB1和HRF四个公开数据集上的实验结果表明,AUC值分别为0.9867、0.9869、0.9910和0.9887,优于现有最先进方法。
原文摘要 · Abstract (English)
Purpose: Automatic and accurate segmentation of fundus vessel images has become an essential prerequisite for computer-aided diagnosis of ophthalmic diseases such as diabetes mellitus. The task of high-precision retinal vessel segmentation still faces difficulties due to the low contrast between the branch ends of retinal vessels and the background, the long and thin vessel span, and the variable morphology of the optic disc and optic cup in fundus vessel images. Methods: We propose a more advanced U-shaped architecture for a hybrid Transformer and CNN: TransUNext, which integrates an Efficient Self-attention Mechanism into the encoder and decoder of U-Net to capture both local features and global dependencies with minimal computational overhead. Meanwhile, the Global Multi-Scale Fusion (GMSF) module is further introduced to upgrade skip-connections, fuse high-level semantic and low-level detailed information, and eliminate high- and low-level semantic differences. Inspired by ConvNeXt, TransNeXt Block is designed to optimize the computational complexity of each base block in U-Net and avoid the information loss caused by the compressed dimension when the information is converted between the feature spaces of different dimensions. Results: We evaluated the proposed method on four public datasets DRIVE, STARE, CHASE-DB1, and HRF. In the experimental results, the AUC (area under the ROC curve) values were 0.9867, 0.9869, 0.9910, and 0.9887, which exceeded the other state-of-the-art.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。