USM RNN-T model weights binarization

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Rybakov, Oleg, Serdyuk, Dmitriy, Zheng, Chengjian
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866912043354816512
author Rybakov, Oleg
Serdyuk, Dmitriy
Zheng, Chengjian
author_facet Rybakov, Oleg
Serdyuk, Dmitriy
Zheng, Chengjian
contents Large-scale universal speech models (USM) are already used in production. However, as the model size grows, the serving cost grows too. Serving cost of large models is dominated by model size that is why model size reduction is an important research topic. In this work we are focused on model size reduction using weights only quantization. We present the weights binarization of USM Recurrent Neural Network Transducer (RNN-T) and show that its model size can be reduced by 15.9x times at cost of word error rate (WER) increase by only 1.9% in comparison to the float32 model. It makes it attractive for practical applications.
format Preprint
id arxiv_https___arxiv_org_abs_2406_02887
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle USM RNN-T model weights binarization
Rybakov, Oleg
Serdyuk, Dmitriy
Zheng, Chengjian
Audio and Speech Processing
Sound
Large-scale universal speech models (USM) are already used in production. However, as the model size grows, the serving cost grows too. Serving cost of large models is dominated by model size that is why model size reduction is an important research topic. In this work we are focused on model size reduction using weights only quantization. We present the weights binarization of USM Recurrent Neural Network Transducer (RNN-T) and show that its model size can be reduced by 15.9x times at cost of word error rate (WER) increase by only 1.9% in comparison to the float32 model. It makes it attractive for practical applications.
title USM RNN-T model weights binarization
topic Audio and Speech Processing
Sound
url https://arxiv.org/abs/2406.02887