Scaling Properties of Continuous Diffusion Spoken Language Models

· Source: Apple Machine Learning Research · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Speech and Natural Language Processing · Depth: Expert, quick

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

Method

The phoneme Jensen-Shannon divergence (pJSD) metric quantifies SLM linguistic quality by measuring the divergence between phoneme distributions.

In practice

Topics

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Apple Machine Learning Research.