Cold Fusion: Training Seq2Seq Models Together with Language Models
This method enhances speech recognition and similar tasks by efficiently leveraging language information, though it is incremental as it builds on existing Seq2Seq and language model techniques.
The authors tackled the problem of improving sequence-to-sequence models by integrating pre-trained language models during training, resulting in faster convergence, better generalization, and near-complete domain transfer with less than 10% labeled data.
Sequence-to-sequence (Seq2Seq) models with attention have excelled at tasks which involve generating natural language sentences such as machine translation, image captioning and speech recognition. Performance has further been improved by leveraging unlabeled data, often in the form of a language model. In this work, we present the Cold Fusion method, which leverages a pre-trained language model during training, and show its effectiveness on the speech recognition task. We show that Seq2Seq models with Cold Fusion are able to better utilize language information enjoying i) faster convergence and better generalization, and ii) almost complete transfer to a new domain while using less than 10% of the labeled training data.