面向真实声学场景的高效流式音视频目标说话人提取 Efficient Streaming Audio-Visual Target Speaker Extraction for Real-World Acoustic Scenes
发布RealSSA真实声学基准,提出单遍Falcon模型,延迟降11倍精度超SOTA
前置知识
音视频目标说话人提取(AVTSE)
从多人混合语音中分离出指定目标说话人的声音,与一般「语音分离」的区别在于:它利用目标说话人的视觉流(通常是嘴部动作视频)作为注册线索来定位要提取谁,因此只需要唇动视频+混合音频两类输入,不需要预先知道说话人数。经典实现是视觉骨干提取唇部特征、音频编码器提取时频特征,融合后预测掩码作用于混合谱。
本文的全部工作都建立在 AVTSE 任务设定上,RealSSA 基准和 Falcon 模型分别针对这个任务的数据与模型两端。
STFT 编码与复数掩码
把波形经短时傅里叶变换(STFT)映射到时频域,得到每个时频点的复数(实部+虚部)表示。复数掩码方法让网络预测一个同样形状的复数掩码 $M$,与编码特征做复数乘法(实部相乘相减、虚部相乘相加),再经逆变换重建波形。相比只预测幅度掩码,复数掩码能修正相位,是现代语音分离的主流做法。
Falcon 的音频编码器、掩码生成器和解码器都围绕 STFT 复数谱展开,其对比损失也直接在 STFT 编码器空间计算。
流式(因果)推理与算法前瞻
流式系统逐帧处理音频、边录边出结果,模型只能使用当前帧及之前的信息(因果性);若模型设计上允许回看未来的帧,就构成「算法前瞻」(lookahead)。固定前瞻量决定了端到端延迟下限。流式化通常要求:单向时序模型、左对齐填充的卷积、免缓存的归一化,且下采样次数受前瞻预算约束。
本文面向视频会议、助听设备等延迟敏感场景,视觉分支 120ms 的固定前瞻直接限制了音频侧最多 15 帧(8ms/帧)前瞻,从而决定了 Separator 的最大深度。
SI-SNR 与 SI-SNRi
尺度不变信噪比(SI-SNR)先对估计波形和参考波形做尺度对齐(投影去除幅度差异),再计算残差信噪比,是语音分离的标准重建指标;SI-SNRi 是分离结果相对混合输入的 SI-SNR 改善量(单位 dB),越高越好。它衡量「离干净目标有多近」,但不显式惩罚「离干扰有多近」。
SI-SNR 只拉近估计与目标的这一盲区,正是本文提出编码器空间说话人对比损失的动机;论文主表全部以 SI-SNRi/SDRi 为核心指标。
近场/远场干扰与混响 RT60
近场指声源距麦克风较近(本文取 3m 内),直达声占主导;远场声源经房间反射后直达声弱、混响能量占主导(3m 约对应典型室内的临界距离)。RT60 是混响时间:声源停止后声能衰减 60dB 所需时间,小会议室约 0.3–0.6s,大厅可达 2s。真实录音中近/远场干扰、混响、环境噪声与突发噪声同时存在。
RealSSA 数据集的核心设计就是让这些声学因素联合变化;对比损失也刻意只用近场干扰者做负样本,理解近/远场物理才能明白这一取舍。
研究动机
音视频目标说话人提取(AVTSE)在视频会议、助听设备等延迟敏感场景需求迫切,但现有因果系统存在两个互相耦合的短板。数据侧:主流基准 LRS2-2Mix、LRS3-2Mix、VoxCeleb2-2Mix 把混合固定为两个近场说话人,SNR 受控、房间声学与背景噪声恒定,完全没有近场/远场空间几何、可控混响和突发噪声;LRS3+WHAM! 噪声协议虽加入 SNR 约 -6 到 +3 dB 的背景噪声,但仍保持双说话人、无混响、无突发噪声。所有基准都用同源视频的 held-out 划分做评估,没有任何一个配备真实录音评测集,因此论文里报告的提升很难预测模型部署到真实设备后的表现。模型侧:AV-TFGridNet 这类高性能因果架构有 6.2M 参数、GPU 推理延迟 567.8ms,超出边缘设备实时预算;SwiftNet 这类轻量方案靠反复调用同一小分离器 12 次来恢复容量,0.5M 参数却要 45.3G MACs、GPU 延迟 172.1ms,每多迭代一轮就把省下的计算量加倍还回去。训练目标上,SI-SNR 只把估计拉向干净目标,不显式推开近场干扰者,近场残余泄漏随同时活跃的干扰说话人数增加而恶化。
本文的目标是论文目标分两条线。数据线:构建 RealSSA 基准,让说话人数量、空间几何、环境声学三个维度联合变化而非一次只动一个轴——RealSSA-Sim 提供四种可控仿真场景(小型会议室、中型室内、大型公共场所、户外),近场干扰说话人 1–5 个、含远场干扰的场景再加 10–15 个远场说话人,RT60 覆盖 0.05 到 2.0 秒、SIR 覆盖 -2 到 15 dB,并按 30%–50% 的概率采样突发噪声事件;RealSSA-Real 提供仅用于评测、绝不参与训练的真实手机录音,覆盖室内、公共场所与户外。模型线:提出因果模型 Falcon,在单次编码器–解码器前向内完成多尺度时频分离,配合编码器空间说话人对比损失,目标是分离质量持平或超过最强因果基线(RealSSA-Sim 上超过 AV-TFGridNet 的 7.6 dB SI-SNRi),同时相对 AV-TFGridNet 把参数减少 91.9%、计算量降低 20.2 倍、GPU 推理延迟降低 11.1 倍,达到边缘设备可负担的实时水平。
与已有工作不同的是,本文的独特切入有三点。第一,现有基准一次只变化一个轴,作者强调干扰说话人数、近/远场几何、混响、环境噪声和突发噪声必须联合采样,才贴近真实声学场景的分布;并首次在 AVTSE 领域用真实录音评测集(RealSSA-Real)打通「仿真训练→真机录音评测」的泛化诊断,Table 6 显示仅这一改动就让五个架构的 DNSMOS 全部提升。第二,在效率路线上,作者看清了轻量模型的隐性代价:SwiftNet 用同一紧凑块迭代 12 次换容量,每帧计算被成倍放大,于是改为把频轴建模、时间轴建模和联合时频建模分配到单次 U-Net 前向的不同分辨率层级,一次遍历即得四个时频尺度。第三,在训练目标上,先前抑制残余干扰要么调整训练混合分布、要么在分离器输出端加路由门控,本文第一个直接在共享 STFT 编码器的特征空间里,把估计值从每个近场干扰者的表征显式推开,且该分支训练后即丢弃,推理零成本、可跨骨干迁移。
核心方法
直觉上,Falcon 把「小模型反复跑 12 遍」换成「一遍走完从粗到细的四层时频加工」:如同流水线上按分辨率分工的四道工序,一次通过就完成过去需要多轮迭代才能积累的建模容量。技术路线:输入 16kHz 单通道混合波形 $x = s + \sum_{k=1}^{K} b_k + f + n$(目标语音、$K$ 个近场干扰、远场干扰之和、环境噪声)与目标说话人 96×96 灰度嘴部视频。视觉侧用冻结的 FRCNN 骨干(沿用 CTCNet)提取唇部特征,经视觉瓶颈得到 $V_0 \in \mathbb{R}^{C_v \times T_v}$;音频侧做 STFT(窗长 256、帧移 128,即每帧 8ms),把实部、虚部、幅度拼成三通道张量输入 2D 卷积得到 $A_0 \in \mathbb{R}^{C_e \times T \times N_f}$。两路分别经 AudioBlock(与分离器同构)和 SwiftNet 的 VideoBlock(因果 LSTM 聚合时序并上采样到音频帧率)精炼后,视觉特征沿频率维扩展并与音频特征逐点相加得到融合特征 $F$。Separator(非对称 U-Net,隐藏维 $C_h = 64$)单遍前向把 $F$ 映射为目标特征 $E$,掩码生成器据此预测时频复数掩码 $M$,与 $A_0$ 做复数乘法得到掩码嵌入 $\hat{A}$,最后经音频解码器与 iSTFT 重建目标语音 $\hat{s}$。训练目标为 $\mathcal{L} = -\mathrm{SI\text{-}SNR}(\hat{s}, s) + 1.0 \cdot \mathcal{L}_{spk}$。
核心创新有两个。其一是单遍多尺度 Separator:时频建模集中在编码器瓶颈(时间-频率分辨率均降 8 倍处)形成一个全局场景上下文 $G = R^3(S^3)$,再通过全局上下文注入模块(GCI)$\tilde{S}^l = \phi_{loc}(S^l) \odot \sigma(\phi_{gate}(G)) + \phi_{glb}(G)$ 把粗尺度信息门控地注入每一层跳连,让全局语境强调或抑制特定频带;解码器每层配一个分辨率各异的时频精炼块 TFRB,一次前向即覆盖四个时频尺度,替代 SwiftNet 把同一模块迭代 12 次的做法——同样 0.5M 参数,MACs 从 45.3G 降到 10.6G,GPU 延迟从 172.1ms 降到 51.3ms。其二是编码器空间说话人对比损失:查询是估计目标在编码器空间的表征 $\hat{A}$,正样本是干净目标 $sg(E_a(s))$、负样本是每个近场干扰者 $sg(E_a(b_k))$(远场干扰因残差已低于 SI-SNR 误差底而被排除出负样本集),三者共用同一音频编码器且参考路径停梯度;损失为 $\mathcal{L}_{spk} = 1 - \cos(q, p) + \sum_{k=1}^{K} \frac{\exp(\tau s_k)}{\sum_{j=1}^{K} \exp(\tau s_j)} \mathrm{ReLU}(s_k - m)$,其中 margin $m = 0.5$(对应 60° 角度间隔,估计一旦足够远离干扰排斥项自动失效,剩余残差交给 SI-SNR),温度 $\tau$ 从 0 线性退火到 1,训练结束后整个分支丢弃,推理零开销。
方法步骤详情
完整流程分六步。第一步视觉编码:25fps 的 96×96 灰度嘴部视频输入冻结的 FRCNN,经视觉瓶颈输出 $V_0 \in \mathbb{R}^{C_v \times T_v}$。第二步音频编码:混合波形做 STFT(窗 256、跳 128),实部/虚部/幅度三通道过 2D 卷积得 $A_0 \in \mathbb{R}^{C_e \times T \times N_f}$,再经音频瓶颈投影到统一通道维 $C$。第三步模态精炼与融合:$A$ 过 AudioBlock 得 $\tilde{A}$;$V_0$ 过 VideoBlock(因果 LSTM、上采样到 8ms 音频帧率)得 $\tilde{V} \in \mathbb{R}^{C \times T}$,沿频率轴扩展后与 $\tilde{A}$ 相加得融合特征 $F$。第四步 Separator:深度 1×1 卷积加 PReLU 生成残差支路 $R$;逐点 1×1 卷积投影到 $C_h = 64$ 后经流式因果层归一化 $\mathrm{LN}^c$(逐时频位置、免缓存);编码器 4 级,$D^0$ 为步幅 1 深度 3×3 卷积、$D^1$–$D^3$ 为步幅 2 深度 5×5 卷积(时间轴左对齐填充、频率轴对称填充,无时间前瞻),瓶颈处 TFRB 输出全局上下文 $G$;GCI 把 $G$ 注入各层跳连;解码器按 $l = 2, 1, 0$ 粗到细执行 $Y^l = R^l(\mathrm{Id}^l(\tilde{S}^l, Y^{l+1}) + S^l)$,加 $S^l$ 项做恒等残差保梯度。第五步掩码与重建:掩码生成器输出 $M$,按实/虚拆分后与 $A_0$ 复数相乘得 $\hat{A}$,音频解码器加 iSTFT 重建 $\hat{s}$。第六步训练:$\mathcal{L}_{recon} + 1.0 \cdot \mathcal{L}_{spk}$,对比支路仅训练期激活;AdamW 学习率 $1 \times 10^{-3}$、权重衰减 0.01,验证损失 15 个 epoch 不降则减半,30 个 epoch 早停。
技术新颖性
与已有工作的本质区别在三个层面。架构上,因果 AVTSE 此前只有两条路:重型单块(AV-TFGridNet 6.2M 参数、每帧 213.8G MACs;AV-Mossformer2 56.7M 参数)或紧凑块多次迭代(SwiftNet-12 复用同一块 12 次,0.5M 参数但 45.3G MACs),Falcon 的 level-specific 单遍多尺度是第三条路——四个 TFRB 各自工作在不同分辨率,参数保持 0.5M 而 MACs 仅 10.6G。训练目标上,先前抑制干扰依赖调混合分布或输出端门控,本文首次在共享编码器特征空间做说话人对比,且组合了三个技巧:softmax 加权负样本、60° margin 排斥、温度退火——margin 保证估计足够远离干扰后排斥自动失效,退火让嵌入空间先均匀探索再聚焦最难负样本(固定 $\tau = 1$ 只得 7.4 dB,退火 0→1 得 7.7 dB)。因果设计上,$\mathrm{LN}^c$ 逐时频位置归一化且免缓存;2D 注意力用块因果掩码,允许同一帧内全频段交互而不引入时间前瞻;下采样深度受视觉分支 120ms 固定前瞻约束——音频帧移 8ms,最多 15 帧前瞻,恰好容纳 3 级下采样,第 4 级会破坏音视频对齐。该损失还跨骨干有效:SwiftNet-12 加 0.2 dB、Dolphin 加 0.3 dB。
实验结果
主实验覆盖五个数据集与三类指标。分离质量(Table 3):Falcon 在全部四个模拟基准上 SI-SNRi/SDRi 均列第一——RealSSA-Sim 7.7/8.2 dB(最强基线 AV-TFGridNet 7.6/8.1)、LRS2-2Mix 14.1/14.7、LRS3-2Mix 16.4/16.7(超过 AV-Mossformer2 的 16.2/16.6)、VoxCeleb2-2Mix 10.5/11.8(比 Swift-Net-12 高 1.1 dB);RealSSA-Real 上 DNSMOS 3.276 与 Swift-Net-12 并列第一,明显高于 AV-TFGridNet 的 3.030 和 Dolphin 的 2.960。语音可懂度与说话人一致性(Table 4):RealSSA-Sim WER 25.47% 为所有提取系统最低,比 AV-TFGridNet 的 27.31% 低 1.84 个百分点(干净目标参照 11.83%);RealSSA-Real WER 32.74% 居第二,比 AV-Mossformer2 的 32.00% 差 0.74 点;两组说话人相似度 43.05/42.23 均为最高,超 Swift-Net-12 的 41.79/40.04 达 1.26/2.19 点。效率(Table 5):Falcon 以 0.5M 参数取得 10.6G MACs、GPU 延迟 51.3ms、CPU RAM 125.3MB、GPU 显存 364.7MB,全部指标最优;相对 AV-TFGridNet 参数减 91.9%、MACs 减 95.0%(20.2 倍)、GPU 延迟降 91.0%(11.1 倍),相对 Swift-Net-12 MACs 减 76.6%、GPU 延迟降 70.2%。跨域泛化(Table 6):用 RealSSA-Sim 替换 VoxCeleb2 训练后,五个架构在 RealSSA-Real 上 DNSMOS 全部提升、NISQA 四个提升,Falcon 从 NISQA 2.431→3.615、DNSMOS 2.878→3.276。消融(Table 7–9):下采样深度 1→2→3 级 SI-SNRi 从 7.1→7.4→7.7 dB,3 级是视觉 120ms 前瞻允许的上限;对比损失给 Falcon 加 0.4 dB、SwiftNet-12 加 0.2 dB、Dolphin 加 0.3 dB;$\tau$ 退火 0→1 优于固定 $\tau=0$(7.5 dB)、固定 $\tau=1$(7.4 dB)和退火 0→2(7.5 dB)。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| RealSSA-Sim 目标说话人提取(含 1–5 近场 + 可选 10–15 远场干扰、混响、突发噪声) | SI-SNRi (dB) | 7.7 | 7.6(AV-TF-GridNet) | +0.1 dB,同时参数减少 91.9%、GPU 延迟降低 11.1× |
| LRS3-2Mix | SI-SNRi (dB) | 16.4 | 16.2(AV-Mossformer2) | +0.2 dB(SDRi 16.7 vs 16.6) |
| VoxCeleb2-2Mix | SI-SNRi (dB) | 10.5 | 9.4(Swift-Net-12) | +1.1 dB,为四个基准中相对提升最大 |
| RealSSA-Sim 语音可懂度 | WER (%) | 25.47 | 27.31(AV-TF-GridNet) | -1.84 个百分点(干净目标参照 11.83%) |
| RealSSA-Real 真实录音感知质量 | DNSMOS | 3.276 | 3.030(AV-TF-GridNet) | +0.246,与 Swift-Net-12 并列第一 |
| 推理效率(2 秒 16kHz 输入 + 50 帧嘴部视频) | GPU 延迟 (ms) | 51.3 | 567.8(AV-TF-GridNet) | 降低 11.1×;MACs 10.6G vs 213.8G(20.2×) |
| 仿真训练对真实录音的泛化(RealSSA-Real) | NISQA | 3.615(RealSSA-Sim 训练的 Falcon) | 2.431(同架构用 VoxCeleb2 训练) | +1.184;五个架构 DNSMOS 全部提升 |
局限与改进
作者承认的局限:Falcon 的分离器与对比损失都在单通道上运行,附录 A.2 模拟的四元麦克风阵列留作未来多通道扩展;RealSSA-Real 仅约 84 分钟、10 名参与者、8 个环境,规模偏小;对比损失依赖仿真中近场干扰者的干净单源波形,无法直接用于无标注真实录音。我的补充观察:RealSSA-Sim 声学配置虽丰富,但房间冲激响应由 gpuRIR 仿真生成,与真实房间的系统性失配仍可能存在,而唯一的真实评测集又太小,难以支撑细粒度的统计结论;DNSMOS/NISQA 是非侵入式客观指标,与主观 MOS 的相关性有限,WER 只用 Mega-ASR 一个识别系统评估,结论可能偏向该系统的声学特性;效率对比中质量最强的基线 AV-TFGridNet 是 2023 年架构,若与同为 0.5M 参数的 Swift-Net-12 相比,优势缩小为 MACs 减 76.6%、GPU 延迟降 70.2%,而 RealSSA-Sim SI-SNRi 只提升 0.4 dB;另外 RealSSA-Real 的 WER 落后 AV-Mossformer2 0.74 点,说明极端噪声条件下仍有改进空间。
独立分析的弱点
第一,单通道输入使模型只能依靠频谱与唇动线索抑制空间邻近干扰,遇到与目标音色相近的同性别近场干扰者时,编码器空间的推斥力可能不足以完全消除泄漏——改进方向是引入小型麦克风阵列前端(如流式 MVDR 或神经波束形成)与 Falcon 的掩码分支级联。第二,对比损失需要每个近场干扰者的干净参考波形,真实部署中根本拿不到,这把该技术锁死在仿真训练域——可用自监督替代:以模型第一版分离结果构造伪负样本,或用预训练说话人验证模型在真实混合上打伪标签再算对比项。第三,视觉分支继承 CTCNet 的 FRCNN 且冻结,侧脸、遮挡、低光下唇部特征会退化,论文只在附录 M 做了视觉完全缺失的实验,缺乏对低质量视觉的系统性鲁棒性评估——改进方向是训练时对视觉特征做 dropout 式退化增强,或加一个轻量的视觉质量门控决定音视频融合权重。第四,效率数据在特定 GPU/CPU 平台测得,边缘 NPU(手机 DSP 等)实测延迟可能不同,且 TFRB 中的深度 LSTM 与 2D 注意力对 NPU 并行化不友好——可探索用全卷积或状态空间模型替换 LSTM,并做 INT8 量化验证。第五,25fps 固定帧率的视频输入假设限制了输入源类型,低帧率或变帧率摄像头场景需要自适应插值策略。
未来方向
作者明确提出的方向:把分离器与编码器空间对比损失扩展到多通道(利用附录 A.2 已模拟的四元阵列数据);让对比损失适配无标注真实录音,摆脱对干净干扰源波形的依赖;扩大 RealSSA-Real 的规模与环境覆盖。基于本文成果可以延伸的工作:把 Falcon 作为流式前端接入实时 ASR 或同声传译系统,量化端到端的 WER 与延迟收益,验证「低层分离质量提升能传导到下游任务」;探索声纹注册与唇动线索的自适应融合,在视觉缺失或低质量时无缝退化为纯音频目标说话人提取;把「单遍多尺度替代多轮迭代」的设计原则迁移到其他流式语音增强任务(去混响、去噪、回声消除);对比损失中 softmax 负样本加权、角 margin 与温度退火的组合可推广到说话人表征学习;在说话人数动态变化的开放场景中研究无需预设干扰数的持续分离与在线说话人计数。
复现评估
论文声明代码与 demo 见 https://cslikai.cn/Falcon/。Falcon 本身只有 0.5M 参数、隐藏维 $C_h = 64$,单张消费级 GPU 即可训练,推理只需 51.3ms GPU 延迟和 364.7MB 显存,工程落地门槛很低。数据方面:RealSSA-Sim 约 153 小时(100,826 训练 / 9,133 验证 / 9,169 测试),生成管线依赖 gpuRIR 做房间冲激响应仿真,语料来自 LRS3(目标,需从 TED 视频爬取或申请)、LibriSpeech(干扰者)、WHAM!(持续背景噪声)与 Hive(突发噪声),整条数据管线有一定工程量;RealSSA-Real 仅 84 分钟评测集,涉及隐私处理(嘴部灰度裁剪、16kHz 重采样、25fps 转码)。若只复现 Falcon 及两个关键消融(下采样深度、对比损失),单卡数天可完成;若要复现主表需重训 9 个基线(含 56.7M 参数的 AV-Mossformer2 因果化版本),算力需求较大。训练超参在附录 D、延迟测量协议在附录 E 均有交代,复现信息总体充分,主要不确定因素是 RealSSA 数据集本身的开源发布情况。
论文图表
四种真实声学场景的照片(小型房间、咖啡店、购物商场、户外街道),直观展示标准固定双说话人基准没有覆盖的因素:说话人数变化、近场/远场干扰并存、混响和突发环境噪声。
这是论文动机的直观呈现:真实部署场景与 LRS2/LRS3/VoxCeleb2 这类受控基准之间的差距,正是 RealSSA 数据集四个场景类型设计的出发点。