Mixed Cross Entropy Loss for Neural Machine Translation
Not on the map. Why?
Abstract (arXiv)
In neural machine translation, cross entropy (CE) is the standard loss function in two training methods of auto-regressive models, i.e., teacher forcing and scheduled sampling. In this paper, we propose mixed cross entropy loss (mixed CE) as a substitute for CE in both training approaches. In teacher forcing, the model trained with CE regards the translation problem as a one-to-one mapping process, while in mixed CE this process can be relaxed to one-to-many. In scheduled sampling, we show that mixed CE has the potential to encourage the training and testing behaviours to be similar to each other, more effectively mitigating the exposure bias problem. We demonstrate the superiority of mixed CE over CE on several machine translation datasets, WMT'16 Ro-En, WMT'16 Ru-En, and WMT'14 En-De in both teacher forcing and scheduled sampling setups. Furthermore, in WMT'14 En-De, we also find mixed CE consistently outperforms CE on a multi-reference set as well as a challenging paraphrased reference set. We also found the model trained with mixed CE is able to provide a better probability distribution defined over the translation output space. Our code is available at https://github.com/haorannlp/mix.
Key facts. Cited by 8 papers in the map (data through 7 Oct 2026). Builds on 39 of its 54 references that are in the map. First posted on arXiv 30 Jun 2021.
Builds on
39 of 54 references.
References · 39
- 2017Attention Is All You Needcited by 59,932
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2015Rethinking the Inception Architecture for Computer Visioncited by 5,451
- 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,530
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,016
- 2014Sequence to Sequence Learning with Neural Networkscited by 2,974
- 2015Neural Machine Translation of Rare Words with Subword Unitscited by 2,898
- 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,708
- 2019fairseq: A Fast, Extensible Toolkit for Sequence Modelingcited by 1,486
- 2019When Does Label Smoothing Help?cited by 985
- 2015Improving Neural Machine Translation Models with Monolingual Datacited by 784
- 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 779
27 more
- 2016Sequence-Level Knowledge Distillationcited by 761
- 2020Learning from Noisy Labels with Deep Neural Networks: A Surveycited by 580
- 2015Sequence Level Training with Recurrent Neural Networkscited by 568
- 2018Born Again Neural Networkscited by 539
- 2017Regularizing Neural Networks by Penalizing Confident Output Distributionscited by 461
- 2014Training Deep Neural Networks on Noisy Labels with Bootstrappingcited by 322
- 2017Synthetic and Natural Noise Both Break Neural Machine Translationcited by 304
- 2016Professor Forcing: A New Algorithm for Training Recurrent Networkscited by 214
- 2018Deterministic Non-Autoregressive Neural Sequence Modeling by Iterative Refinementcited by 194
- 2016An Actor-Critic Algorithm for Sequence Predictioncited by 158
- 2020On Exposure Bias, Hallucination and Domain Shift in Neural Machine Translationcited by 127
- 2016Sequence-to-Sequence Learning as Beam-Search Optimizationcited by 126
- 2018Analyzing Uncertainty in Neural Machine Translationcited by 126
- 2019Bridging the Gap between Training and Inference for Neural Machine Translationcited by 123
- 2019Generalization in Generation: A closer look at Exposure Biascited by 123
- 2015Minimum Risk Training for Neural Machine Translationcited by 114
- 2020BLEU might be Guilty but References are not Innocentcited by 109
- 2016Reward Augmented Maximum Likelihood for Neural Structured Predictioncited by 80
- 2019Mixture Models for Diverse Machine Translation: Tricks of the Tradecited by 77
- 2019Self-Knowledge Distillation in Natural Language Processingcited by 60
- 2017Classical Structured Prediction Losses for Sequence to Sequence Learningcited by 55
- 2019Scheduled Sampling for Transformerscited by 46
- 2018Denoising Neural Machine Translation Training with Trusted Data and Online Data Selectioncited by 43
- 2017SEARNN: Training RNNs with Global-Local Lossescited by 17
- 2019Parallel Scheduled Samplingcited by 17
- 2017Differentiable Scheduled Sampling for Credit Assignmentcited by 13
- 2019Controllable Dual Skew Divergence Loss for Neural Machine Translationcited by 2
Built on by
Citing papers per year: 2021: 1; 2022: 1; 2023: 2; 2024: 2; 2025: 1; 2026: 1.
8 papers.
- 2024GuardT2I: Defending Text-to-Image Models from Adversarial Promptscited by 61
- 2022Teacher Forcing Recovers Reward Functions for Text Generationcited by 11
- 2021Improving Scheduled Sampling with Elastic Weight Consolidation for Neural Machine Translationcited by 1
- 2023Magmaw: Modality-Agnostic Adversarial Attacks on Machine Learning-Based Wireless Communication Systemscited by 1
- 2023Dynamic Scheduled Sampling with Imitation Loss for Neural Text Generationcited by 0
- 2024Improving Autoregressive Training with Dynamic Oraclescited by 0
- 2025Relative-Based Scaling Law for Neural Language Modelscited by 0
- 2026LaCy: What Small Language Models Can and Should Learn is Not Just a Question of Losscited by 0
- 2026LaCy: What Small Language Models Can and Should Learn is Not Just a Question of Loss12 Feb 2026
- 2025Relative-Based Scaling Law for Neural Language Models23 Oct 2025
- 2024Improving Autoregressive Training with Dynamic Oracles13 Jun 2024
- 2024GuardT2I: Defending Text-to-Image Models from Adversarial Prompts3 Mar 2024
- 2023Magmaw: Modality-Agnostic Adversarial Attacks on Machine Learning-Based Wireless Communication Systems1 Nov 2023
- 2023Dynamic Scheduled Sampling with Imitation Loss for Neural Text Generation31 Jan 2023
- 2022Teacher Forcing Recovers Reward Functions for Text Generation17 Oct 2022
- 2021Improving Scheduled Sampling with Elastic Weight Consolidation for Neural Machine Translation13 Sep 2021