HAINAN: Fast and Accurate Transducer for Hybrid-Autoregressive ASR

Hainan Xu, Travis M. Bartley, Vladimir Bataev, Boris Ginsburg
10/3/2024
cs.LG

No confident code implementation found for this paper.

Abstract

We present Hybrid-Autoregressive INference TrANsducers (HAINAN), a novel architecture for speech recognition that extends the Token-and-Duration Transducer (TDT) model. Trained with randomly masked predictor network outputs, HAINAN supports both autoregressive inference with all network components and non-autoregressive inference without the predictor. Additionally, we propose a novel semi-autoregressive inference paradigm that first generates an initial hypothesis using non-autoregressive inference, followed by refinement steps where each token prediction is regenerated using parallelized autoregression on the initial hypothesis. Experiments on multiple datasets across different languages demonstrate that HAINAN achieves efficiency parity with CTC in non-autoregressive mode and with TDT in autoregressive mode. In terms of accuracy, autoregressive HAINAN outperforms TDT and RNN-T, while non-autoregressive HAINAN significantly outperforms CTC. Semi-autoregressive inference further enhances the model's accuracy with minimal computational overhead, and even outperforms TDT results in some cases. These results highlight HAINAN's flexibility in balancing accuracy and speed, positioning it as a strong candidate for real-world speech recognition applications.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{xu2024hainan,
  title  = {HAINAN: Fast and Accurate Transducer for Hybrid-Autoregressive ASR},
  author = {Hainan Xu and Travis M. Bartley and Vladimir Bataev and Boris Ginsburg},
  year   = {2024},
  eprint = {2410.02597},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2410.02597v3},
  journal = {ICLR 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2410.02597.svg)](https://codeofpaper.com/paper/2410.02597)

Discussion