English

Quantifying the Plausibility of Context Reliance in Neural Machine Translation

Computation and Language 2024-03-14 v2 Artificial Intelligence Human-Computer Interaction Machine Learning

Abstract

Establishing whether language models can use contextual information in a human-plausible way is important to ensure their trustworthiness in real-world settings. However, the questions of when and which parts of the context affect model generations are typically tackled separately, with current plausibility evaluations being practically limited to a handful of artificial benchmarks. To address this, we introduce Plausibility Evaluation of Context Reliance (PECoRe), an end-to-end interpretability framework designed to quantify context usage in language models' generations. Our approach leverages model internals to (i) contrastively identify context-sensitive target tokens in generated texts and (ii) link them to contextual cues justifying their prediction. We use \pecore to quantify the plausibility of context-aware machine translation models, comparing model rationales with human annotations across several discourse-level phenomena. Finally, we apply our method to unannotated model translations to identify context-mediated predictions and highlight instances of (im)plausible context usage throughout generation.

Keywords

Cite

@article{arxiv.2310.01188,
  title  = {Quantifying the Plausibility of Context Reliance in Neural Machine Translation},
  author = {Gabriele Sarti and Grzegorz Chrupała and Malvina Nissim and Arianna Bisazza},
  journal= {arXiv preprint arXiv:2310.01188},
  year   = {2024}
}

Comments

ICLR 2024 Camera Ready. Code: https://github.com/gsarti/pecore. Artifacts: https://huggingface.co/collections/gsarti/pecore-iclr-2024-65edab42e28439e21b612c2e