Research map / Large language models / Reasoning and chain of thought
Exploring Diverse Generation Paths via Inference-time Stiefel Activation Steering
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Proposes STARS, a training-free inference-time method that steers hidden activations of concurrent generation runs with orthogonal additive vectors optimized on the Stiefel manifold (volume maximization), with a closed-form one-step update for per-token latency.
Problem
Language model generations collapse onto homogeneous high-probability outputs, so multiple concurrent sampled runs (e.g., for best-of-N or test generation) lack diversity; sampling-based fixes do not guarantee diversity across runs.
Approach
At each token, jointly compute orthogonal, fixed-norm additive steering vectors for all N concurrent runs by maximizing the log-determinant (volume) of the steered activation matrix on the scaled Stiefel manifold; a closed-form SVD-based one-step update with greedy stepsize replaces iterative Riemannian gradient descent for real-time use.
Key results
- On TESTEVAL with Qwen3-1.7B at T=1.0, STARS_0.5 raises overall line coverage from 59.95% (sampling) to 91.05%, and branch coverage from 55.05% to 86.81%.
- On LiveIdeaBench with Qwen2.5-3B-Instruct at T=0.2, STARS_0.5 improves fluency from 2.68 (sampling) to 5.09 and average score from 5.01 to 5.59.
- Reported one-step update reaches about 2% optimality gap using about 3% of the runtime of full Riemannian gradient descent (appendix claim).
Surprising findings
- Standard temperature sampling on Gemma-1.1-2b-it gives near-zero execution and assertion correctness scores that STARS improves only marginally, while on Qwen3-1.7B STARS raises correctness substantially at all temperatures.
- Per-question runtime overhead is small on most settings but reaches about 2 seconds (Qwen2.5-3B: 3.02s to 5.01s), which the authors call negligible.
Limitations
- No theoretical convergence guarantee for the one-step heuristic update; only empirical optimality gap evidence.
- Evaluated only on two benchmarks (test case generation and idea generation) with small models (1.7B-3B).
- Steering hyperparameter alpha and layer choice require tuning on a subset of data.
Experimental setup
Datasets: TESTEVAL LiveIdeaBench
Benchmarks: TESTEVAL LiveIdeaBench
Metrics: syntax correctness execution correctness assertion correctness line coverage branch coverage cov@k originality feasibility clarity fluency
Baselines: temperature sampling
Builds on: activation steering polar retraction temperature sampling
Closest papers by meaning
- Test-time Diverse Reasoning by Riemannian Activation Steering (AAAI 2026 · similarity 0.88)
- Conditional Optimal Bridge for Riemannian Activation Steering (NeurIPS 2026 · similarity 0.87)
- Training-Free Guidance Beyond Differentiability: Scalable Path Steering with Tree Search in Diffusion and Flow Models (NeurIPS 2025 · similarity 0.85)
- Test-Time Scaling in the Wild: Why Exploitation, Not Exploration, Is the Bottleneck (NeurIPS 2026 · similarity 0.85)
- Efficient Test-Time Scaling via Hierarchical Search and Self-Verification for Discrete Diffusion Language Models (ICML 2026 · similarity 0.85)
- ATTS: Asynchronous Test-Time Scaling via Conformal Prediction (ICLR 2026 · similarity 0.85)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
