Highly performing deep neural networks come at the cost of computational complexity that limits their practicality for deployment on portable devices. We propose the low-rank transformer (LRT), a memory-efficient and fast neural architecture that significantly reduces the parameters and boosts the speed of training and inference for end-to-end speech recognition. Our approach reduces the number of parameters of the network by more than 50% and speeds up the inference time by around 1.35x compared to the baseline transformer model. The experiments show that our LRT model generalizes better and yields lower error rates on both validation and test sets compared to an uncompressed transformer model. The LRT model outperforms those from existing works on several datasets in an end-to-end setting without using an external language model or acoustic data.
@article{arxiv.1910.13923,
title = {Lightweight and Efficient End-to-End Speech Recognition Using Low-Rank Transformer},
author = {Genta Indra Winata and Samuel Cahyawijaya and Zhaojiang Lin and Zihan Liu and Pascale Fung},
journal= {arXiv preprint arXiv:1910.13923},
year = {2020}
}
Comments
The first two authors contributed equally to this work. Accepted as an oral presentation in ICASSP 2020