Contrastive Token Learning with Similarity Decay for Repetition Suppression in Machine Translation

Huangyu Dai, Ben Chen, Kaidi Chen, Ying Han, Zihan Liang +1 more
9/30/2024
cs.CLcs.AI

No confident code implementation found for this paper.

Abstract

For crosslingual conversation and trade, Neural Machine Translation (NMT) is pivotal yet faces persistent challenges with monotony and repetition in generated content. Traditional solutions that rely on penalizing text redundancy or token reoccurrence have shown limited efficacy, particularly for lengthy article and e-commerce descriptions with inherent redundancy, even with the advent of Large Language Models (LLMs). This paper investigates the underlying causes of textual repetition through the lens of information entropy, attributing the phenomenon to the elevated uncertainty within the input text. To address this, a novel algorithm named Contrastive Token Learning with Similarity Decay (CTSD) is introduced, which modulates the suppression of tokens dynamically, informed by varying attention weights and inter-token distances. Furthermore, an e-commerce dataset comprised of title texts of online real items is compiled and released susceptible to hallucination translations to benchmark the algorithm. Extensive evaluations demonstrate that CTSD significantly outperforms existing approaches in precision and generalizability. Additional online A/B testing underscores its practical value, showing marked improvements in user engagement and conversion. Notably, this method has been implemented with full traffic on eight multilingual sites of alibaba.com, the largest B2B e-commerce platform in the world.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{dai2024contrastive,
  title  = {Contrastive Token Learning with Similarity Decay for Repetition Suppression in Machine Translation},
  author = {Huangyu Dai and Ben Chen and Kaidi Chen and Ying Han and Zihan Liang and Wen Jiang},
  year   = {2024},
  eprint = {2409.19877},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2409.19877v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2409.19877.svg)](https://codeofpaper.com/paper/2409.19877)

Discussion