SLM-SS: Token-Based Speech Separation with Speech Language Models
Organizations: Auditory Cognition and Computational Acoustics Lab, Shanghai Jiao Tong University, Shanghai, 200240, P. R. China
Abstract
Speech separation (SS) has witnessed substantial progress in recent years with the advent of neural network-based methods, achieving notable improvements on signal-level evaluation metrics. Existing speech separation systems are mostly optimized with signal-level objectives, and often yield outputs that are acceptable in waveform metrics but suboptimal for intelligibility-sensitive downstream tasks. To address this gap, we propose SLM-SS, a token-based generative speech separation framework that models multi-talker mixtures with speech language models. We first discretize clean speech into multi-codebook token sequences with neural codecs and serialize multi-speaker targets via Serialized Output Training (SOT). An encoder-decoder autoregressive speech language model predicts the zero-order sequence to establish speaker ordering and coarse linguistic content, and a non-autoregressive module then refines higher-order residual codebooks for efficient reconstruction. We further introduce a per-talker non-autoregressive strategy that simplifies high-order modeling and improves the stability. On Libri2Mix, SLM-SS with per-talker NAR achieves WER, approaching codec-only reconstruction. The WavTokenizer variant eliminates NAR refinement and achieves the highest mean subjective quality score among evaluated separation systems. Its UTMOS and DNSMOS advantages over discriminative baselines persist on unseen noisy and Chinese mixtures without fine-tuning.
Figures & tables
| Speaker & Linguistic Consistency | Signal | Non-intrusive | Subjective | ||||||
| Method | Spk Sim. | WER | TER | LPS | SBS | PESQ | UTMOS | DNSMOS | Score |
| GT | - | 5.19 | - | 1.000 | 1.000 | 4.64 | 4.09 | 3.59 | 78.41 4.08 |
| GT-Encodec32 | 93.5 | 6.03 | 24.7 | 0.975 | 0.957 | 3.14 | 3.18 | 3.50 | - |
| GT-Encodec8 (UB) | 92.8 | 6.31 | 39.0 | 0.970 | 0.944 | 2.62 | 2.95 | 3.38 | - |
| GT-WavTokenizer | 92.7 | 7.58 | ∗ 47.3 | 0.964 | 0.939 | 2.33 | 4.01 | 3.67 | - |
| BSRNN | 92.9 | 20.02 | 65.1 | 0.916 | 0.903 | 2.37 | 2.96 | 3.02 | - |
| Temperature | Spk Sim. | WER | TER | LPS | SBS |
|---|---|---|---|---|---|
| 0.5 | 38.9 | 49.08 | 69.3 | 0.581 | 0.695 |
| 0.9 | 73.1 | 10.20 | 56.9 | 0.900 | 0.845 |
| 1.0 | 91.7 | 7.24 | 45.8 | 0.954 | 0.913 |
| 1.1 | 77.8 | 9.73 | 52.0 | 0.949 | 0.895 |
| 1.5 | 54.2 | 64.55 | 87.8 | 0.178 | 0.497 |
| Train | Test | Spk sim | WER | TER | LPS | SBS |
|---|---|---|---|---|---|---|
| 2Mix | 2Mix | 90.2 | 8.84 | 51.1 | 0.936 | 0.900 |
| 3Mix | 3Mix | 87.3 | 12.12 | 63.7 | 0.895 | 0.870 |
| Joint | 2Mix | 90.2 | 8.80 | 50.7 | 0.939 | 0.903 |
| 3Mix | 88.8 | 9.94 | 61.3 | 0.915 | 0.875 |
| NAR | #Talker | Spk Sim. | WER | TER | LPS | SBS |
|---|---|---|---|---|---|---|
| Unfinetuned | 2Mix | 90.1 | 8.83 | 50.5 | 0.937 | 0.903 |
| 3Mix | 88.7 | 9.93 | 61.3 | 0.912 | 0.871 | |
| 4Mix | 79.5 | 16.19 | 70.1 | 0.833 | 0.794 | |
| Finetuned | 2Mix | 90.1 | 8.83 | 50.6 | 0.938 | 0.903 |
| 3Mix | 88.7 | 9.92 | 61.4 | 0.911 | 0.872 | |
| 4Mix | 84.6 | 11.10 | 66.7 | 0.889 | 0.852 |
| Dataset | Decode method | Spk Sim. | WER | LPS | SBS |
|---|---|---|---|---|---|
| 2Mix | Greedy Search | 91.9 | 9.66 | 0.904 | 0.915 |
| Top-30 RS | 92.0 | 9.08 | 0.917 | 0.919 | |
| Beam Search | 92.3 | 8.97 | 0.923 | 0.921 | |
| 3Mix | Greedy Search | 87.2 | 11.41 | 0.906 | 0.910 |
| Top-30 RS | 87.8 | 11.03 | 0.908 | 0.911 | |
| Beam Search | 88.8 | 10.18 | 0.915 | 0.913 |
| Configuration | #Params | GMACs | Mem | RTF | WER | PESQ | UTMOS | DNSMOS |
|---|---|---|---|---|---|---|---|---|
| WavLM Base+ Encoder (94M) | ||||||||
| + 23M Decoder (b-t) | 117M | 37.14 | 0.72GiB | 0.29 | 72.87 | 1.22 | 2.21 | 3.07 |
| + 48M Decoder (b-s) | 142M | 69.52 | 1.44GiB | 0.52 | 63.88 | 1.34 | 2.58 | 3.16 |
| + 121M Decoder (b-m) | 215M | 161.01 | 2.18GiB | 0.75 | 36.91 | 1.65 | 3.01 | 3.29 |
| + 212M Decoder (b-l) | 306M | 348.73 | 2.90GiB | 0.91 | 25.22 | 1.91 | 3.32 | 3.40 |
| WavLM Large Encoder (315M) | ||||||||
| Model | WER/CER | PESQ | UTMOS | DNSMOS |
|---|---|---|---|---|
| (a) Libri2Mix-noisy | ||||
| SLM-SS-WT (l-s) | 30.46 | 1.60 | 2.96 | 3.20 |
| SLM-SS-WT (l-l) | 25.54 | 1.61 | 2.98 | 3.22 |
| TF-LocoFormer | 25.96 | 1.64 | 2.11 | 2.56 |
| MossFormer2 | 33.89 | 1.39 | 1.54 | 2.35 |
| (b) AISHELL1-2Mix | ||||
Speech Separation
Speech separation isolates individual voices from overlapping speech, supporting clearer communication, transcription, and speaker-aware audio processing.
See all 36 papers