mlsanityData to 7 Oct 2026

mbrs: A Library for Minimum Bayes Risk Decoding

arXiv 2408.04167 ↗First posted 8 Aug 2024cs.CLcited by 13

Not on the map. Why?

Abstract (arXiv)

Minimum Bayes risk (MBR) decoding is a decision rule of text generation tasks that outperforms conventional maximum a posterior (MAP) decoding using beam search by selecting high-quality outputs based on a utility function rather than those with high-probability. Typically, it finds the most suitable hypothesis from the set of hypotheses under the sampled pseudo-references. mbrs is a library of MBR decoding, which can flexibly combine various metrics, alternative expectation estimations, and algorithmic variants. It is designed with a focus on speed measurement and calling count of code blocks, transparency, reproducibility, and extensibility, which are essential for researchers and developers. We published our mbrs as an MIT-licensed open-source project, and the code is available on GitHub. GitHub: https://github.com/naist-nlp/mbrs

Key facts. Cited by 13 papers in the map (data through 7 Oct 2026). Builds on 18 of its 32 references that are in the map. First posted on arXiv 8 Aug 2024.

Builds on

18 of 32 references.

References · 18

  1. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
  2. 2019fairseq: A Fast, Extensible Toolkit for Sequence Modelingcited by 1,486
  3. 2020BLEURT: Learning Robust Metrics for Text Generationcited by 1,196
  4. 2020COMET: A Neural Framework for MT Evaluationcited by 1,024
  5. 2020Beyond English-Centric Multilingual Machine Translationcited by 559
  6. 2017Six Challenges for Neural Machine Translationcited by 334
  7. 2023xCOMET: Transparent Machine Translation Evaluation through Fine-grained Error Detectioncited by 260
  8. 2020Is MAP Decoding All You Need? The Inadequacy of the Mode in Neural Machine Translationcited by 137
  9. 2018Analyzing Uncertainty in Neural Machine Translationcited by 126
  10. 2021High Quality Rather than High Model Probability: Minimum Bayes Risk Decoding with Neural Metricscited by 104
  11. 2021Sampling-Based Approximations to Minimum Bayes Risk Decoding for Neural Machine Translationcited by 63
  12. 2021Understanding the Properties of Minimum Bayes Risk Decoding in Neural Machine Translationcited by 56
6 more
  1. 2023Faster Minimum Bayes Risk Decoding with Confidence-based Pruningcited by 34
  2. 2024Linear-time Minimum Bayes Risk Decoding with Reference Aggregationcited by 21
  3. 2024Efficient Minimum Bayes Risk Decoding using Low-Rank Matrix Completion Algorithmscited by 13
  4. 2024Centroid-Based Efficient Minimum Bayes Risk Decodingcited by 12
  5. 2024Generating Diverse and High-Quality Texts by Minimum Bayes Risk Decodingcited by 11
  6. 2023Model-Based Minimum Bayes Risk Decoding for Text Generationcited by 10

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 1; 2025: 5; 2026: 7.

13 papers.

Sort:

By area: