提出新模型AttUKAN与对比损失,提升视网膜血管分割精度
Novel Extraction of Discriminative Fine-Grained Feature to Improve Retinal Vessel Segmentation
- 用注意力门增强KAN网络,更好捕捉细粒度特征
- 设计标签引导的像素级对比损失,区分血管与背景像素对
- 在5个数据集上达最优性能,适合医学图像分割研究者
视网膜血管分割是多种严重眼病早期检测的关键方法。尽管神经网络推动了该领域进展,但仍面临挑战:需对眼底图像中每个像素进行分类,尤其依赖图像内部的判别性特征提取。现有方法多关注解码器输出与标签的差异,忽视编码器提供的细粒度特征表示。为此,本文提出新型注意力U型柯尔莫戈洛夫-阿诺德网络AttUKAN及标签引导的像素级对比损失。通过在KAN中引入注意力门,抑制无关特征激活并实现非线性建模,提升模型敏感性和可解释性;同时设计新型对比损失,指导模型区分前景血管像素对与背景像素对,强化判别性特征学习。在DRIVE、STARE、CHASE_DB1、HRF及自研数据集上实验,AttUKAN分别取得82.50%、81.14%、81.34%、80.21%和80.09%的F1分数,以及70.24%、68.64%、68.59%、67.21%和66.94%的MIoU,优于11种现有网络。定量与定性结果均表明其达到当前最优性能。
原文摘要 · Abstract (English)
Retinal vessel segmentation is a vital early detection method for several severe ocular diseases. Despite significant progress in retinal vessel segmentation with the advancement of Neural Networks, there are still challenges to overcome. Specifically, retinal vessel segmentation aims to predict the class label for every pixel within a fundus image, with a primary focus on intra-image discrimination, making it vital for models to extract more discriminative features. Nevertheless, existing methods primarily focus on minimizing the difference between the output from the decoder and the label, but ignore fully using feature-level fine-grained representations from the encoder. To address these issues, we propose a novel Attention U-shaped Kolmogorov-Arnold Network named AttUKAN along with a novel Label-guided Pixel-wise Contrastive Loss for retinal vessel segmentation. Specifically, we implement Attention Gates into Kolmogorov-Arnold Networks to enhance model sensitivity by suppressing irrelevant feature activations and model interpretability by non-linear modeling of KAN blocks. Additionally, we also design a novel Label-guided Pixel-wise Contrastive Loss to supervise our proposed AttUKAN to extract more discriminative features by distinguishing between foreground vessel-pixel pairs and background pairs. Experiments are conducted across four public datasets including DRIVE, STARE, CHASE_DB1, HRF and our private dataset. AttUKAN achieves F1 scores of 82.50%, 81.14%, 81.34%, 80.21% and 80.09%, along with MIoU scores of 70.24%, 68.64%, 68.59%, 67.21% and 66.94% in the above datasets, which are the highest compared to 11 networks for retinal vessel segmentation. Quantitative and qualitative results show that our AttUKAN achieves state-of-the-art performance and outperforms existing retinal vessel segmentation methods. Our code will be available at https://github.com/stevezs315/AttUKAN.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。