arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.03344cs.SDcs.AIeess.AS
收藏

Audio Mamba:用于音频表示学习的双向状态空间模型

Audio Mamba: Bidirectional State Space Model for Audio Representation Learning

Korea Advanced Institute of Science and Technology(韩国科学技术院)

Mehmet Hamza Erol, Arda Senocak, Jiu Feng, Joon Son Chung

AI总结:

针对音频分类中 Audio Spectrogram Transformers 自注意力机制带来的二次方计算开销问题,提出首个无自注意力且纯基于状态空间模型的 Audio Mamba (AuM),在六个基准数据集上取得与成熟 AST 相当或更优的性能。

摘要:

Transformers 迅速成为音频分类的首选,超越了基于 CNN 的方法。然而,Audio Spectrogram Transformers (ASTs) 因自注意力机制表现出二次方复杂度扩展。消除此开销极具吸引力。近期,状态空间模型 (SSMs)(如 Mamba)在此方面已展现语言与视觉任务的潜力。本研究探讨音频分类是否必须依赖自注意力。通过提出 Audio Mamba (AuM)(首个无自注意力、纯 SSM 的音频分类模型),我们旨在解答此问题。我们在包含六个不同基准的多种音频数据集上评估了 AuM,其表现与成熟的 AST 模型相当或更优。

英文摘要:

Transformers have rapidly become the preferred choice for audio classification, surpassing methods based on CNNs. However, Audio Spectrogram Transformers (ASTs) exhibit quadratic scaling due to self-attention. The removal of this quadratic self-attention cost presents an appealing direction. Recently, state space models (SSMs), such as Mamba, have demonstrated potential in language and vision tasks in this regard. In this study, we explore whether reliance on self-attention is necessary for audio classification tasks. By introducing Audio Mamba (AuM), the first self-attention-free, purely SSM-based model for audio classification, we aim to address this question. We evaluate AuM on various audio datasets - comprising six different benchmarks - where it achieves comparable or better performance compared to well-established AST model.

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

↑