← 返回 2026-05-18

AuralSAM2: 通过金字塔音视觉特征提示让 SAM2 听见声音 AuralSAM2: Enabling SAM2 Hear Through Pyramid Audio-Visual Feature Prompting

Yuyuan Liu, Yuanhong Chen, Chong Wang, Junlin Han, Junde Wu, Can Peng, Jingkun Chen, Yu Tian, Gustavo Carneiro 📅 2026-05-14 👍 3 2026-07-13 08:36
SAM2 多模态融合 对比学习 视频分割 音频视觉分割

用外部 AuralFuser 模块给 SAM2 注入音频,金字塔提示避免稀释。

前置知识

Segment Anything Model 2 (SAM2)

Meta 提出的视频分割基础模型,扩展了 SAM 的人机交互提示式分割范式,支持点/框等提示在视频帧间传播目标遮罩。其核心由图像编码器 $h_\phi^{SAM2}$、记忆库和带双向交叉注意力的遮罩解码器 $g_\phi^{SAM2}$ 组成。

本文在 SAM2 上扩展音频模态,必须理解其提示工程和冻结骨干的推理流程,才能看出本文如何在不破坏原范式的前提下加入音频。

Audio-Visual Segmentation (AVS)

音视觉分割任务,要求模型在给定音频信号和视频帧的条件下,逐帧输出发声物体的像素级二值遮罩。代表基准 AVSBench 包含单源 (V1s) 与多源 (V1m) 子集,扩展的 V2 加入 70 个语义类别。

AVS 是 AuralSAM2 的主战场,论文 75.58 MJ / 84.12 MF(V1m,Hiera-l)等关键指标都来自该基准。

提示工程 (Prompt Engineering) 与提示稀释 (Prompt Dilution)

提示工程指在 SAM2 中给定点/框等空间提示以引导分割的人机交互流程。论文提出的「音频提示稀释」特指音频信号在 backbone 逐层传播中,跨注意力强度从浅层 ~0.1 衰减到深层约 0.01 的现象。

这是文章技术创新的立足点:现有 adapter 方法无法解决稀释问题,AuralSAM2 的稀疏+密集提示正是为缓解这一现象设计。

对比学习 (Contrastive Learning) 与 InfoNCE

通过最大化正样本对相似度、最小化负样本对相似度来学习表征。InfoNCE 损失形式为 $\ell_{ctrs} = -\log \frac{\exp(e \cdot e^+/\tau)}{\sum \exp(e \cdot e^-/\tau)}$,$\tau$ 为温度参数。

AudioCon 是基于 InfoNCE 框架的改造,它以音频为锚点而非对称对齐,是本文核心损失之一。

特征金字塔网络 (FPN)

通过自顶向下路径与横向连接 (Lateral Layer) 将不同尺度的特征融合,生成多尺度表征。在 SAM2 编码器中早期层对应小尺寸 patch embedding,浅层捕获边缘,深层捕获语义。

AuralFuser 在 $k=3$ 层金字塔上对齐音频与视觉,是缓解音频提示稀释的关键结构设计。

研究动机

当前将音频模态融入 SAM2 的两条路线都存在严重缺陷。第一条路线(AL-REF、SAM4AVS)依赖 GPT-4V/Grounded-SAM 等大型基础模型把音频转成文本再生成框提示,结果准确率低(V1m 上仅 48.6 MJ)且推理慢,因为每帧都要调外部模型。第二条路线(GAVS、SAMA-AVS)在图像编码器或解码器中插入 adapter 做音视觉融合,在 AVSBench (V1m) 上可达 67.7 MJ,但需要两次 SAM2 前向:在 prompt 工程场景中,已被 adapter 改写的图像特征必须与人类提供的新点/框提示重新融合,导致 ensemble 比纯 AVS 慢约 6.5 FPS(在 V1m 上 FPS 从 17.8 降至 8.7–9.9)。更深层的问题是「音频提示稀释」:每个视频片段产生 $> 10^6$ 个稠密视觉 token,而音频只贡献约 10 个粗 embedding,跨注意力信号随网络加深不断衰减,最后变成纯视觉主导,分割结果不再受音频调控。

