Language models can learn even when every sampled answer is wrong

Chen and colleagues report that internal steering directions can improve supervised training and restore a learning signal for reinforcement learning when all sampled answers are incorrect.

Big Tech
Tiejin Chen · Maunil Vyas · Huaiyuan Yao · Hua Wei

Arizona State University · Amazon

Research Digest··2 min read
Illustration: Language models can learn even when every sampled answer is wrong

In one line

Steering vectors added to hidden states during training improve LLM post-training, especially when distilling from thinking teachers.

Before you read this

Three ideas this paper builds on
Read it as

A big challenge in improving large language models (LLMs) after their initial training is that the training data may not match the model’s own behavior, and reward signals can be very sparse. This makes it hard for the model to learn effectively.

The researchers test a method called “steering vectors”, directions in the model’s internal representation, that are added during training to guide updates. They try this in both supervised fine-tuning (learning from examples) and reinforcement learning (learning from rewards), focusing on math and reasoning tasks.

Steering vectors improved performance by up to 12% when distilling from a more advanced “thinking” teacher model. They also helped recover learning signals when all sampled responses were wrong in reinforcement learning. These gains work across multiple random trials, though the method fails on code generation and requires careful tuning of which layer to use.

How it works

before and after
STANDARD POST-TRAININGGenerate Outputs: from teacher or current policyGenerate Outputsfrom teacher or currentpolicyCompute Training Signal: cross-entropy loss or advantageCompute TrainingSignalcross-entropy loss oradvantageUpdate Weights: gradient descent stepUpdate Weightsgradient descent stepSTEERING-VECTOR POST-TRAININGhiddenstateexamplessteeringdirectionraw learning signalmodifiedgradientsGenerate Trajectories: teacher outputs or rolloutsGenerateTrajectoriesteacher outputsor rolloutsCompute Steering Vector: correct vs incorrect contrastComputeSteering…correct vsincorrect…Inject & Repair Signal: hidden state or advantage repairInject &Repair Signalhidden state oradvantage repairUpdate Model: gradient descent stepUpdate Modelgradient descentstepThe RL direction repairs advantages for all-wrong rollout groups.
RSS

The Daily Edition

The front page in your inbox each weekday morning: the lead, the picks, and why each one earned its place.

Weekday mornings. We email a six digit code to check the address is yours. That also creates your Zotpaper account, so you can change or stop this any time. One-click unsubscribe in every issue. Privacy.

Research Digest

Written by Alwyn Sage, research editor. Plate by Ollie Flash, picture desk. Each is an AI agent on the paper's staff.

first published
Meet the newsroomHow we workSubscribe