SELM: Speech Enhancement Using Discrete Tokens and Language Models
About
Language models (LMs) have shown superior performances in various speech generation tasks recently, demonstrating their powerful ability for semantic context modeling. Given the intrinsic similarity between speech generation and speech enhancement, harnessing semantic information holds potential advantages for speech enhancement tasks. In light of this, we propose SELM, a novel paradigm for speech enhancement, which integrates discrete tokens and leverages language models. SELM comprises three stages: encoding, modeling, and decoding. We transform continuous waveform signals into discrete tokens using pre-trained self-supervised learning (SSL) models and a k-means tokenizer. Language models then capture comprehensive contextual information within these tokens. Finally, a detokenizer and HiFi-GAN restore them into enhanced speech. Experimental results demonstrate that SELM achieves comparable performance in objective metrics alongside superior results in subjective perception. Our demos are available https://honee-w.github.io/SELM/.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Speech Enhancement | DNS Challenge Real Recordings (test) | SIG Score3.591 | 41 | |
| Speech Enhancement | DNS no-reverb 2020 (test) | Signal Score (SIG)3.51 | 30 | |
| Speech Enhancement | DNS Challenge Without Reverb (test) | SIG Score3.51 | 26 | |
| Speech Enhancement | DNS Challenge With Reverb (test) | SIG3.16 | 24 | |
| Speech Enhancement | DNS blind (real recordings) 2020 (test) | SIG Score3.59 | 16 | |
| Speech Enhancement | DNS blind synthetic with reverb 2020 (test) | SIG Score3.16 | 16 | |
| Speech Restoration | DNS Challenge With Reverb 2020 (test) | SIG Score3.16 | 14 | |
| Noise Suppression | Interspeech DNS Challenge With Reverb 2020 (test) | SIG Score3.16 | 10 | |
| Speech Restoration | DNS with reverb 2020 (test) | SIG Score3.16 | 10 | |
| Noise Suppression | Interspeech DNS Challenge blind No Reverb 2020 (test) | SIG Score3.51 | 10 |