CIE: Controlling Language Model Text Generations Using Continuous Signals

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Samuel, Vinay, Diddee, Harshita, Zhang, Yiming, Ippolito, Daphne
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866918144629538816
author Samuel, Vinay
Diddee, Harshita
Zhang, Yiming
Ippolito, Daphne
author_facet Samuel, Vinay
Diddee, Harshita
Zhang, Yiming
Ippolito, Daphne
contents Aligning language models (LMs) with user intent is becoming increasingly relevant to enhance user experience. This calls for designing methods that can allow users to control the properties of the language that LMs generate, for example, controlling the length of the generation or the complexity of the language that gets chosen. Most existing work attempts to integrate users' control by conditioning LM generations on natural language prompts or discrete control signals, which are often brittle and hard to scale. In this work, we are interested in continuous control signals, ones that exist along a spectrum that can't easily be captured in a natural language prompt or via existing techniques in conditional generation. Through a case study in controlling the precise response-length of generations, we demonstrate how an LM can be finetuned to expect a control vector that is interpolated between a "low" and a "high" token embedding. Our method more reliably exerts response-length control than in-context learning methods or fine-tuning methods that represent the control signal as a discrete signal.
format Preprint
id arxiv_https___arxiv_org_abs_2505_13448
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle CIE: Controlling Language Model Text Generations Using Continuous Signals
Samuel, Vinay
Diddee, Harshita
Zhang, Yiming
Ippolito, Daphne
Computation and Language
Artificial Intelligence
Aligning language models (LMs) with user intent is becoming increasingly relevant to enhance user experience. This calls for designing methods that can allow users to control the properties of the language that LMs generate, for example, controlling the length of the generation or the complexity of the language that gets chosen. Most existing work attempts to integrate users' control by conditioning LM generations on natural language prompts or discrete control signals, which are often brittle and hard to scale. In this work, we are interested in continuous control signals, ones that exist along a spectrum that can't easily be captured in a natural language prompt or via existing techniques in conditional generation. Through a case study in controlling the precise response-length of generations, we demonstrate how an LM can be finetuned to expect a control vector that is interpolated between a "low" and a "high" token embedding. Our method more reliably exerts response-length control than in-context learning methods or fine-tuning methods that represent the control signal as a discrete signal.
title CIE: Controlling Language Model Text Generations Using Continuous Signals
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2505.13448