arXiv · 2304.13134
LAST: Scalable Lattice-Based Speech Modelling in JAX
Abstract
We introduce LAST, a LAttice-based Speech Transducer library in JAX. With an emphasis on flexibility, ease-of-use, and scalability, LAST implements differentiable weighted finite state automaton (WFSA) algorithms needed for training \& inference that scale to a large WFSA such as a recognition lattice over the entire utterance. Despite these WFSA algorithms being well-known in the literature, new challenges arise from performance characteristics of modern architectures, and from nuances in automatic differentiation. We describe a suite of generally applicable techniques employed in LAST to address these challenges, and demonstrate their effectiveness with benchmarks on TPUv3 and V100 GPU.
Explore related subjects
Keep this discovery
Ke Wu, Ehsan Variani, Tom Bagby, Michael Riley. 2023-04-25. LAST: Scalable Lattice-Based Speech Modelling in JAX. https://arxiv.org/abs/2304.13134
Cite the original work for its findings. Save a collection to share your selection of sources.