cs.CVApr 10, 2026

Benchmarking CNN- and Transformer-Based Models for Surgical Instrument Segmentation in Robotic-Assisted Surgery

Authors: Sara Ameli

Abstract

Accurate segmentation of surgical instruments in robotic-assisted surgery is critical for enabling context-aware computer-assisted interventions, such as tool tracking, workflow analysis, and autonomous decision-making. In this study, we benchmark five deep learning architectures-UNet, UNet, DeepLabV3, Attention UNet, and SegFormer on the SAR-RARP50 dataset for multi-class semantic segmentation of surgical instruments in real-world radical prostatectomy videos. The models are trained with a compound loss function combining Cross Entropy and Dice loss to address class imbalance and capture fine object boundaries. Our experiments reveal that while convolutional models such as UNet and Attention UNet provide strong baseline performance, DeepLabV3 achieves results comparable to SegFormer, demonstrating the effectiveness of atrous convolution and multi-scale context aggregation in capturing complex surgical scenes. Transformer-based architectures like SegFormer further enhance global contextual understanding, leading to improved generalization across varying instrument appearances and surgical conditions. This work provides a comprehensive comparison and practical insights for selecting segmentation models in surgical AI applications, highlighting the trade-offs between convolutional and transformer-based approaches.

Figures & tables

Explore similar work

CardsList
  1. TEDi: Temporal Memory-Enhanced and Denoising Transformer for Surgical Instrument Segmentation

    Sep 15, 2026Jiahong Yuan, Weiming Mi, Tao Zhang +1Video Object SegmentationMedical Image Segmentation

  2. Benchmarking CNN-based Models against Transformer-based Models for Abdominal Multi-Organ Segmentation on the RATIC Dataset

    Mar 19, 2026Lukas Bayer, Sheethal Bhat, Andreas MaierMedical Image SegmentationHybrid CNN-Transformer Architectures

  3. SIRA: Reasoning-Aware Surgical Instrument Segmentation via Query-Anchored Alignment

    Sep 18, 2026Zhibo Zhang, Qijie Wang, Zengqiang YanMedical Image SegmentationReasoning Segmentation