Minimum Bayes Risk Decoding for Error Span Detection in Reference-Free Automatic Machine Translation Evaluation
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
- MAP decoding's probability-utility correlation assumption often fails.
- MBR decoding maximizes expected utility, not just likelihood.
- Utility functions are critical for MBR performance.
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
- Use MBR decoding for higher ESD accuracy.
- Employ SoftF1 for robust span-level utility.
- Consider MBR distillation for efficient inference.
Topics
- Minimum Bayes Risk
- Error Span Detection
- Machine Translation Evaluation
- SoftF1 Utility Function
- MBR Distillation
- Large Language Models
Code references
Best for: Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CL updates on arXiv.org.