Saved in:
Bibliographic Details
Main Authors: Dang, Shaoxiang, Matsumoto, Tetsuya, Takeuchi, Yoshinori, Kudo, Hiroaki
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2412.18217
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912169154576384
author Dang, Shaoxiang
Matsumoto, Tetsuya
Takeuchi, Yoshinori
Kudo, Hiroaki
author_facet Dang, Shaoxiang
Matsumoto, Tetsuya
Takeuchi, Yoshinori
Kudo, Hiroaki
contents The topic of speech separation involves separating mixed speech with multiple overlapping speakers into several streams, with each stream containing speech from only one speaker. Many highly effective models have emerged and proliferated rapidly over time. However, the size and computational load of these models have also increased accordingly. This is a disaster for the community, as researchers need more time and computational resources to reproduce and compare existing models. In this paper, we propose U-mamba-net: a lightweight Mamba-based U-style model for speech separation in complex environments. Mamba is a state space sequence model that incorporates feature selection capabilities. U-style network is a fully convolutional neural network whose symmetric contracting and expansive paths are able to learn multi-resolution features. In our work, Mamba serves as a feature filter, alternating with U-Net. We test the proposed model on Libri2mix. The results show that U-Mamba-Net achieves improved performance with quite low computational cost.
format Preprint
id arxiv_https___arxiv_org_abs_2412_18217
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle U-Mamba-Net: A highly efficient Mamba-based U-net style network for noisy and reverberant speech separation
Dang, Shaoxiang
Matsumoto, Tetsuya
Takeuchi, Yoshinori
Kudo, Hiroaki
Sound
Machine Learning
Audio and Speech Processing
The topic of speech separation involves separating mixed speech with multiple overlapping speakers into several streams, with each stream containing speech from only one speaker. Many highly effective models have emerged and proliferated rapidly over time. However, the size and computational load of these models have also increased accordingly. This is a disaster for the community, as researchers need more time and computational resources to reproduce and compare existing models. In this paper, we propose U-mamba-net: a lightweight Mamba-based U-style model for speech separation in complex environments. Mamba is a state space sequence model that incorporates feature selection capabilities. U-style network is a fully convolutional neural network whose symmetric contracting and expansive paths are able to learn multi-resolution features. In our work, Mamba serves as a feature filter, alternating with U-Net. We test the proposed model on Libri2mix. The results show that U-Mamba-Net achieves improved performance with quite low computational cost.
title U-Mamba-Net: A highly efficient Mamba-based U-net style network for noisy and reverberant speech separation
topic Sound
Machine Learning
Audio and Speech Processing
url https://arxiv.org/abs/2412.18217