XLSR-MamBo: Scaling the Hybrid Mamba-Attention Backbone for Audio Deepfake Detection¶
会议: ACL 2026 Findings
arXiv: 2601.02944
代码: GitHub
领域: AI安全 / 语音伪造检测
关键词: 音频深度伪造检测, Mamba, 混合架构, 状态空间模型, XLSR
一句话总结¶
提出 XLSR-MamBo 框架,系统探索 Mamba-Attention 混合架构在音频深度伪造检测中的四种拓扑设计和多种 SSM 变体(Mamba2、Hydra、GDN),其中 MamBo-3-Hydra 利用 Hydra 的原生双向建模达到多个基准上的竞争性能,且增加骨干深度可有效缓解浅层模型的性能不稳定。
研究背景与动机¶
领域现状:音频深度伪造检测(ADD)已从手工特征转向端到端架构。XLSR 作为前端特征提取器搭配 Conformer 等注意力分类器是主流方案。近期 Mamba 等状态空间模型(SSM)因线性复杂度受到关注。
现有痛点:纯因果 SSM 是单向的,难以捕捉全局频域伪造痕迹所需的基于内容的检索能力。现有双向 Mamba 扩展依赖手工设计的双分支策略(如正反向拼接),存在结构冗余。Transformer 的二次复杂度限制了长序列效率。
核心矛盾:SSM 擅长高效时序压缩和局部高频伪影捕捉,Attention 擅长全局关联和内容检索——深度伪造信号同时表现为局部高频伪影和全局频谱不一致,单一机制都不够。
本文目标:系统探索 SSM-Attention 混合架构在 ADD 中的最优拓扑组合,并评估深度缩放对性能稳定性的影响。
切入角度:受 Jamba、Zamba 等 LLM 混合架构启发,但针对 ADD 任务进行定制化探索,特别引入 Hydra(原生双向 SSM)替代启发式双向策略。
核心 idea:SSM 和 Attention 的互补性(时序压缩 vs 内容检索)在 ADD 中尤为重要,Hydra 的原生双向参数化比双分支策略更优雅,增加 SSM 堆叠深度 N 可缓解性能不稳定。
方法详解¶
整体框架¶
XLSR-MamBo 的出发点是:深度伪造信号既有局部高频伪影、又有全局频谱不一致,SSM 擅长高效时序压缩与局部捕捉、Attention 擅长全局关联与内容检索,单靠一种机制都不够,于是把两者混合进同一个骨干。原始音频先经冻结的 XLSR 前端提取特征 \(X \in \mathbb{R}^{T \times 1024}\),过 RMSNorm 并线性投影到隐藏维 \(D=128\),再经 \(L=5\) 层 MamBo 混合层编码,最后由门控注意力池化聚合成句级表示、线性层输出真伪二分类 logits。论文的真正工作不在某个固定结构,而在系统比较 SSM 与 Attention 的多种拓扑组合及其深度缩放行为——下面三个关键设计都作用在这套 MamBo 混合骨干上。
%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400, 'subGraphTitleMargin': {'top': 8, 'bottom': 16}}}}%%
flowchart TD
A["原始音频"] --> B["冻结 XLSR 前端<br/>提取特征 X (T×1024)"]
B --> C["RMSNorm + 线性投影<br/>降到隐藏维 D=128"]
C --> D
subgraph MB["MamBo 混合骨干(L=5 层)"]
direction TB
D["四种 MamBo 拓扑<br/>层内混合 / 层间交替选型"] --> E["Hydra 原生双向 SSM<br/>准可分矩阵免双分支"]
E --> F["深度缩放(堆叠 N)<br/>单元内连叠 N 个 SSM 块"]
end
F --> G["门控注意力池化<br/>聚合为句级表示"]
G --> H["线性层 → 真伪二分类 logits"]
关键设计¶
1. 四种 MamBo 拓扑:层内混合还是层间交替
为了搞清 SSM 与 Attention 该怎么拼,作者枚举了四种拓扑:MamBo-1 用纯 SSM 直接替换多头注意力;MamBo-2(Mamer)在 SSM 之后接一层 MHA 来替换 FFN,属于层内混合;MamBo-3 让 Mamba 层与 Transformer 层交替堆叠;MamBo-4 让 Mamba 层与 Mamer 层交替堆叠。前两者探索"层内"如何融合两种算子,后两者探索"层间"如何穿插,且每种拓扑都可换装不同 SSM 变体(Mamba、Mamba2、Hydra、GDN)。这种全因子式的探索把不同伪造痕迹对处理方式的偏好暴露出来,实验结论是层间交替(MamBo-3)整体最优。
2. Hydra 原生双向 SSM:用准可分矩阵免掉双分支启发式
深度伪造的伪影可能分布在整段音频里,检测需要非因果的全局上下文,但纯因果 SSM 是单向的,过去的双向 Mamba 又依赖正反向拼接这类手工双分支策略、结构冗余。Hydra 改从参数化层面解决:把前向与反向扫描统一写成一个准可分矩阵,其下三角承载过去信息、上三角承载未来信息,整体计算形如 \(\text{shift}(SS(X)) + \text{flip}(\text{shift}(SS(\text{flip}(X)))) + DX\),在线性复杂度内就原生地完成双向处理。它既补上了"因果一致性违反"这一伪造线索所需的非因果视野,又比双分支更优雅、无冗余。
3. 深度缩放(Stacking N):靠堆叠层数压住浅层的不稳定
实验中发现浅层模型性能方差很大、跑同一配置结果忽高忽低,原因是 SSM 层数太少、表征深度不足以稳定地刻画复杂伪造痕迹。作者引入堆叠超参数 \(N\),允许在单个单元里连续叠 \(N\) 个 SSM 块来加深表征。扫描下来 \(N=3\) 在性能与稳定性之间最佳,而 \(N=1\) 的浅层模型方差明显偏大——这条"加深可缓解不稳定"的规律对实际部署有直接价值。
损失函数 / 训练策略¶
使用 FocalLoss 处理类别不平衡。AdamW 优化器(\(lr=10^{-5}\)),10% 线性 warmup + 余弦衰减。混合精度训练(BF16/FP32),最多 20 epoch,早停 patience=7。在 ASVspoof 2019 LA 训练集上训练,跨数据集评估泛化性。
实验关键数据¶
主实验¶
| 模型 | ASV21LA EER↓ | ASV21DF EER↓ | ITW EER↓ |
|---|---|---|---|
| XLSR-Conformer (基线) | ~1.0 | ~2.5 | ~5.0 |
| MamBo-1-Mamba (N=1) | 1.19 | 2.08 | 4.65 |
| MamBo-3-Hydra (N=3) | 最优 | 竞争性 | 竞争性 |
| RawBMamba | - | - | - |
消融实验¶
| 配置 | ASV21LA | 说明 |
|---|---|---|
| MamBo-1 (纯SSM) | 基线 | SSM 替换 Attention |
| MamBo-2 (Mamer) | 略优 | 层内混合有帮助 |
| MamBo-3 (交替) | 最优 | 层间交替效果最好 |
| N=1 vs N=3 | 方差↓ | 深度缩放显著提升稳定性 |
关键发现¶
- MamBo-3(Mamba-Transformer 交替)在多数基准上表现最优,证明层间交替优于层内混合
- Hydra 在 MamBo-3 中表现最佳,其原生双向建模比 Mamba 的启发式双分支更有效
- 增加 SSM 堆叠深度 N 从 1 到 3 显著降低性能方差,浅层模型的不稳定性是实际部署的隐患
- 在 DFADD 数据集上对扩散和流匹配合成方法保持鲁棒,证明泛化能力
- GDN 的 delta rule 记忆管理在某些场景下也表现不错
亮点与洞察¶
- 系统化的拓扑探索(4 种设计 × 4 种 SSM 变体 × 不同深度)为 SSM-Attention 混合架构在语音任务中的应用提供了全面的设计指南。这种方法论可迁移到其他语音任务
- Hydra 的原生双向能力在 ADD 中的优势验证了"因果一致性违反"作为伪造检测线索的假设
- "深度缩放缓解浅层不稳定"是实用的工程洞察,对实际部署有直接指导意义
局限与展望¶
- 仅在 ASVspoof 2019 LA 训练集上训练,训练数据多样性有限
- 模型规模较小(D=128, L=5),更大规模模型的表现未探索
- ITW 数据集上的性能仍有提升空间
- 未探索端到端训练(XLSR 参数冻结)
- 未来可探索更多混合拓扑和跨语言泛化
相关工作与启发¶
- vs XLSR-Conformer: 纯注意力架构,本文混合 SSM 在效率和性能上均有改进
- vs RawBMamba: 手工双向 Mamba 策略,本文用 Hydra 原生双向替代更优雅
- vs Jamba/Samba: LLM 领域的混合架构,本文首次将此范式系统化应用于 ADD
评分¶
- 新颖性: ⭐⭐⭐⭐ 系统化探索 SSM-Attention 混合在 ADD 中的应用,Hydra 引入有新意
- 实验充分度: ⭐⭐⭐⭐⭐ 四种拓扑 × 四种变体 × 多深度 × 多数据集,非常全面
- 写作质量: ⭐⭐⭐⭐ 背景知识详实,实验组织清晰
- 价值: ⭐⭐⭐⭐ 为 ADD 领域的架构选择提供了系统化参考