Audio-Native Speech Recognition with a Frozen Discrete-Diffusion Language Model
Summary
A new audio-native speech recognition system utilizes a frozen 26B mixture-of-experts DiffusionGemma language model, which employs uniform, random-token discrete diffusion for parallel transcript refinement. This approach contrasts with traditional autoregressive decoders. The system integrates a frozen Whisper encoder for acoustic features, a lightweight projector, and low-rank adapters, training only 42M parameters (0.16% of the backbone). Initially, standard training objectives failed to ground the audio, but applying a Connectionist Temporal Classification (CTC) loss through the frozen output head resolved this issue. The resulting model achieves a 6.6% Word Error Rate on LibriSpeech test-clean and transcribes utterances in approximately eight parallel steps, regardless of length. A single adapter, trained on six languages, was evaluated on English, Hindi, and Mandarin.
Key takeaway
For Machine Learning Engineers developing speech recognition systems, this research demonstrates a viable alternative to autoregressive decoders. You should consider discrete diffusion language models for their ability to transcribe speech in parallel, offering consistent processing times regardless of utterance length. This approach, particularly with CTC loss for audio grounding, could significantly improve inference efficiency and scalability in your ASR deployments.
Key insights
Discrete diffusion language models can perform parallel speech transcription by refining whole transcripts.
Principles
- Diffusion models offer parallel decoding for ASR.
- CTC loss can ground audio in frozen LMs.
- Low-rank adapters enable modality transfer.
Method
An audio-native interface projects Whisper features into a frozen DiffusionGemma backbone via LoRA, then uses CTC loss through the output head to enable parallel speech transcription.
In practice
- Explore discrete diffusion for ASR tasks.
- Use CTC loss to stabilize audio-LM integration.
- Apply LoRA for efficient multimodal adaptation.
Topics
- Automatic Speech Recognition
- Diffusion Models
- Language Models
- Whisper Encoder
- Connectionist Temporal Classification
- Low-Rank Adapters
Best for: Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.