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.
