CLSDASFeb 20, 2024

OWSM-CTC: An Open Encoder-Only Speech Foundation Model for Speech Recognition, Translation, and Language Identification

NVIDIA
arXiv:2402.12654v343 citationsh-index: 19Has CodeACL
Originality Incremental advance
AI Analysis

This work addresses the need for faster and more robust speech models for multilingual recognition and translation, though it is incremental as it builds on existing CTC and OWSM frameworks.

The paper tackles the challenge of scaling non-autoregressive models for speech-to-text tasks by proposing OWSM-CTC, an encoder-only speech foundation model based on CTC, trained on 180k hours of data for multilingual ASR, ST, and LID. It achieves competitive ASR results, up to 24% relative improvement on ST, and is 3-4 times faster for inference with 20x speed-up for long-form ASR.

There has been an increasing interest in large speech models that can perform multiple tasks in a single model. Such models usually adopt an encoder-decoder or decoder-only architecture due to their popularity and good performance in many domains. However, autoregressive models can be slower during inference compared to non-autoregressive models and also have potential risks of hallucination. Though prior studies observed promising results of non-autoregressive models for certain tasks at small scales, it remains unclear if they can be scaled to speech-to-text generation in diverse languages and tasks. Inspired by the Open Whisper-style Speech Model (OWSM) project, we propose OWSM-CTC, a novel encoder-only speech foundation model based on Connectionist Temporal Classification (CTC). It is trained on 180k hours of public audio data for multilingual automatic speech recognition (ASR), speech translation (ST), and language identification (LID). Compared to encoder-decoder OWSM, our OWSM-CTC achieves competitive results on ASR and up to 24% relative improvement on ST, while it is more robust and 3 to 4 times faster for inference. OWSM-CTC also improves the long-form ASR result with 20x speed-up. We will publicly release our code, pre-trained model, and training logs to promote open science in speech foundation models.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes