基于视觉基础模型的多任务框架,提升内镜手术识别与分割精度
EndoARSS: Adapting Spatially-Aware Foundation Model for Efficient Activity Recognition and Semantic Segmentation in Endoscopic Surgery
- 以DINOv2为基础,融合低秩适配与空间感知注意力机制
- 在三个新数据集上实现超越现有模型的准确率与鲁棒性
- 适合医疗AI研究者及机器人手术系统开发者参考
内镜手术是机器人辅助微创手术的金标准,具有早期疾病检测和精准干预的优势。然而,手术场景复杂,不同手术阶段变化大,目标与背景特征混淆,传统深度学习模型易受跨任务干扰,导致下游任务性能不佳。为此,本文提出面向内镜手术活动识别与语义分割的多任务学习框架EndoARSS。该框架基于DINOv2基础模型,采用低秩适配实现高效微调,并引入任务高效的共享低秩适配器缓解不同任务间的梯度冲突;同时设计空间感知多尺度注意力模块,通过全局空间信息的跨域学习增强特征表达区分度。为评估效果,我们构建了三个新数据集:MTLESD、MTLEndovis和MTLEndovis-Gen,均包含活动识别与语义分割的精细标注。大量实验表明,EndoARSS在多个基准上表现优异,显著优于现有模型,验证了其在提升内镜手术AI系统安全性与效率方面的潜力。
原文摘要 · Abstract (English)
Endoscopic surgery is the gold standard for robotic-assisted minimally invasive surgery, offering significant advantages in early disease detection and precise interventions. However, the complexity of surgical scenes, characterized by high variability in different surgical activity scenarios and confused image features between targets and the background, presents challenges for surgical environment understanding. Traditional deep learning models often struggle with cross-activity interference, leading to suboptimal performance in each downstream task. To address this limitation, we explore multi-task learning, which utilizes the interrelated features between tasks to enhance overall task performance. In this paper, we propose EndoARSS, a novel multi-task learning framework specifically designed for endoscopy surgery activity recognition and semantic segmentation. Built upon the DINOv2 foundation model, our approach integrates Low-Rank Adaptation to facilitate efficient fine-tuning while incorporating Task Efficient Shared Low-Rank Adapters to mitigate gradient conflicts across diverse tasks. Additionally, we introduce the Spatially-Aware Multi-Scale Attention that enhances feature representation discrimination by enabling cross-spatial learning of global information. In order to evaluate the effectiveness of our framework, we present three novel datasets, MTLESD, MTLEndovis and MTLEndovis-Gen, tailored for endoscopic surgery scenarios with detailed annotations for both activity recognition and semantic segmentation tasks. Extensive experiments demonstrate that EndoARSS achieves remarkable performance across multiple benchmarks, significantly improving both accuracy and robustness in comparison to existing models. These results underscore the potential of EndoARSS to advance AI-driven endoscopic surgical systems, offering valuable insights for enhancing surgical safety and efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。