Shallow Fusion of Weighted Finite-State Transducer and Language Model for Text Normalization

Code

• github.com/NVIDIA/NeMo
  OfficialIn paperpytorch★ 16,967
• github.com/MindSpore-scientific-2/code-8/tree/main/shallow-rnns
  mindspore★ 0
• github.com/MindSpore-scientific/code-1/tree/main/shallow-rnns
  mindspore★ 0

Abstract

Text normalization (TN) systems in production are largely rule-based using weighted finite-state transducers (WFST). However, WFST-based systems struggle with ambiguous input when the normalized form is context-dependent. On the other hand, neural text normalization systems can take context into account but they suffer from unrecoverable errors and require labeled normalization datasets, which are hard to collect. We propose a new hybrid approach that combines the benefits of rule-based and neural systems. First, a non-deterministic WFST outputs all normalization candidates, and then a neural language model picks the best one -- similar to shallow fusion for automatic speech recognition. While the WFST prevents unrecoverable errors, the language model resolves contextual ambiguity. The approach is easy to extend and we show it is effective. It achieves comparable or better results than existing state-of-the-art TN models.

Tasks

Reproductions