arXiv:2510.18213cs.CV2025-10

用指数移动平均稳定超声视频中的肿瘤分割,提升精度与实时性。

EMA-SAM: Exponential Moving-average for SAM-based PTMC Segmentation

  • 引入置信加权指数移动平均指针,保持肿瘤特征在帧间的连续性。
  • 在PTMC-RFA数据集上,Dice提升至0.86,误报减少29%。
  • 轻量级设计仅增加0.1%计算量,保持30FPS实时推理速度。

乳头状甲状腺微小癌(PTMC)越来越多地采用射频消融(RFA)治疗,但超声视频中因对比度低、探头运动和热伪影导致病灶分割困难。近期的通用图像分割模型SAM-2虽在静态图像表现良好,但在介入超声中因帧间独立设计出现预测不稳定与时间漂移。本文提出EMA-SAM,通过在记忆库中引入置信加权指数移动平均指针,构建稳定的肿瘤潜在原型。该设计在探头压力变化和气泡遮挡下仍能保持时间一致性,并在清晰证据重现时快速适应。在自建的PTMC-RFA数据集(124分钟,13名患者)上,EMA-SAM将最大Dice从0.82提升至0.86,最大IoU从0.72升至0.76,同时降低29%误报率。在外部基准如VTUS和结肠镜息肉视频数据集上,相较SAM-2实现2–5个Dice点的持续提升。重要的是,EMA指针引入的计算开销低于0.1%,在单张A100 GPU上保持约30FPS的实时吞吐能力。这些结果表明EMA-SAM是应对介入超声严苛需求的鲁棒高效框架。

原文摘要 · Abstract (English)

Papillary thyroid microcarcinoma (PTMC) is increasingly managed with radio-frequency ablation (RFA), yet accurate lesion segmentation in ultrasound videos remains difficult due to low contrast, probe-induced motion, and heat-related artifacts. The recent Segment Anything Model 2 (SAM-2) generalizes well to static images, but its frame-independent design yields unstable predictions and temporal drift in interventional ultrasound. We introduce \textbf{EMA-SAM}, a lightweight extension of SAM-2 that incorporates a confidence-weighted exponential moving average pointer into the memory bank, providing a stable latent prototype of the tumour across frames. This design preserves temporal coherence through probe pressure and bubble occlusion while rapidly adapting once clear evidence reappears. On our curated PTMC-RFA dataset (124 minutes, 13 patients), EMA-SAM improves \emph{maxDice} from 0.82 (SAM-2) to 0.86 and \emph{maxIoU} from 0.72 to 0.76, while reducing false positives by 29\%. On external benchmarks, including VTUS and colonoscopy video polyp datasets, EMA-SAM achieves consistent gains of 2--5 Dice points over SAM-2. Importantly, the EMA pointer adds \textless0.1\% FLOPs, preserving real-time throughput of $\sim$30\,FPS on a single A100 GPU. These results establish EMA-SAM as a robust and efficient framework for stable tumour tracking, bridging the gap between foundation models and the stringent demands of interventional ultrasound. Codes are available here \hyperref[code {https://github.com/mdialameh/EMA-SAM}.

医学影像视频分割SAM实时分割

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。