arXiv:2503.06700cs.CV2025-03被引 14

用记忆机制让SAM2同时理解多模态和语义,提升分割精度。

MemorySAM: Memorize Modalities and Semantics with Segment Anything Model 2 for Multi-modal Semantic Segmentation

  • 将多模态数据视为序列帧,用记忆机制提取共性特征。
  • 在真实与合成数据集上分别达到65.38%和52.88%的准确率。
  • 适合做跨模态语义分割的研究者和工程师参考。

多模态语义分割(MMSS)旨在从多种传感器获取的视觉模态中进行像素级预测。近期大视觉模型Segment Anything Model 2(SAM2)在图像和视频上展现出强大的零样本分割能力。将其扩展至MMSS时面临两大挑战:如何适配多模态数据?如何更好理解语义?受视频中跨帧相关性的启发,我们提出将多模态数据视为同一场景的连续帧序列。核心思想是“记忆”模态无关信息和“记忆”与目标场景相关的语义。为此,我们在多模态数据上应用SAM2的记忆机制以捕获模态无关特征;同时设计仅在训练阶段使用的语义原型记忆模块(SPMM),在训练过程中存储类别级原型,帮助SAM2从实例分割过渡到语义分割。通过迭代施加全局与局部原型间的原型适应损失,对齐并优化其语义理解。大量实验表明,所提方法MemorySAM在合成与真实世界基准上均显著优于现有最先进方法(在DELIVER上达65.38%,在MCubeS上达52.88%)。代码将公开。

原文摘要 · Abstract (English)

Research has focused on Multi-Modal Semantic Segmentation (MMSS), where pixel-wise predictions are derived from multiple visual modalities captured by diverse sensors. Recently, the large vision model, Segment Anything Model 2 (SAM2), has shown strong zero-shot segmentation performance on both images and videos. When extending SAM2 to MMSS, two issues arise: 1. How can SAM2 be adapted to multi-modal data? 2. How can SAM2 better understand semantics? Inspired by cross-frame correlation in videos, we propose to treat multi-modal data as a sequence of frames representing the same scene. Our key idea is to ''memorize'' the modality-agnostic information and 'memorize' the semantics related to the targeted scene. To achieve this, we apply SAM2's memory mechanisms across multi-modal data to capture modality-agnostic features. Meanwhile, to memorize the semantic knowledge, we propose a training-only Semantic Prototype Memory Module (SPMM) to store category-level prototypes across training for facilitating SAM2's transition from instance to semantic segmentation. A prototypical adaptation loss is imposed between global and local prototypes iteratively to align and refine SAM2's semantic understanding. Extensive experimental results demonstrate that our proposed MemorySAM outperforms SoTA methods by large margins on both synthetic and real-world benchmarks (65.38% on DELIVER, 52.88% on MCubeS). Source code will be made publicly available.

多模态分割SAM2语义记忆原型学习

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