Audio-Native Speech Recognition with a Frozen Discrete-Diffusion Language Model

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, quick

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

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

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 Artificial Intelligence.