Saved in:
Bibliographic Details
Main Authors: Watts, Jake R., Sokol, Joel
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2407.16994
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915517757915136
author Watts, Jake R.
Sokol, Joel
author_facet Watts, Jake R.
Sokol, Joel
contents We propose an approach for preventing unsafe or otherwise low-quality large language model (LLM) outputs by leveraging the stochasticity of LLMs, an approach we call Repeated Checking with Regeneration (RCR). In this system, LLM checkers vote on the acceptability of a generated output, regenerating it if a threshold of disapproval is reached, until sufficient checkers approve. Based on our estimators for cost and failure rate and experimental data tailored to the application, our algorithm achieves a desired expected failure rate at Pareto-optimal cost. The failure rate provably decreases exponentially as a function of cost, and the models reasonably estimate the actual performance of such a system in action, even with limited data. This approach does not depend on the language model used, and could allow cheap, small LLMs to control, constrain, or at some tasks even outperform very complex and costly ones.
format Preprint
id arxiv_https___arxiv_org_abs_2407_16994
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle A Voter-Based Stochastic Rejection-Method Framework for Asymptotically Safe Language Model Outputs
Watts, Jake R.
Sokol, Joel
Artificial Intelligence
Computation and Language
Machine Learning
We propose an approach for preventing unsafe or otherwise low-quality large language model (LLM) outputs by leveraging the stochasticity of LLMs, an approach we call Repeated Checking with Regeneration (RCR). In this system, LLM checkers vote on the acceptability of a generated output, regenerating it if a threshold of disapproval is reached, until sufficient checkers approve. Based on our estimators for cost and failure rate and experimental data tailored to the application, our algorithm achieves a desired expected failure rate at Pareto-optimal cost. The failure rate provably decreases exponentially as a function of cost, and the models reasonably estimate the actual performance of such a system in action, even with limited data. This approach does not depend on the language model used, and could allow cheap, small LLMs to control, constrain, or at some tasks even outperform very complex and costly ones.
title A Voter-Based Stochastic Rejection-Method Framework for Asymptotically Safe Language Model Outputs
topic Artificial Intelligence
Computation and Language
Machine Learning
url https://arxiv.org/abs/2407.16994