Minimum Bayes Risk Decoding for Error Span Detection in Reference-Free Automatic Machine Translation Evaluation

· Source: cs.CL updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Natural Language Processing · Depth: Expert, extended

Summary

Minimum Bayes Risk (MBR) decoding has been introduced for Error Span Detection (ESD) in reference-free automatic machine translation evaluation. State-of-the-art generative ESD methods typically use Maximum a Posteriori (MAP) decoding, which assumes model probabilities correlate with human annotation similarity; however, this assumption often fails. The proposed MBR decoding approach employs sentence- and span-level similarity metrics as utility functions to select candidate hypotheses. Experimental results on the WMT24 Metrics Shared Task, using models like Llama-3.3-70B-Inst and Gemma-2-27b-it, show MBR decoding, particularly with the novel SoftF1 utility function, significantly outperforms the MAP baseline across system, sentence, and span levels at N=256 candidate hypotheses. To address MBR's high computational cost, MBR distillation using Direct Preference Optimization (DPO) enables a standard greedy model to match MBR performance, eliminating inference-time latency.

Key takeaway

For Machine Learning Engineers developing or deploying automatic machine translation evaluation systems, you should consider implementing Minimum Bayes Risk (MBR) decoding. This approach, particularly with the SoftF1 utility function, significantly enhances error span detection accuracy over traditional MAP decoding. If computational cost is a concern, explore MBR distillation to achieve comparable performance with efficient greedy search, resolving inference latency bottlenecks in your production environments.

Key insights

Minimum Bayes Risk decoding improves Error Span Detection by optimizing for utility over model probability, especially with SoftF1.

Principles

Method

MBR decoding for ESD involves generating N candidate hypotheses, then selecting the one maximizing expected utility against support hypotheses using similarity functions like SoftF1. MBR distillation uses DPO to transfer this to a greedy model.

In practice

Topics

Code references

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CL updates on arXiv.org.