In speech separation, both CNN- and Transformer-based models have
demonstrated robust separation capabilities, garnering significant attention
within the research community. However, CNN-based methods have limited
modelling capability for long-sequence audio, leading to suboptimal separation
performance. Conversely, Transformer-based methods are limited in practical
applications due to their high computational complexity. Notably, within
computer vision, Mamba-based methods have been celebrated for their formidable
performance and reduced computational requirements. In this paper, we propose a
network architecture for speech separation using a state-space model, namely
SPMamba. We adopt the TF-GridNet model as the foundational framework and
substitute its Transformer component with a bidirectional Mamba module, aiming
to capture a broader range of contextual information. Our experimental results
reveal an important role in the performance aspects of Mamba-based models.
SPMamba demonstrates superior performance with a significant advantage over
existing separation models in a dataset built on Librispeech. Notably, SPMamba
achieves a substantial improvement in separation quality, with a 2.42 dB
enhancement in SI-SNRi compared to the TF-GridNet. The source code for SPMamba
is publicly accessible at this https URL .

本文以状态空间模型为基础，提出了一种用于语音分离的网络架构，即 SPMamba。通过将 TF-GridNet 模型的 Transformer 组件替换为双向 Mamba 模块，旨在捕捉更广泛的上下文信息。实验结果表明，基于 Mamba 的模型在性能方面发挥了重要作用，SPMamba 在建立在 Librispeech 数据集上的模型中表现出优越的性能，与 TF-GridNet 相比，在 SI-SNRi 方面提高了 2.42 dB。SPMamba 的源代码可在此 https URL 提交。