← Search

Kenneth Li

5 accepted papers

2024

Q-Probe: A Lightweight Approach to Reward Maximization for Language Models

ICML 2024poster

We present an approach called Q-probing to adapt a pre-trained language model to maximize a task-specific reward function. At a high level, Q-probing sits between heavier approaches such as finetuning and lighter approaches such as few shot prompting, but can also be combined with either. The idea i…

2023

Emergent World Representations: Exploring a Sequence Model Trained on a Synthetic Task

ICLR 2023top-5%

Language models show a surprising range of capabilities, but the source of their apparent competence is unclear. Do these networks just memorize a collection of surface statistics, or do they rely on internal representations of the process that generates the sequences they see? We investigate this q…

2023

Inference-Time Intervention: Eliciting Truthful Answers from a Language Model

NeurIPS 2023spotlight

We introduce Inference-Time Intervention (ITI), a technique designed to enhance the "truthfulness" of large language models (LLMs). ITI operates by shifting model activations during inference, following a learned set of directions across a limited number of attention heads. This intervention signifi…