本文的目标是本文目标是在不破坏 SAM2 原有人机交互提示范式的前提下,让 SAM2 真正「听见」声音,并在 prompt engineering 场景下也能高效工作。具体指标上要在 AVSBench (V1m) 上显著超越 GAVS、SAMA-AVS 等 adapter-based SOTA(Jaccard 提升 3.9%、MF 提升 4.6 个点),同时把引入音频后的 FPS 衰减控制在约 2.3 帧(从 17.4 降至 14.1),同时具备为 SAM (HQ-SAM) 注入同类能力时的泛化性。

与已有工作不同的是,已有方法都把音频当作「伪视觉」或「次要信号」——要么翻译成框、要么塞进 adapter 改写图像特征。作者从一条新视角切入:保留 SAM2 的图像 encoder 完全冻结,把音频当作独立的特征级提示源,通过外部 AuralFuser 模块跨多尺度金字塔同时生成稀疏(全局上下文)和稠密(像素级定位)两组提示,并在最后一个 stage 通过 AudioCon 让视觉特征围绕音频聚拢,缓解视觉主导。这一设计首次把「不修改 SAM2 视觉特征、又能被人类提示调用」两件事同时达成。

核心方法

AuralSAM2 = 冻结的 SAM2(Hiera-b+ 或 Hiera-l backbone) + 外部 AuralFuser 模块。直觉上可以把它理解为:让 SAM2 在原本接收「点/框」提示的入口处,再多接收两个从音频派生的提示流(稀疏提示 + 稠密提示),同时不影响图像特征也不需要二次前向。具体地,AuralFuser 用 VGGish 压缩波形、用 RoBERTa 编码可选文本、在 SAM2 三层 Hiera 特征上用不同 patch size 做 patch embedding 构造金字塔;随后每层独立做 self-attention 与 cross-attention 融合(融合块来自 TPAVI 改造),最后通过公式 (5) 把稀疏提示注入 mask token、稠密提示注入 dense embedding。视觉特征 + 音频表征不被强制对齐为相同 token 数,而是保留原本的极端不平衡,再用 AudioCon 把视觉特征朝音频锚点拉齐。训练时 SAM2 与 RoBERTa 完全冻结。

与 adapter-based 方法本质区别有三:(i) 图像编码器不被改写,因此 promptable 分割一次前向即可完成,不需要 ensemble;(ii) 提示信号分两路提供——稀疏提示 $r_a^{(k)} \in \mathbb{R}^{B \times L}$ 用 Select_a 从音-文融合特征中抽取,编码全局上下文;稠密提示 $r_v^{(k)} \in \mathbb{R}^{B \times H' \times W' \times L}$ 直接叠加到 SAM2 dense embedding 上保留像素级粒度;(iii) 引入 AudioCon 把视觉特征作为 query、音频作为 prototype+anchor,对抗视觉 embedding 之间互相吸引造成的「同质化」。

方法步骤详情

方法分四步。(1) 编码:VGGish 压音频到 $z_a\in\mathbb{R}^{B\times L}$,RoBERTa 可选处理文本;Q-pooling 取 SAM2 三层 $\{z_v^{(k)}\}$,合并 $z_c=[z_a,z_t]$。(2) 金字塔:$k\in\{1,2,3\}$ 用 patch size $p^{(k)}\in\{4,2,1\}$ 做 embedding,1x1 卷积自顶向下累积。(3) 融合:每层 self-attention,再用基于 TPAVI 改造的 CrossFusion 输出稀疏提示 $R_a$ 与稠密提示 $R_v$。(4) 注入:按公式 (5) 叠加 $\tilde{p}_{mask}^k=p_{mask}^k+r_a^{(k)}$、$\tilde{p}_d^k=p_d^k+r_v^{(k)}$ 后送入 $g_\phi^{SAM2}$。训练目标 $L=\ell_{SAM2}+\ell_{ctrs}$,$\ell_{SAM2}$ 按 20:1:1:1 加权,AudioCon $\tau=0.10$。

技术新颖性

