Dynamic Sparsity Neural Networks for Automatic Speech Recognition
arXiv:2005.10627
Abstract
In automatic speech recognition (ASR), model pruning is a widely adopted technique that reduces model size and latency to deploy neural network models on edge devices with resource constraints. However, multiple models with different sparsity levels usually need to be separately trained and deployed to heterogeneous target hardware with different resource specifications and for applications that have various latency requirements. In this paper, we present Dynamic Sparsity Neural Networks (DSNN) that, once trained, can instantly switch to any predefined sparsity configuration at run-time. We demonstrate the effectiveness and flexibility of DSNN using experiments on internal production datasets with Google Voice Search data, and show that the performance of a DSNN model is on par with that of individually trained single sparsity networks. Our trained DSNN model, therefore, can greatly ease the training process and simplify deployment in diverse scenarios with resource constraints.
arXiv commentICASSP 2021. (c) 2021 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works
References in corpus (12)
- TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systems
- Sequence Transduction with Recurrent Neural Networks
- Rethinking the Value of Network Pruning
- To prune, or not to prune: exploring the efficacy of pruning for model compression
- SNIP: Single-shot Network Pruning based on Connection Sensitivity
- A Closer Look at Memorization in Deep Networks
- Multi-Scale Dense Networks for Resource Efficient Image Classification
- AutoSlim: Towards One-Shot Architecture Search for Channel Numbers
- Stabilizing the Lottery Ticket Hypothesis
- Block-Sparse Recurrent Neural Networks
- Optimizing Speech Recognition For The Edge
- BigNAS: Scaling Up Neural Architecture Search with Big Single-Stage Models