用手术记忆增强时序建模,提升长视频手术阶段识别准确率
MoSFormer: Augmenting Temporal Context with Memory of Surgery for Surgical Phase Recognition
- 引入手术长期记忆与短期印象融合机制,突破固定窗口限制
- 在BernBypass70上达88.0%视频级准确率,相位级F1达66.3
- 适合需要全流程理解的手术视频分析场景
从视频中进行手术阶段识别可支持多种下游应用。基于Transformer的滑动窗口方法虽能捕捉丰富时空特征,但受限于内存与计算资源,实际使用中仍需固定上下文窗口,难以维持长手术过程中的时序一致性,常导致预测碎片化和整体理解不足。为此,我们提出手术记忆(MoS)框架,通过融合语义可解释的长期手术历史与短期印象来增强时序建模。MoSFormer采用精心设计的编码与融合机制集成MoS,引入步骤过滤优化历史表示,并构建记忆缓存流水线以提升训练与推理稳定性,缓解捷径学习与过拟合问题。在多个基准测试中表现优异:在挑战性伯恩绕道70(BernBypass70)数据集上,达到88.0%视频级准确率,相位级精度70.7、召回率68.7、F1分数66.3,显著优于基线模型(+2.1%视频级准确率,相位级指标提升分别为+4.6、+3.6、+3.8)。消融与反事实推理验证了长短记忆组件的独立与协同增益。定性结果表明时序一致性明显改善,增强的时序上下文实现了全过程理解,为更全面的手术视频分析铺平道路。
原文摘要 · Abstract (English)
Surgical phase recognition from video enables various downstream applications. Transformer-based sliding window approaches have set the state-of-the-art by capturing rich spatial-temporal features. However, while transformers can theoretically handle arbitrary-length sequences, in practice they are limited by memory and compute constraints, resulting in fixed context windows that struggle with maintaining temporal consistency across lengthy surgical procedures. This often leads to fragmented predictions and limited procedure-level understanding. To address these challenges, we propose Memory of Surgery (MoS), a framework that enriches temporal modeling by incorporating both semantic interpretable long-term surgical history and short-term impressions. MoSFormer, our enhanced transformer architecture, integrates MoS using a carefully designed encoding and fusion mechanism. We further introduce step filtering to refine history representation and develop a memory caching pipeline to improve training and inference stability, mitigating shortcut learning and overfitting. MoSFormer demonstrates state-of-the-art performance on multiple benchmarks. On the Challenging BernBypass70 benchmark, it attains 88.0 video-level accuracy and phase-level metrics of 70.7 precision, 68.7 recall, and 66.3 F1 score, outperforming its baseline with 2.1 video-level accuracy and phase-level metrics of 4.6 precision, 3.6 recall, and 3.8 F1 score. Further studies confirms the individual and combined benefits of long-term and short-term memory components through ablation and counterfactual inference. Qualitative results shows improved temporal consistency. The augmented temporal context enables procedure-level understanding, paving the way for more comprehensive surgical video analysis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。