Forcing Diffuse Distributions out of Language Models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zhang, Yiming, Schwarzschild, Avi, Carlini, Nicholas, Kolter, Zico, Ippolito, Daphne
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929452003360768
author Zhang, Yiming
Schwarzschild, Avi
Carlini, Nicholas
Kolter, Zico
Ippolito, Daphne
author_facet Zhang, Yiming
Schwarzschild, Avi
Carlini, Nicholas
Kolter, Zico
Ippolito, Daphne
contents Despite being trained specifically to follow user instructions, today's instructiontuned language models perform poorly when instructed to produce random outputs. For example, when prompted to pick a number uniformly between one and ten Llama-2-13B-chat disproportionately favors the number five, and when tasked with picking a first name at random, Mistral-7B-Instruct chooses Avery 40 times more often than we would expect based on the U.S. population. When these language models are used for real-world tasks where diversity of outputs is crucial, such as language model assisted dataset construction, their inability to produce diffuse distributions over valid choices is a major hurdle. In this work, we propose a fine-tuning method that encourages language models to output distributions that are diffuse over valid outcomes. The methods we introduce generalize across a variety of tasks and distributions and make large language models practical for synthetic dataset generation with little human intervention.
format Preprint
id arxiv_https___arxiv_org_abs_2404_10859
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Forcing Diffuse Distributions out of Language Models
Zhang, Yiming
Schwarzschild, Avi
Carlini, Nicholas
Kolter, Zico
Ippolito, Daphne
Computation and Language
Machine Learning
Despite being trained specifically to follow user instructions, today's instructiontuned language models perform poorly when instructed to produce random outputs. For example, when prompted to pick a number uniformly between one and ten Llama-2-13B-chat disproportionately favors the number five, and when tasked with picking a first name at random, Mistral-7B-Instruct chooses Avery 40 times more often than we would expect based on the U.S. population. When these language models are used for real-world tasks where diversity of outputs is crucial, such as language model assisted dataset construction, their inability to produce diffuse distributions over valid choices is a major hurdle. In this work, we propose a fine-tuning method that encourages language models to output distributions that are diffuse over valid outcomes. The methods we introduce generalize across a variety of tasks and distributions and make large language models practical for synthetic dataset generation with little human intervention.
title Forcing Diffuse Distributions out of Language Models
topic Computation and Language
Machine Learning
url https://arxiv.org/abs/2404.10859