Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper explores the use of a discrete diffusion language model for automatic speech recognition, challenging the conventional autoregressive decoder approach. By leveraging a frozen Whisper encoder and a lightweight projector, the authors train a 26B mixture-of-experts model called DiffusionGemma, achieving a significant reduction in parameter training while maintaining performance. The model demonstrates a 6.6% word error rate on the LibriSpeech test-clean dataset, transcribing speech in parallel across multiple languages with remarkable efficiency.
A discrete diffusion model can achieve competitive speech recognition accuracy while transcribing in parallel, breaking the mold of traditional autoregressive methods.
Automatic speech recognition is dominated by autoregressive decoders that emit one token at a time. We ask whether a discrete diffusion language model can transcribe speech instead, refining a whole transcript in parallel over a small number of denoising steps. We train an audio-native interface for DiffusionGemma, a 26B mixture-of-experts model that generates text by uniform, random-token discrete diffusion rather than the absorbing-mask scheme common to recent diffusion language models. A frozen Whisper encoder supplies acoustic features, a lightweight projector maps them into the model embedding space, and low-rank adapters let the frozen backbone attend to the new modality. About 42M parameters are trained, which is 0.16 percent of the backbone. We find that the natural training objectives fail to ground the audio because their gradient reaches the projector only through attention that has already dismissed it. A connectionist temporal classification loss applied through the frozen output head breaks this deadlock. The resulting model reaches 6.6 percent word error rate on LibriSpeech test-clean, transcribes in roughly eight parallel steps regardless of utterance length, and uses a single adapter trained on six languages, which we evaluate here on English, Hindi, and Mandarin.