Token-level and sequence-level loss smoothing for RNN language models

Abstract : Despite the effectiveness of recurrent neu-ral network language models, their maximum likelihood estimation suffers from two limitations. It treats all sentences that do not match the ground truth as equally poor, ignoring the structure of the output space. Second, it suffers from "exposure bias": during training tokens are predicted given ground-truth sequences, while at test time prediction is conditioned on generated output sequences. To overcome these limitations we build upon the recent reward augmented maximum likelihood approach i.e. sequence-level smoothing that encourages the model to predict sentences close to the ground truth according to a given performance metric. We extend this approach to token-level loss smoothing, and propose improvements to the sequence-level smoothing approach. Our experiments on two different tasks, image captioning and machine translation, show that token-level and sequence-level loss smoothing are complementary, and significantly improve results.
Complete list of metadatas

Cited literature [48 references]  Display  Hide  Download

https://hal.inria.fr/hal-01790879
Contributor : Thoth Team <>
Submitted on : Monday, May 14, 2018 - 10:19:13 AM
Last modification on : Tuesday, April 2, 2019 - 1:47:51 AM
Long-term archiving on : Tuesday, September 25, 2018 - 10:25:32 PM

File

paper.pdf
Files produced by the author(s)

Identifiers

  • HAL Id : hal-01790879, version 1

Citation

Maha Elbayad, Laurent Besacier, Jakob Verbeek. Token-level and sequence-level loss smoothing for RNN language models. ACL - 56th Annual Meeting of the Association for Computational Linguistics, Jul 2018, Melbourne, Australia. pp.2094-2103. ⟨hal-01790879⟩

Share

Metrics

Record views

467

Files downloads

513