用不对称特征增强提升手术场景分割精度
Surgical Scene Segmentation by Transformer With Asymmetric Feature Enhancement
- 设计不对称特征增强模块,融合局部细节与多尺度信息
- 在多个手术分割数据集上超越现有最佳方法,尤其擅长细粒度结构识别
- 适合关注手术视觉理解与医学图像分割的研究者
手术场景分割是机器人辅助腹腔镜手术理解的基础任务,包含多种解剖结构和手术器械,相似的局部纹理与细微结构使分割极具挑战。基于视觉的变换器方法虽有潜力,但仍面临两大问题:一是缺乏内部块间信息融合,导致性能不佳;二是未能针对性建模解剖结构与器械的特性。为此,本文提出一种新型基于变换器的框架TAFE,通过不对称特征增强模块,先强化局部信息,再以多尺度交互注意力策略将优化后的特征金字塔主动融入变换器编码器的嵌入表示中。该方法在多个不同手术分割任务中均优于当前最优方法,并展现出优异的细粒度结构识别能力。代码已开源:https://github.com/cyuan-sjtu/ViT-asym。
原文摘要 · Abstract (English)
Surgical scene segmentation is a fundamental task for robotic-assisted laparoscopic surgery understanding. It often contains various anatomical structures and surgical instruments, where similar local textures and fine-grained structures make the segmentation a difficult task. Vision-specific transformer method is a promising way for surgical scene understanding. However, there are still two main challenges. Firstly, the absence of inner-patch information fusion leads to poor segmentation performance. Secondly, the specific characteristics of anatomy and instruments are not specifically modeled. To tackle the above challenges, we propose a novel Transformer-based framework with an Asymmetric Feature Enhancement module (TAFE), which enhances local information and then actively fuses the improved feature pyramid into the embeddings from transformer encoders by a multi-scale interaction attention strategy. The proposed method outperforms the SOTA methods in several different surgical segmentation tasks and additionally proves its ability of fine-grained structure recognition. Code is available at https://github.com/cyuan-sjtu/ViT-asym.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。