Skipformer: A Skip-and-Recover Strategy for Efficient Speech Recognition

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhu, Wenjing, Sun, Sining, Shan, Changhao, Fan, Peng, Yang, Qing
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916253629677568
author Zhu, Wenjing
Sun, Sining
Shan, Changhao
Fan, Peng
Yang, Qing
author_facet Zhu, Wenjing
Sun, Sining
Shan, Changhao
Fan, Peng
Yang, Qing
contents Conformer-based attention models have become the de facto backbone model for Automatic Speech Recognition tasks. A blank symbol is usually introduced to align the input and output sequences for CTC or RNN-T models. Unfortunately, the long input length overloads computational budget and memory consumption quadratically by attention mechanism. In this work, we propose a "Skip-and-Recover" Conformer architecture, named Skipformer, to squeeze sequence input length dynamically and inhomogeneously. Skipformer uses an intermediate CTC output as criteria to split frames into three groups: crucial, skipping and ignoring. The crucial group feeds into next conformer blocks and its output joint with skipping group by original temporal order as the final encoder output. Experiments show that our model reduces the input sequence length by 31 times on Aishell-1 and 22 times on Librispeech corpus. Meanwhile, the model can achieve better recognition accuracy and faster inference speed than recent baseline models. Our code is open-sourced and available online.
format Preprint
id arxiv_https___arxiv_org_abs_2403_08258
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Skipformer: A Skip-and-Recover Strategy for Efficient Speech Recognition
Zhu, Wenjing
Sun, Sining
Shan, Changhao
Fan, Peng
Yang, Qing
Computation and Language
Machine Learning
Conformer-based attention models have become the de facto backbone model for Automatic Speech Recognition tasks. A blank symbol is usually introduced to align the input and output sequences for CTC or RNN-T models. Unfortunately, the long input length overloads computational budget and memory consumption quadratically by attention mechanism. In this work, we propose a "Skip-and-Recover" Conformer architecture, named Skipformer, to squeeze sequence input length dynamically and inhomogeneously. Skipformer uses an intermediate CTC output as criteria to split frames into three groups: crucial, skipping and ignoring. The crucial group feeds into next conformer blocks and its output joint with skipping group by original temporal order as the final encoder output. Experiments show that our model reduces the input sequence length by 31 times on Aishell-1 and 22 times on Librispeech corpus. Meanwhile, the model can achieve better recognition accuracy and faster inference speed than recent baseline models. Our code is open-sourced and available online.
title Skipformer: A Skip-and-Recover Strategy for Efficient Speech Recognition
topic Computation and Language
Machine Learning
url https://arxiv.org/abs/2403.08258