Scaling Properties of Continuous Diffusion Spoken Language Models
Summary
Continuous diffusion (CD) spoken language models (SLMs) are being explored as a more viable alternative to discrete autoregressive (AR) SLMs, which currently lag behind text and text-speech models in performance due to computational and data demands. Researchers introduced the phoneme Jensen-Shannon divergence (pJSD) metric to quantify SLM linguistic quality. Analysis reveals that CD SLMs exhibit scaling laws for validation loss and pJSD, similar to AR models. They also show optimal token-to-parameter ratios decreasing as compute scales. Notably, at higher compute, loss becomes insensitive to data and model sizes, suggesting potential for faster inference. Scaling CD SLMs to 16B parameters with tens of millions of hours of conversational data enables the generation of emotive, prosodic, multi-speaker, multilingual speech, although achieving long-form coherence remains a significant challenge.
Key takeaway
For Machine Learning Engineers developing spoken language models, exploring continuous diffusion (CD) architectures is crucial. You should consider CD SLMs for their scaling law properties and potential for efficient inference, especially when targeting high-compute environments. While scaling to 16B parameters enables emotive, multi-speaker, multilingual speech, you must address the significant challenge of achieving long-form coherence in generated audio.
Key insights
Continuous diffusion SLMs exhibit scaling laws, offering potential for efficient inference at high compute.
Principles
- CD SLMs mirror AR scaling behavior.
- Optimal token-to-parameter ratios decrease with compute.
- Higher compute broadens optimal model/data allocation.
Method
The phoneme Jensen-Shannon divergence (pJSD) metric quantifies SLM linguistic quality by measuring the divergence between phoneme distributions.
In practice
- Scale CD SLMs to 16B parameters for emotive speech.
- Utilize insensitivity to data/model size for fast inference.
Topics
- Continuous Diffusion Models
- Spoken Language Models
- Scaling Laws
- Phoneme Jensen-Shannon Divergence
- Speech Synthesis
- Efficient Inference
Best for: Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Apple Machine Learning Research.