mbrs: A Library for Minimum Bayes Risk Decoding
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
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2019fairseq: A Fast, Extensible Toolkit for Sequence Modelingcited by 1,486
- 2020BLEURT: Learning Robust Metrics for Text Generationcited by 1,196
- 2020COMET: A Neural Framework for MT Evaluationcited by 1,024
- 2020Beyond English-Centric Multilingual Machine Translationcited by 559
- 2017Six Challenges for Neural Machine Translationcited by 334
- 2023xCOMET: Transparent Machine Translation Evaluation through Fine-grained Error Detectioncited by 260
- 2020Is MAP Decoding All You Need? The Inadequacy of the Mode in Neural Machine Translationcited by 137
- 2018Analyzing Uncertainty in Neural Machine Translationcited by 126
- 2021High Quality Rather than High Model Probability: Minimum Bayes Risk Decoding with Neural Metricscited by 104
- 2021Sampling-Based Approximations to Minimum Bayes Risk Decoding for Neural Machine Translationcited by 63
- 2021Understanding the Properties of Minimum Bayes Risk Decoding in Neural Machine Translationcited by 56
6 more
- 2023Faster Minimum Bayes Risk Decoding with Confidence-based Pruningcited by 34
- 2024Linear-time Minimum Bayes Risk Decoding with Reference Aggregationcited by 21
- 2024Efficient Minimum Bayes Risk Decoding using Low-Rank Matrix Completion Algorithmscited by 13
- 2024Centroid-Based Efficient Minimum Bayes Risk Decodingcited by 12
- 2024Generating Diverse and High-Quality Texts by Minimum Bayes Risk Decodingcited by 11
- 2023Model-Based Minimum Bayes Risk Decoding for Text Generationcited by 10
Built on by
Citing papers per year: 2024: 1; 2025: 5; 2026: 7.
13 papers.
Sort:
By area:
- 2025Revisiting Compositional Generalization Capability of Large Language Models Considering Instruction Following Abilitycited by 10
- 2025gec-metrics: A Unified Library for Grammatical Error Correction Evaluationcited by 9
- 2024Diversity Explains Inference Scaling Laws: Through a Case Study of Minimum Bayes Risk Decodingcited by 8
- 2025From SALAMANDRA to SALAMANDRATA: BSC Submission for WMT25 General Machine Translation Shared Taskcited by 3
- 2025Agreement-Constrained Probabilistic Minimum Bayes Risk Decodingcited by 2
- 2025Case-Based Decision-Theoretic Decoding with Quality Memoriescited by 1
- 2026Edit-level Majority Voting Mitigates Over-Correction in LLM-based Grammatical Error Correctioncited by 1
- 2026Maastricht University at AMIYA: Adapting LLMs for Dialectal Arabic using Fine-tuning and MBR Decodingcited by 0
- 2026HalluCiteChecker: A Lightweight Toolkit for Hallucinated Citation Detection and Verification in the Era of AI Scientistscited by 0
- 2026Enhancing Factuality through Consensus and Consistency in Summarization Using Minimum Bayes Risk Decodingcited by 0
- 2026MLLP-VRAIN UPV system for the IWSLT 2026 Simultaneous Speech Translation taskcited by 0
- 2026Non-Autoregressive Minimum Bayes' Risk Decoding for Fast Speech Recognitioncited by 0
- 2026Overfitting Mitigation via Singular Value Decomposition in Minimum Bayes Risk Decoding1 Sep 2026
- 2026Non-Autoregressive Minimum Bayes' Risk Decoding for Fast Speech Recognition16 Jun 2026
- 2026MLLP-VRAIN UPV system for the IWSLT 2026 Simultaneous Speech Translation task15 Jun 2026
- 2026Enhancing Factuality through Consensus and Consistency in Summarization Using Minimum Bayes Risk Decoding28 May 2026
- 2026Edit-level Majority Voting Mitigates Over-Correction in LLM-based Grammatical Error Correction13 May 2026
- 2026HalluCiteChecker: A Lightweight Toolkit for Hallucinated Citation Detection and Verification in the Era of AI Scientists29 Apr 2026
- 2026Maastricht University at AMIYA: Adapting LLMs for Dialectal Arabic using Fine-tuning and MBR Decoding10 Feb 2026
- 2025Agreement-Constrained Probabilistic Minimum Bayes Risk Decoding1 Dec 2025
- 2025Case-Based Decision-Theoretic Decoding with Quality Memories16 Sep 2025
- 2025From SALAMANDRA to SALAMANDRATA: BSC Submission for WMT25 General Machine Translation Shared Task18 Aug 2025
- 2025Revisiting Compositional Generalization Capability of Large Language Models Considering Instruction Following Ability18 Jun 2025
- 2025gec-metrics: A Unified Library for Grammatical Error Correction Evaluation26 May 2025
- 2025gec-metrics: A Unified Library for Grammatical Error Correction Evaluationcited by 9
- 2024Diversity Explains Inference Scaling Laws: Through a Case Study of Minimum Bayes Risk Decodingcited by 8
- 2025From SALAMANDRA to SALAMANDRATA: BSC Submission for WMT25 General Machine Translation Shared Taskcited by 3
- 2025Agreement-Constrained Probabilistic Minimum Bayes Risk Decodingcited by 2
- 2025Case-Based Decision-Theoretic Decoding with Quality Memoriescited by 1
- 2026Edit-level Majority Voting Mitigates Over-Correction in LLM-based Grammatical Error Correctioncited by 1
- 2026Maastricht University at AMIYA: Adapting LLMs for Dialectal Arabic using Fine-tuning and MBR Decodingcited by 0
- 2026HalluCiteChecker: A Lightweight Toolkit for Hallucinated Citation Detection and Verification in the Era of AI Scientistscited by 0
- 2026Enhancing Factuality through Consensus and Consistency in Summarization Using Minimum Bayes Risk Decodingcited by 0
- 2026MLLP-VRAIN UPV system for the IWSLT 2026 Simultaneous Speech Translation taskcited by 0
- 2026Non-Autoregressive Minimum Bayes' Risk Decoding for Fast Speech Recognitioncited by 0
- 2026Overfitting Mitigation via Singular Value Decomposition in Minimum Bayes Risk Decodingcited by 0
- 2026Overfitting Mitigation via Singular Value Decomposition in Minimum Bayes Risk Decoding1 Sep 2026
- 2026Non-Autoregressive Minimum Bayes' Risk Decoding for Fast Speech Recognition16 Jun 2026
- 2026MLLP-VRAIN UPV system for the IWSLT 2026 Simultaneous Speech Translation task15 Jun 2026
- 2026Enhancing Factuality through Consensus and Consistency in Summarization Using Minimum Bayes Risk Decoding28 May 2026
- 2026Edit-level Majority Voting Mitigates Over-Correction in LLM-based Grammatical Error Correction13 May 2026
- 2026HalluCiteChecker: A Lightweight Toolkit for Hallucinated Citation Detection and Verification in the Era of AI Scientists29 Apr 2026
- 2026Maastricht University at AMIYA: Adapting LLMs for Dialectal Arabic using Fine-tuning and MBR Decoding10 Feb 2026
- 2025Agreement-Constrained Probabilistic Minimum Bayes Risk Decoding1 Dec 2025
- 2025Case-Based Decision-Theoretic Decoding with Quality Memories16 Sep 2025
- 2025From SALAMANDRA to SALAMANDRATA: BSC Submission for WMT25 General Machine Translation Shared Task18 Aug 2025
- 2025gec-metrics: A Unified Library for Grammatical Error Correction Evaluation26 May 2025
- 2024Diversity Explains Inference Scaling Laws: Through a Case Study of Minimum Bayes Risk Decoding19 Oct 2024