Centroid-Based Efficient Minimum Bayes Risk Decoding
Not on the map. Why?
Abstract (arXiv)
Minimum Bayes risk (MBR) decoding achieved state-of-the-art translation performance by using COMET, a neural metric that has a high correlation with human evaluation. However, MBR decoding requires quadratic time since it computes the expected score between a translation hypothesis and all reference translations. We propose centroid-based MBR (CBMBR) decoding to improve the speed of MBR decoding. Our method clusters the reference translations in the feature space, and then calculates the score using the centroids of each cluster. The experimental results show that our CBMBR not only improved the decoding speed of the expected score calculation 5.7 times, but also outperformed vanilla MBR decoding in translation quality by up to 0.5 COMET in the WMT'22 EnleftrightarrowJa, EnleftrightarrowDe, EnleftrightarrowZh, and WMT'23 EnleftrightarrowJa translation tasks.
Key facts. Cited by 12 papers in the map (data through 7 Oct 2026). Builds on 15 of its 32 references that are in the map. First posted on arXiv 17 Feb 2024.
Builds on
15 of 32 references.
References · 15
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2019BERTScore: Evaluating Text Generation with BERTcited by 5,462
- 2019Unsupervised Cross-lingual Representation Learning at Scalecited by 3,907
- 2020BLEURT: Learning Robust Metrics for Text Generationcited by 1,196
- 2020COMET: A Neural Framework for MT Evaluationcited by 1,024
- 2017SemEval-2017 Task 1: Semantic Textual Similarity - Multilingual and Cross-lingual Focused Evaluationcited by 998
- 2016Bag of Tricks for Efficient Text Classificationcited by 869
- 2020Language-agnostic BERT Sentence Embeddingcited by 621
- 2020Beyond English-Centric Multilingual Machine Translationcited by 559
- 2020Is MAP Decoding All You Need? The Inadequacy of the Mode in Neural Machine Translationcited by 137
- 2021Sampling-Based Approximations to Minimum Bayes Risk Decoding for Neural Machine Translationcited by 63
- 2023Epsilon Sampling Rocks: Investigating Sampling Strategies for Minimum Bayes Risk Decoding for Machine Translationcited by 57
Built on by
Citing papers per year: 2024: 6; 2025: 5; 2026: 1.
12 papers.
- 2024Mitigating Metric Bias in Minimum Bayes Risk Decodingcited by 15
- 2024Improving Minimum Bayes Risk Decoding with Multi-Promptcited by 13
- 2024mbrs: A Library for Minimum Bayes Risk Decodingcited by 13
- 2024Diversity Explains Inference Scaling Laws: Through a Case Study of Minimum Bayes Risk Decodingcited by 8
- 2024A Bayesian Optimization Approach to Machine Translation Rerankingcited by 2
- 2025Early-Exit and Instant Confidence Translation Quality Estimationcited by 2
- 2025Re-evaluating Minimum Bayes Risk Decoding for Automatic Speech Recognitioncited by 2
- 2025Agreement-Constrained Probabilistic Minimum Bayes Risk Decodingcited by 2
- 2025Case-Based Decision-Theoretic Decoding with Quality Memoriescited by 1
- 2024Unveiling the Power of Source: Source-based Minimum Bayes Risk Decoding for Neural Machine Translationcited by 0
- 2025Minimum Bayes Risk Decoding for Error Span Detection in Reference-Free Automatic Machine Translation Evaluationcited by 0
- 2026Quit While You're Ahead: Quit for Efficient Candidate Generation in Machine Translation Rerankingcited by 0
- 2026Quit While You're Ahead: Quit for Efficient Candidate Generation in Machine Translation Reranking1 Sep 2026
- 2025Minimum Bayes Risk Decoding for Error Span Detection in Reference-Free Automatic Machine Translation Evaluation8 Dec 2025
- 2025Agreement-Constrained Probabilistic Minimum Bayes Risk Decoding1 Dec 2025
- 2025Re-evaluating Minimum Bayes Risk Decoding for Automatic Speech Recognition22 Oct 2025
- 2025Case-Based Decision-Theoretic Decoding with Quality Memories16 Sep 2025
- 2025Early-Exit and Instant Confidence Translation Quality Estimation20 Feb 2025
- 2024A Bayesian Optimization Approach to Machine Translation Reranking14 Nov 2024
- 2024Mitigating Metric Bias in Minimum Bayes Risk Decoding5 Nov 2024
- 2024Diversity Explains Inference Scaling Laws: Through a Case Study of Minimum Bayes Risk Decoding19 Oct 2024
- 2024mbrs: A Library for Minimum Bayes Risk Decoding8 Aug 2024
- 2024Improving Minimum Bayes Risk Decoding with Multi-Prompt22 Jul 2024
- 2024Unveiling the Power of Source: Source-based Minimum Bayes Risk Decoding for Neural Machine Translation17 Jun 2024