Are Generative Models Underconfident? Better Quality Estimation with Boosted Model Probability

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Dinh, Tu Anh, Niehues, Jan
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912586740531200
author Dinh, Tu Anh
Niehues, Jan
author_facet Dinh, Tu Anh
Niehues, Jan
contents Quality Estimation (QE) is estimating quality of the model output during inference when the ground truth is not available. Deriving output quality from the models' output probability is the most trivial and low-effort way. However, we show that the output probability of text-generation models can appear underconfident. At each output step, there can be multiple correct options, making the probability distribution spread out more. Thus, lower probability does not necessarily mean lower output quality. Due to this observation, we propose a QE approach called BoostedProb, which boosts the model's confidence in cases where there are multiple viable output options. With no increase in complexity, BoostedProb is notably better than raw model probability in different settings, achieving on average +0.194 improvement in Pearson correlation to ground-truth quality. It also comes close to or outperforms more costly approaches like supervised or ensemble-based QE in certain settings.
format Preprint
id arxiv_https___arxiv_org_abs_2502_11115
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Are Generative Models Underconfident? Better Quality Estimation with Boosted Model Probability
Dinh, Tu Anh
Niehues, Jan
Computation and Language
I.2.7
Quality Estimation (QE) is estimating quality of the model output during inference when the ground truth is not available. Deriving output quality from the models' output probability is the most trivial and low-effort way. However, we show that the output probability of text-generation models can appear underconfident. At each output step, there can be multiple correct options, making the probability distribution spread out more. Thus, lower probability does not necessarily mean lower output quality. Due to this observation, we propose a QE approach called BoostedProb, which boosts the model's confidence in cases where there are multiple viable output options. With no increase in complexity, BoostedProb is notably better than raw model probability in different settings, achieving on average +0.194 improvement in Pearson correlation to ground-truth quality. It also comes close to or outperforms more costly approaches like supervised or ensemble-based QE in certain settings.
title Are Generative Models Underconfident? Better Quality Estimation with Boosted Model Probability
topic Computation and Language
I.2.7
url https://arxiv.org/abs/2502.11115