CONTESTS: a Framework for Consistency Testing of Span Probabilities in Language Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wagner, Eitan, Slavutsky, Yuli, Abend, Omri
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914960570843136
author Wagner, Eitan
Slavutsky, Yuli
Abend, Omri
author_facet Wagner, Eitan
Slavutsky, Yuli
Abend, Omri
contents Although language model scores are often treated as probabilities, their reliability as probability estimators has mainly been studied through calibration, overlooking other aspects. In particular, it is unclear whether language models produce the same value for different ways of assigning joint probabilities to word spans. Our work introduces a novel framework, ConTestS (Consistency Testing over Spans), involving statistical tests to assess score consistency across interchangeable completion and conditioning orders. We conduct experiments on post-release real and synthetic data to eliminate training effects. Our findings reveal that both Masked Language Models (MLMs) and autoregressive models exhibit inconsistent predictions, with autoregressive models showing larger discrepancies. Larger MLMs tend to produce more consistent predictions, while autoregressive models show the opposite trend. Moreover, for both model types, prediction entropies offer insights into the true word span likelihood and therefore can aid in selecting optimal decoding strategies. The inconsistencies revealed by our analysis, as well their connection to prediction entropies and differences between model types, can serve as useful guides for future research on addressing these limitations.
format Preprint
id arxiv_https___arxiv_org_abs_2409_19984
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle CONTESTS: a Framework for Consistency Testing of Span Probabilities in Language Models
Wagner, Eitan
Slavutsky, Yuli
Abend, Omri
Computation and Language
Artificial Intelligence
Although language model scores are often treated as probabilities, their reliability as probability estimators has mainly been studied through calibration, overlooking other aspects. In particular, it is unclear whether language models produce the same value for different ways of assigning joint probabilities to word spans. Our work introduces a novel framework, ConTestS (Consistency Testing over Spans), involving statistical tests to assess score consistency across interchangeable completion and conditioning orders. We conduct experiments on post-release real and synthetic data to eliminate training effects. Our findings reveal that both Masked Language Models (MLMs) and autoregressive models exhibit inconsistent predictions, with autoregressive models showing larger discrepancies. Larger MLMs tend to produce more consistent predictions, while autoregressive models show the opposite trend. Moreover, for both model types, prediction entropies offer insights into the true word span likelihood and therefore can aid in selecting optimal decoding strategies. The inconsistencies revealed by our analysis, as well their connection to prediction entropies and differences between model types, can serve as useful guides for future research on addressing these limitations.
title CONTESTS: a Framework for Consistency Testing of Span Probabilities in Language Models
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2409.19984