CD-Net: 基于跨维特征融合的OCTA图像视网膜血管分割网络
CD-Net: A Cross-Dimensional Feature Fusion Network for Retinal Vessel Segmentation Based on OCTA Images
-
摘要: 在光学相干断层扫描血管造影(OCTA)图像中, 提取视网膜血管对于计算机辅助诊断以及治疗多种眼部疾病具有重要意义. 然而, 现有方法未能充分利用不同维度数据间视网膜血管特征, 缺乏对视网膜血管拓扑形态的关注. 针对上述问题, 提出一种新颖的基于跨维特征融合的OCTA视网膜血管分割网络(CD-Net), 网络由跨维学习路径及平面学习路径组成, 联合3D体数据和2D投影图作为输入. 首先, 针对不同维度数据间的互补性及特异性, 在跨维学习路径中设计了跨维学习块(CDLB)及血管拓扑增强模块(VTEM), 以更精细地捕获微小血管特征, 突出平面血管的拓扑形态; 其次, 为了将3D特征压缩至2D特征, 保留更为丰富的血管特征信息, 采用跨步压缩模块(SCM)逐步将特征压缩至2D; 然后, 将两种编码块分支中的特征在通道及空间层面进行有效融合, 获取跨维的血管特征; 最后, 在平面学习路径中, 每个跨维学习块生成的互补特征合并至对应的平面编码块中, 进一步融合跨维血管特征, 获得最终的2D分割结果. 在OCTA-500数据集上进行的实验结果表明, CD-Net在DSC上达到90.39%, 性能优于文中对比方法.Abstract: Extracting retinal vessels from Optical Coherence Tomography Angiography (OCTA) images is crucial for computer-aided diagnosis and treatment of various ocular diseases. However, existing methods fail to fully leverage the retinal vessel features across different dimensions and lack attention to the topological morphology of retinal vessels. To address these issues, a novel OCTA retinal vessel segmentation network based on cross-dimensional feature fusion, named CD-Net, is proposed. This network comprises a cross-dimensional learning path and a planar learning path, jointly utilizing 3D volumetric data and 2D projection images as input.Firstly, considering the complementarity and specificity of data across different dimensions, the cross-dimensional learning block (CDLB) and vessel topology enhancement module (VTEM) are designed within the cross-dimensional learning path. These components aim to capture finer features of microvasculature and highlight the topological morphology of planar vessels. Secondly, to compress 3D features into 2D while retaining richer vessel feature information, a stride compression module (SCM) is employed to progressively reduce the dimensionality of features to 2D. Thirdly, features from the two encoding branches are effectively fused at both the channel and spatial levels to obtain cross-dimensional vessel features. Finally, in the planar learning path, the complementary features generated by each cross-dimensional learning block are merged into the corresponding planar encoding block, further integrating cross-dimensional vessel features to achieve the final 2D segmentation result. Experimental results on the OCTA-500 dataset demonstrate that CD-Net achieves a Dice similarity coefficient of 90.39%, outperforming the comparison methods mentioned in the text.