Conv-TasNet: Surpassing Ideal Time-Frequency Magnitude Masking for Speech Separation

20 Sep 2018  ·  Yi Luo, Nima Mesgarani ·

Single-channel, speaker-independent speech separation methods have recently seen great progress. However, the accuracy, latency, and computational cost of such methods remain insufficient... The majority of the previous methods have formulated the separation problem through the time-frequency representation of the mixed signal, which has several drawbacks, including the decoupling of the phase and magnitude of the signal, the suboptimality of time-frequency representation for speech separation, and the long latency in calculating the spectrograms. To address these shortcomings, we propose a fully-convolutional time-domain audio separation network (Conv-TasNet), a deep learning framework for end-to-end time-domain speech separation. Conv-TasNet uses a linear encoder to generate a representation of the speech waveform optimized for separating individual speakers. Speaker separation is achieved by applying a set of weighting functions (masks) to the encoder output. The modified encoder representations are then inverted back to the waveforms using a linear decoder. The masks are found using a temporal convolutional network (TCN) consisting of stacked 1-D dilated convolutional blocks, which allows the network to model the long-term dependencies of the speech signal while maintaining a small model size. The proposed Conv-TasNet system significantly outperforms previous time-frequency masking methods in separating two- and three-speaker mixtures. Additionally, Conv-TasNet surpasses several ideal time-frequency magnitude masks in two-speaker speech separation as evaluated by both objective distortion measures and subjective quality assessment by human listeners. Finally, Conv-TasNet has a significantly smaller model size and a shorter minimum latency, making it a suitable solution for both offline and real-time speech separation applications. read more

PDF Abstract
Task Dataset Model Metric Name Metric Value Global Rank Uses Extra
Training Data
Result Benchmark
Multi-task Audio Source Seperation MTASS Conv-TasNet SDRi (Speech) 11.80 # 2
SDRi (Music) 8.35 # 2
SDRi (Noise) 8.07 # 2
SDRi (Average) 9.41 # 2
Music Source Separation MUSDB18 Conv-TasNet SDR (vocals) 6.81 # 11
SDR (drums) 6.08 # 13
SDR (other) 4.37 # 12
SDR (bass) 5.66 # 8
SDR (avg) 5.73 # 13
Music Source Separation MUSDB18 Conv-TasNet (extra) SDR (vocals) 6.74 # 12
SDR (drums) 7.11 # 6
SDR (other) 4.44 # 10
SDR (bass) 7.00 # 5
SDR (avg) 6.32 # 5
Speech Separation WSJ0-2mix Conv-TasNet SI-SDRi 15.3 # 15