When Do Self-Supervised Representations Help Speech Restoration?
Organizations: School of Electrical Engineering, KAIST, Daejeon, South Korea · Department of AI Systems, KAIST, Daejeon, South Korea
Abstract
Self-supervised learning (SSL) representations are widely used in speech restoration (SR), and the prevailing trend is SSL feature cleansing: fine-tuning the SSL to recover clean features from degraded speech, then conditioning the restoration model on them. However, the representational capability of SSL when encoding degraded speech has been underexplored, and we ask whether cleansing always benefits SR training. Across SSL encoders and perceptual quality proxies, we discover perceptual correlation collapse---under severe degradation, cleansing degraded SSL features toward clean ones misaligns with the SR objective. We hypothesize the restoration backbone can supply the SSL with clues to resolve this collapse, since its intermediate layers already produce partially restored features. We therefore propose Early Restoration Injection (ERI), which routes intermediate features from the restoration backbone into the SSL through a lightweight conditioning network. ERI excels over the unconditioned baseline and, on the VoiceFixer benchmark, achieves competitive performance among multiple baselines.
Figures & tables
| Model | # Params (M) total / trainable | OVRL ↑ | UTMOS ↑ | PESQ ↑ | ESTOI ↑ | LPS ↑ |
| Degraded | - | 2.39 | 2.63 | 1.94 | 0.58 | 0.84 |
| VoiceFixer | 112 / 112 | 3.00 | 3.44 | 2.04 | 0.56 | 0.80 |
| TF-GridNet | 14.4 / 14.4 | 3.06 | 3.41 | 2.47 | 0.68 | 0.88 |
| AnyEnhance | 364 / 324 | 3.14 | 3.75 | 2.45 | 0.67 | 0.88 |
| SIDON | 250 / 57.4 | 3.20 | 3.97 | 2.05 | 0.58 | 0.88 |
| SEMamba++ | 2.7 / 2.7 | 3.10 | 3.75 | 2.76 | 0.70 | 0.87 |
| ERI | SSL | Stage 1 steps (k) | ||||
| ✓ | WavLM | 500 (Full) | 3.97 | 4.06 | 2.50 | 0.90 |
| ✗ | 3.93 | 4.03 | 2.42 | 0.88 | ||
| ✓ | WavLM | 300 | 3.95 | 4.04 | 2.41 | 0.89 |
| ✗ | 3.88 | 3.99 | 2.41 | 0.88 | ||
| ✓ | HuBERT | 500 | 3.93 | 4.03 | 2.41 | 0.87 |
| ✗ | 3.86 | 3.96 | 2.33 | 0.83 |
| Design | Params | ||||
| Ours (default) | 24M | 3.95 | 4.04 | 2.41 | 0.89 |
| ERI layer 2 1 | 24M | 3.91 | 4.03 | 2.42 | 0.88 |
| w/o sg on SSL cond. fus. | 24M | 3.92 | 4.02 | 2.42 | 0.88 |
| AdaLN-based ERI | 32M | 3.94 | 4.04 | 2.43 | 0.89 |
Self-Supervised Learning
Self-supervised learning uses structure within unlabeled data to train representations and models for downstream tasks.
See all 256 papers