用图神经网络提升腹腔镜手术图像的解剖结构分割精度
Graph Neural Networks for Surgical Scene Segmentation
- 结合视觉变压器与图网络,显式建模解剖区域间的空间关系
- 在两个数据集上实现mIoU提升7-8%、mDice提升6%
- 特别擅长分割细小、稀有且关键的解剖结构
准确识别肝胆解剖结构对预防腹腔镜胆囊切除术中的手术并发症至关重要。深度学习模型常因遮挡、长距离依赖和难以捕捉罕见结构的细微几何特征而表现不佳。本文提出两种基于图的分割方法,通过整合视觉变换器(ViT)特征编码器与图神经网络(GNN),增强手术场景分析中的空间与语义理解。其一采用静态k近邻图结合初始残差与身份映射的图卷积网络(GCNII),实现稳定的长程信息传播;其二采用动态可微图生成器(DGG)与图注意力网络(GAT),支持自适应拓扑学习。两个模型在Endoscapes-Seg50和CholecSeg8k基准上进行评估。结果表明,所提方法相较当前最优基线,在平均交并比(mIoU)上提升达7-8%,平均骰子系数(mDice)提升6%,并产生具有解剖一致性的预测,尤其在细小、稀有及安全关键结构上表现突出。结论指出,该图模型结合了ViT的全局上下文与图的关联推理,提升了可解释性与可靠性,为更安全的腹腔镜与机器人辅助手术提供了精准关键解剖特征识别的技术路径。
原文摘要 · Abstract (English)
Purpose: Accurate identification of hepatocystic anatomy is critical to preventing surgical complications during laparoscopic cholecystectomy. Deep learning models often struggle with occlusions, long-range dependencies, and capturing the fine-scale geometry of rare structures. This work addresses these challenges by introducing graph-based segmentation approaches that enhance spatial and semantic understanding in surgical scene analyses. Methods: We propose two segmentation models integrating Vision Transformer (ViT) feature encoders with Graph Neural Networks (GNNs) to explicitly model spatial relationships between anatomical regions. (1) A static k Nearest Neighbours (k-NN) graph with a Graph Convolutional Network with Initial Residual and Identity Mapping (GCNII) enables stable long-range information propagation. (2) A dynamic Differentiable Graph Generator (DGG) with a Graph Attention Network (GAT) supports adaptive topology learning. Both models are evaluated on the Endoscapes-Seg50 and CholecSeg8k benchmarks. Results: The proposed approaches achieve up to 7-8% improvement in Mean Intersection over Union (mIoU) and 6% improvement in Mean Dice (mDice) scores over state-of-the-art baselines. It produces anatomically coherent predictions, particularly on thin, rare and safety-critical structures. Conclusion: The proposed graph-based segmentation methods enhance both performance and anatomical consistency in surgical scene segmentation. By combining ViT-based global context with graph-based relational reasoning, the models improve interpretability and reliability, paving the way for safer laparoscopic and robot-assisted surgery through a precise identification of critical anatomical features.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。