Dual-path RNN: efficient long sequence modeling for time-domain single-channel speech separation

Code

• github.com/mpariente/asteroid
  pytorch★ 2,548
• github.com/yluo42/TAC
  pytorch★ 304
• github.com/JusperLee/Dual-Path-RNN-Pytorch
  pytorch★ 0
• github.com/sp-uhh/dual-path-rnn
  tf★ 0
• github.com/fgnt/padertorch
  pytorch★ 0
• github.com/saurjya/asteroid
  pytorch★ 0
• github.com/lyqcom/dprnn
  mindspore★ 0
• github.com/ZhangYu-x/dprnn_mindspore
  mindspore★ 0

Abstract

Recent studies in deep learning-based speech separation have proven the superiority of time-domain approaches to conventional time-frequency-based methods. Unlike the time-frequency domain approaches, the time-domain separation systems often receive input sequences consisting of a huge number of time steps, which introduces challenges for modeling extremely long sequences. Conventional recurrent neural networks (RNNs) are not effective for modeling such long sequences due to optimization difficulties, while one-dimensional convolutional neural networks (1-D CNNs) cannot perform utterance-level sequence modeling when its receptive field is smaller than the sequence length. In this paper, we propose dual-path recurrent neural network (DPRNN), a simple yet effective method for organizing RNN layers in a deep structure to model extremely long sequences. DPRNN splits the long sequential input into smaller chunks and applies intra- and inter-chunk operations iteratively, where the input length can be made proportional to the square root of the original sequence length in each operation. Experiments show that by replacing 1-D CNN with DPRNN and apply sample-level modeling in the time-domain audio separation network (TasNet), a new state-of-the-art performance on WSJ0-2mix is achieved with a 20 times smaller model than the previous best system.

Tasks

Benchmark Results

Reproductions