用3D图结构提升血管连通性分割,解决断续与端点丢失问题。
ViG3D-UNet: Volumetric Vascular Connectivity-Aware Segmentation via 3D Vision Graph Representation
- 基于3D图神经网络与U型架构融合,建模血管拓扑关系。
- 在ASOCA和ImageCAS数据集上实现更连贯的分割结果。
- 适合需要精准血管结构分析的医学影像研究者。
准确的血管分割对冠状动脉可视化和冠心病诊断至关重要。该任务需从体积分空间中提取稀疏的树状血管分支。然而,现有方法因血管分割不连续及端点缺失面临显著挑战。为此,本文提出一种3D视觉图神经网络框架ViG3D-UNet,通过在U型架构中集成3D图表示与聚合,促进连续血管分割。ViG3D模块捕捉体积分空间中的血管连通性与拓扑结构,卷积模块则提取细粒度血管细节。两者通过通道注意力融合形成编码器特征。随后,采用纸夹形偏移解码器,在稀疏特征空间中最小化冗余计算,并恢复特征图尺寸以匹配原始输入。在两个公开数据集ASOCA和ImageCAS上的评估表明,该方法在保持血管分割连通性的同时,实现了更高的分割精度。代码将很快开源。
原文摘要 · Abstract (English)
Accurate vascular segmentation is essential for coronary visualization and the diagnosis of coronary heart disease. This task involves the extraction of sparse tree-like vascular branches from the volumetric space. However, existing methods have faced significant challenges due to discontinuous vascular segmentation and missing endpoints. To address this issue, a 3D vision graph neural network framework, named ViG3D-UNet, was introduced. This method integrates 3D graph representation and aggregation within a U-shaped architecture to facilitate continuous vascular segmentation. The ViG3D module captures volumetric vascular connectivity and topology, while the convolutional module extracts fine vascular details. These two branches are combined through channel attention to form the encoder feature. Subsequently, a paperclip-shaped offset decoder minimizes redundant computations in the sparse feature space and restores the feature map size to match the original input dimensions. To evaluate the effectiveness of the proposed approach for continuous vascular segmentation, evaluations were performed on two public datasets, ASOCA and ImageCAS. The segmentation results show that the ViG3D-UNet surpassed competing methods in maintaining vascular segmentation connectivity while achieving high segmentation accuracy. Our code will be available soon.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。