arXiv · 1808.08987
Large Margin Neural Language Model
Abstract
We propose a large margin criterion for training neural language models. Conventionally, neural language models are trained by minimizing perplexity (PPL) on grammatical sentences. However, we demonstrate that PPL may not be the best metric to optimize in some tasks, and further propose a large margin formulation. The proposed method aims to enlarge the margin between the "good" and "bad" sentences in a task-specific sense. It is trained end-to-end and can be widely applied to tasks that involve re-scoring of generated text. Compared with minimum-PPL training, our method gains up to 1.1 WER reduction for speech recognition and 1.0 BLEU increase for machine translation.
Explore related subjects
Keep this discovery
Jiaji Huang, Yi Li, Wei Ping, Liang Huang. 2018-08-27. Large Margin Neural Language Model. https://arxiv.org/abs/1808.08987
Cite the original work for its findings. Save a collection to share your selection of sources.