arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.02004cs.CVcs.AIcs.SDeess.AS
收藏

SAVE:利用分割一切模型实现音视频分割的简易方法

SAVE: Segment Audio-Visual Easy way using Segment Anything Model

National Cancer Center(国家癌症中心)

Khanh-Binh Nguyen, Chae Jung Park

AI总结:

本文提出轻量级方法SAVE,通过图像编码器适配器和残差音频编码器适配器将SAM高效适配到音视频分割任务,在降低输入分辨率至256像素的同时超越SOTA性能。

摘要:

音视频分割(Audio-Visual Segmentation, AVS)的主要目标是通过在像素级别准确预测分割掩码,精确识别和定位视觉场景中的听觉元素。实现这一目标需要从数据和模型两个方面综合考虑,以有效解决该任务。本研究提出了一种轻量级方法 SAVE,该方法将预训练的分割一切模型(Segment Anything Model, SAM)高效地适配到 AVS 任务中。通过在 Transformer 模块中引入图像编码器适配器以更好地捕捉不同数据集的信息,并提出一种残差音频编码器适配器将音频特征编码为稀疏提示,我们提出的模型在编码阶段实现了有效的音视频融合与交互。我们提出的方法通过将输入分辨率从 1024 像素降低到 256 像素,加快了训练和推理速度,同时与之前的 SOTA 相比取得了更高的性能。大量实验验证了我们的方法,表明我们提出的模型显著优于其他 SOTA 方法。此外,利用在合成数据上预训练的模型提升了在真实 AVSBench 数据上的性能,在输入图像仅为 256 像素的情况下,在 S4(V1S)子集上达到 84.59 mIoU,在 MS3(V1M)集合上达到 70.28 mIoU。当输入为 1024 像素时,性能进一步提升至 S4(V1S)上的 86.16 mIoU 和 MS3(V1M)上的 70.83 mIoU。

英文摘要:

The primary aim of Audio-Visual Segmentation (AVS) is to precisely identify and locate auditory elements within visual scenes by accurately predicting segmentation masks at the pixel level. Achieving this involves comprehensively considering data and model aspects to address this task effectively. This study presents a lightweight approach, SAVE, which efficiently adapts the pre-trained segment anything model (SAM) to the AVS task. By incorporating an image encoder adapter into the transformer blocks to better capture the distinct dataset information and proposing a residual audio encoder adapter to encode the audio features as a sparse prompt, our proposed model achieves effective audio-visual fusion and interaction during the encoding stage. Our proposed method accelerates the training and inference speed by reducing the input resolution from 1024 to 256 pixels while achieving higher performance compared with the previous SOTA. Extensive experimentation validates our approach, demonstrating that our proposed model outperforms other SOTA methods significantly. Moreover, leveraging the pre-trained model on synthetic data enhances performance on real AVSBench data, achieving 84.59 mIoU on the S4 (V1S) subset and 70.28 mIoU on the MS3 (V1M) set with only 256 pixels for input images. This increases up to 86.16 mIoU on the S4 (V1S) and 70.83 mIoU on the MS3 (V1M) with inputs of 1024 pixels.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