Skip to content

A Coin Flip Per Token: Bernoulli Sparse Steering of Large Language Models

Jul 2026 · arXiv.org · Vol abs/2607.05615 · 0 citations · 32 references
Computer Science

TL;DR

Stochastic Token Steering (STS), which gates each token independently with probability $p$, and Stochastic Block Steering (SBS), which gates a leading window once per sequence are introduced; neither requires a reward model or learned gating policy.

Abstract

Activation steering via sparse autoencoders (SAEs) enables behavioral control of large language models without task-specific fine-tuning, but standard methods apply the steering signal at every generated token, incurring constant per-token perturbation that risks degrading fluency. We ask: is dense intervention necessary? We introduce Stochastic Token Steering (STS), which gates each token independently with probability $p$, and Stochastic Block Steering (SBS), which gates a leading window once per sequence; neither requires a reward model or learned gating policy. Across two model families and two behavioral tasks, steering only 50% of the tokens recovers most of the dense-steering effect while preserving fluency, and steering as few as 30% surpasses prompt-based control. The optimal steering magnitude scales inversely with the intervention ratio, revealing that SAE-mediated control is rate-limited: the behavioral outcome depends on cumulative signal dosage across a sequence.

View source

Similar papers

2026

Statistically Grounded Sparse-Feature Interventions for Activation-Space Control in Large Language Models

This work introduces a transparent SAE-feature steering pipeline that first applies a six-condition reliability filter, then ranks sparse features through an unweighted Borda consensus over three complementary statistics: F -test, KSG mutual information, and Cohen's d .

Oshayer Siddique, J. M. A. U. Alam, M. Rafy et al. · 0 citations
#natural language process... Preprint Sep 2026

GAPS: Dimension-Level Gates for Conditional Activation Steering

Activation steering suppresses undesired behaviors in language models by adding a steering vector to the hidden state during generation. Recent conditional methods such as CAST and DSAS improve the behavior-capability trade-off by deciding when to intervene, but once active, they apply the full dense vector to all hidd...

Moghis Fereidouni, Muhammad Umair Haider, Hassan Sajjad et al. · 0 citations
Preprint Jul 2026

Forecasting Side Effects of Activation Steering

Activation steering modifies a language model by adding a learned direction to its hidden activations, enabling targeted behavioral changes without retraining. While effective, steering often produces unintended side effects on other behaviors, making it difficult to deploy safely. We therefore ask: can these side effe...

Yong-Ong Chong, Alson Wei Jie Sim, Peixin Zhang et al. · 0 citations
Jul 2026

Frustratingly Simple Black-Box Adaptation of Language Models via Logit Bias

This work develops a black-box method for learning a single context-independent logit-bias vector, added at every decoding step, without modifying model weights or requiring gradients, and suggests that learned logit bias is a lightweight mechanism for adapting language models under minimal access requirements.

Ofek Cohen, Lior Shani, Aviv Rosenberg et al. · 0 citations
Preprint Aug 2026

XTC: Head-Aware Sampling by Excluding Top Choices

This work introduces XTC (Exclude Top Choices), a lightweight head-aware decoding operator that targets this regime of open-ended generation in which several continuations are plausible but too much probability mass remains concentrated on the most generic choice.

Philippe Weidmann, Allen G. Roush, Judah Goldfeder et al. · 0 citations
Book Open access Aug 2026

Multi-Turn Reinforcement Learning for Large Language Models: From Theory to Practice with Amazon SageMaker AI

This hands-on problem-solving tutorial provides both a rigorous algorithmic and practical introduction to multi-turn RL finetuning for LLMs, and covers state-of-the-art multi-turn RL finetuning algorithms, turn-level vs. trajectory-level reward design, and production grade monitoring for reward hacking detection.

Zhe Wang, Sapana Chaudhary, Jiayu Li et al. · 0 citations

Related blog posts

Microsoft Research Blog Oct 12, 2026

Simulating student learning for better AI tutoring

StudentSim is a research system for evaluating and improving AI tutoring tools before they are tested with learners. It models how individuals may engage with tutors, giving feedback to improve tutor performance across subjects like chess, writing, and mathematics. The post Simulating student learning for better AI tutoring appeared first on Microsoft Research.

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.