Drax: Speech Recognition with Discrete Flow Matching
To address the performance bottleneck in non-autoregressive (NAR) automatic speech recognition (ASR) caused by train-inference distribution mismatch, this paper proposes Drax—the first NAR ASR framework based on discrete flow matching. Drax constructs an audio-conditioned probability flow path that explicitly models intermediate erroneous token trajectories during inference, thereby mitigating distributional shift between training and inference. Theoretically, it establishes a connection between generalization error and cumulative velocity error, providing principled guidance for model design. Drax enables fully parallel decoding and achieves recognition accuracy competitive with state-of-the-art autoregressive models on benchmarks including LibriSpeech, while substantially improving decoding efficiency. Extensive experiments validate the effectiveness and scalability of discrete flow matching for ASR tasks.