技术新颖性体现在三个层次。第一是「外部提示」范式:把 AuralFuser 挂在 SAM2 之外,可在不修改图像 encoder 的前提下给 mask decoder 提供音频条件,等价于把音频变成 SAM 一类原生提示,避免了 adapter-based 方法必须二次前向的代价(在 V1m prompt engineering 上 FPS 14.1 vs SAMA-AVS 9.9)。第二是金字塔提示结构:通过 $k=1,2,3$ 三层独立融合 + 1x1 平滑卷积逐层累积,使音频信号在 early/middle/late 各层都保持非零注意力,与 Fig. 4 中显示的概率密度曲线从 SAMA-AVS 的 $\approx 0.01$ 移到 AuralSAM2 的 $\approx 0.075$ 一致。第三是 AudioCon:打破传统「视觉-视觉+音频-音频」对称对齐,让视觉 embedding 仅被音频锚点吸引而忽略视觉-视觉的相互吸引,从机制上对抗视觉 token 数量级压倒音频的本质不平衡。

Illustration of AuralSAM2 architecture on language-aided AVS dataset
Figure 3: Illustration of AuralSAM2 architecture on language-aided AVS dataset

实验结果

AVSBench V1m (Hiera-l) $\mathcal{M}_\mathcal{J}$ 75.58、$\mathcal{M}_\mathcal{F}$ 84.12、J&F 79.85,比 SAMA-AVS (63.1/69.1) J&F 高约 10 点,比 GAVS (Hiera-b+) $\mathcal{M}_\mathcal{J}$ 高 4.34。Ref-AVS Seen (Hiera-l) J&F 58.68,Unseen J&F 71.53 创 SOTA、超 SAM2-LOVE 69.4。Tab. 6 中 audio+points+box $\mathcal{M}_\mathcal{J}$ 74.26、$\mathcal{M}_\mathcal{F}$ 83.58、FPS 14.1,仅比纯点+框慢 3.3 FPS,远优于 GAVS 的 8.7 FPS。消融显示 AudioCon +1.25 J&F,金字塔 $k=1\to 3$ 累计 +3.05 J&F,dense prompts 单独去除损失最大 (-11.61)。

Comparison with SOTA on Ref-AVS
Table 1: Comparison with SOTA on Ref-AVS
Comparison with SOTA on AVSBench
Table 2: Comparison with SOTA on AVSBench
Component-wise ablation on AVSBench and Ref-AVS
Table 3: Component-wise ablation on AVSBench and Ref-AVS
Ablation on contrastive learning
Table 4: Ablation on contrastive learning
Ablation on prompts
Table 5: Ablation on prompts
PDF of cross-attention intensity between audio cues and pixels on AVSBench (V1m)
Figure 4: PDF of cross-attention intensity between audio cues and pixels on AVSBench (V1m)
Qualitative visualisations on Ref-AVS
Figure 5: Qualitative visualisations on Ref-AVS
查看结构化数据
任务指标本文基线提升
AVSBench V1m (多源) MJ / MF / J&F (%) 75.58 / 84.12 / 79.85 (Hiera-l) SAMA-AVS 63.10 / 69.10 (Hiera-h);GAVS 67.70 / 78.93 (Hiera-b+) +10 J&F vs SAMA-AVS;+1.7 J&F vs GAVS(Hiera-b+)
AVSBench V1s (单源) MJ / MF (%) 86.62 / 93.34 (Hiera-l) AVSegFormer 82.10 / 89.90;GAVS 80.10 / 90.20 MJ +4.52 / MF +3.44
AVSBench V2 (semantic) MJ / MF (%) 50.57 / 56.03 (Hiera-l) Stepping Stones 48.5 / 53.2 MJ +2.07 / MF +2.83
Ref-AVS Seen MJ / MF / J&F (%) 56.16 / 61.19 / 58.68 (Hiera-l) SAMA-AVS 39.2 / 56.2 / 47.7 (Hiera-h);TSAM 43.4 / 56.8 / 50.1 +8.6 J&F vs SAMA-AVS;+8.6 J&F vs TSAM
Ref-AVS Unseen MJ / MF / J&F (%) 68.69 / 74.36 / 71.53 (Hiera-l) SAM2-LOVE 66.5 / 72.3 / 69.4;EEMC 49.54 / 64.8 / 57.2 +2.1 J&F vs SAM2-LOVE
Prompt Engineering (V1m, audio+points+box) MJ / MF / FPS 74.26 / 83.58 / 14.1 SAMA-AVS 69.74 / 80.97 / 9.9;GAVS 71.70 / 81.94 / 8.7 FPS +4.2/+5.4,MJ +2.56/+4.52

