EMA-SAM: Exponential Moving-average for SAM-based PTMC Segmentation
Not on the map. Why?
Abstract (arXiv)
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 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 maxDice from 0.82 (SAM-2) to 0.86 and 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 ~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.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 23 of its 34 references that are in the map. First posted on arXiv 21 Oct 2025.
Builds on
23 of 34 references.
References · 23
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2023Segment Anythingcited by 8,158
- 2016Feature Pyramid Networks for Object Detectioncited by 5,021
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2024SAM 2: Segment Anything in Images and Videoscited by 2,784
- 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,518
- 2017Grad-CAM++: Improved Visual Explanations for Deep Convolutional Networkscited by 936
- 2023Medical SAM Adapter: Adapting Segment Anything Model for Medical Image Segmentationcited by 485
- 2020PraNet: Parallel Reverse Attention Network for Polyp Segmentationcited by 460
- 2017Road Extraction by Deep Residual U-Netcited by 333
- 2023Hiera: A Hierarchical Vision Transformer without the Bells-and-Whistlescited by 287
- 2022UniFormer: Unified Transformer for Efficient Spatiotemporal Representation Learningcited by 182
11 more
- 2024Medical SAM 2: Segment medical images as video via Segment Anything Model 2cited by 151
- 2024SAM2Long: Enhancing SAM 2 for Long Video Segmentation with a Training-Free Memory Treecited by 77
- 2021Progressively Normalized Self-Attention Network for Video Polyp Segmentationcited by 75
- 2024SAM2-UNet: Segment Anything 2 Makes Strong Encoder for Natural and Medical Image Segmentationcited by 63
- 2024Vivim: a Video Vision Mamba for Medical Video Segmentationcited by 51
- 2024Surgical SAM 2: Real-time Segment Anything in Surgical Video by Efficient Frame Pruningcited by 46
- 2023Beyond Adapting SAM: Towards End-to-End Ultrasound Image Segmentation via Auto Promptingcited by 40
- 2021LDNet: Unified Listener Dependent Modeling in MOS Prediction for Synthetic Speechcited by 24
- 2021ACSNet: Action-Context Separation Network for Weakly Supervised Temporal Action Localizationcited by 20
- 2024CC-SAM: SAM with Cross-feature Attention and Context for Ultrasound Image Segmentationcited by 11
- 2024Novel adaptation of video segmentation to 3D MRI: efficient zero-shot knee segmentation with SAM2cited by 8
Built on by
Not cited yet.