高级检索

跨层Transformer与多尺度自适应融合的视网膜血管分割算法

Cross-Layer Transformer and Multi-Scale Adaptive Fusion of Retinal Vascular Segmentation Algorithm

  • 摘要: 针对现有视网膜血管分割存在视盘误分割、主血管纹理模糊和微细分支血管断裂等问题, 提出融合跨层Transformer (CLTransformer)与跨尺度注意的视网膜血管分割算法. 首先设计轻量化残差编解码模块用于编码和解码器部分, 实现血管纹理特征的粗粒度提取;其次在编解码连接处采用多尺度特征选择模块, 用于跨级融合粗粒度特征;再次在网络底部加入CLTransformer模块, 对深层语义信息交叉融合, 以细化视网膜血管特征轮廓;最后使用融合损失函数监督算法的训练和测试. 在DRIVE, STARE和CHASE_DB1数据集上进行实验, 其准确度分别为97.10%, 97.66%和97.62%, 特异性分别为98.64%, 99.03%和98.72%, F1分数分别为83.05%, 84.07%和81.18%.

     

    Abstract: A retina vessel segmentation algorithm that integrates CLTransformer with cross-scale attention is proposed in this paper to address various issues encountered by the existing methods, such as the nocorrect segmentation of the optic disc, the blurring of main vessel texture, and the breaking of microvascular branches. Firstly, a lightweight residual encoder-decoder module is developed for encoding and decoding, which enables the coarse-grained extraction of vessel texture features. Secondly, a multi-scale feature selection module is applied at the point of connection between the encoder and the decoder to fuse the coarse-grained features of different levels. Thirdly, a cross-layer Transformer module is deployed at the bottom of the network to cross-fuse deep semantic information, thus refining vessel feature contours. Finally, a fusion loss function is used to conduct supervision on the training and testing of the algorithm. According to the results of experiments conducted on the DRIVE, STARE, and CHASE_DB1 datasets, the accuracy reaches 97.10%, 97.66%, and 97.62%, respectively; the specificity reaches 98.64%, 99.03%, and 98.72%, respectively; and the F1 score reaches 83.05%, 84.07%, and 81.18%, respectively.

     

/

返回文章
返回