局限与改进

作者明确指出几项边界。第一,AudioCon 经验设定温度 $\tau=0.10$,不同 backbone 是否需要重新调优未系统回答。第二,Hiera-b+ 到 Hiera-l 的 4.12 点 Jaccard 提升建立在更多参数之上,并未给出 Hiera-l 训练总时长/GPU 数。第三,方法只在二元或语言引用 AVS 上评测,未在 source separation、sound event detection 等更广 AVS 任务上验证。第四,promptable 结果(Tab. 6)使用 ground-truth 派生的点框,与真人点击的实际分布存在差异。我的额外观察:稀疏提示作用于 mask token 时与 IoU/object token 共享 self-attention 路径,但消融并未分别屏蔽这两条路径,可能遗漏贡献来源;t-SNE 可视化放在 Supp. Section 4.1,主文未给定量证据。

独立分析的弱点

独立分析三个潜在弱点。(i) Ref-AVS Unseen 子集 $\mathcal{S}$ (Null 错误率) 仍为 0.065,模型对「参考对象不存在」的判断仍有改进空间,可加 null-class 二分类头或在 AudioCon 中加入 null-aware 负采样。(ii) AudioCon 公式 (8) 中负样本仅从视觉侧采样,未明确处理多源 AVS (V1m) 中多发声物体并发,正样本集合 $E_a$ 包含全部 frame embedding,可能导致视觉特征被「和声」拉向平均锚点;可改为 per-source 聚类后做 prototype contrastive。(iii) 金字塔深度 $k=3$ 累计带来 +3.05 J&F 但也增加 $3 \times H' \times W' \times L$ 的稠密提示存储,对 64-frame 视频推理显存压力未报告,可能在边缘设备部署困难。改进方向:使用 top-k 选取最显著稠密 token 或对 $r_v^{(k)}$ 做 channel pruning,并在 AudioCon 中显式处理 Null 与多源场景。

未来方向

作者在文末未单独列 future work,但结合结果可自然延伸三条线:(a) 把 AuralFuser 推广到其它 SAM 系列(如 SAM 3、EfficientSAM、SAM2Long)以及 4D 时空场景,借助提示注入的「外部化」特性避免再次 fine-tune 整个图像 encoder;(b) 引入 promptable audio input——目前音频以整段波形喂入,未来可以让用户在提示中圈选声源时间区间,让 AuralFuser 在 temporal pyramid 上也具备 prompted 能力;(c) 联合训练 AudioCon 与 promptable segmentation,把 promptable 推理 $\to$ 微调 $\to$ 部署的人类反馈闭环打通,配合 Online RLHF 让稀疏提示随用户偏好演化。此外,可以把 V2 semantic 任务的 +2.07 MJ 改进做更系统的「开放词汇音视觉分割」评测。

复现评估

复现门槛中等偏上。代码开放在 https://github.com/yyliu01/AuralSAM2,依赖 SAM2 (Hiera-b+/l)、VGGish、RoBERTa;Adam 优化、学习率 $1e-4$,poly decay $(1-iter/max_{iter})^{0.9}$,单 batch=1。损失权重 20:1:1:1 与 $\tau=0.10$ 已明确。主要未知项:(1) Hiera-l 总 epoch/GPU 时长与 GPU 型号;(2) Q-pooling 窗口与 mixed precision 等预处理细节;(3) AudioCon 负样本的 background 过滤阈值。若公开 config.yaml,单卡 A100 80G 训练 Hiera-b+ 预期 1–3 天;Supp. Sec. 3 提到其他 backbone 结果,暗示作者至少跑了 PVT-v2、Swin-b 等多套。