Improving the Calibration of Confidence Scores in Text Generation Using the Output Distribution's Characteristics
Fuente:
arXiv
Enregistré dans:
| Auteurs principaux: | , , |
|---|---|
| Format: | Preprint |
| Publié: |
2025
|
| Sujets: | |
| Accès en ligne: | |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866913890725527552 |
|---|---|
| author | Flores, Lorenzo Jaime Yu Ernst, Ori Cheung, Jackie Chi Kit |
| author_facet | Flores, Lorenzo Jaime Yu Ernst, Ori Cheung, Jackie Chi Kit |
| contents | Well-calibrated model confidence scores can improve the usefulness of text generation models. For example, users can be prompted to review predictions with low confidence scores, to prevent models from returning bad or potentially dangerous predictions. However, confidence metrics are not always well calibrated in text generation. One reason is that in generation, there can be many valid answers, which previous methods do not always account for. Hence, a confident model could distribute its output probability among multiple sequences because they are all valid. We propose task-agnostic confidence metrics suited to generation, which rely solely on the probabilities associated with the model outputs without the need for further fine-tuning or heuristics. Using these, we are able to improve the calibration of BART and Flan-T5 on summarization, translation, and QA datasets. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2506_00637 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | Improving the Calibration of Confidence Scores in Text Generation Using the Output Distribution's Characteristics Flores, Lorenzo Jaime Yu Ernst, Ori Cheung, Jackie Chi Kit Computation and Language Artificial Intelligence Well-calibrated model confidence scores can improve the usefulness of text generation models. For example, users can be prompted to review predictions with low confidence scores, to prevent models from returning bad or potentially dangerous predictions. However, confidence metrics are not always well calibrated in text generation. One reason is that in generation, there can be many valid answers, which previous methods do not always account for. Hence, a confident model could distribute its output probability among multiple sequences because they are all valid. We propose task-agnostic confidence metrics suited to generation, which rely solely on the probabilities associated with the model outputs without the need for further fine-tuning or heuristics. Using these, we are able to improve the calibration of BART and Flan-T5 on summarization, translation, and QA datasets. |
| title | Improving the Calibration of Confidence Scores in Text Generation Using the Output Distribution's Characteristics |
| topic | Computation and Language Artificial Intelligence |
| url | https://arxiv.org/abs/2506.00637 |