dorsal/arxiv
View SchemaGeoSteer: Faithful Chain-of-Thought Steering via Latent Manifold Gradients
| Authors | Kentaro Kazama, Daiki Shirafuji, Tatsuhiko Saito |
|---|---|
| Categories | |
| ArXiv ID | 2601.10229vv2 |
| URL | https://arxiv.org/abs/2601.10229 |
| License | http://creativecommons.org/licenses/by-nc-nd/4.0/ |
Abstract
Recent advances in Large Language Models (LLMs) have improved multi-step reasoning. Most approaches rely on Chain-of-Thought (CoT) rationales. Previous studies have shown that LLMs often generate logically inconsistent reasoning steps even when their final answers are correct. These inconsistencies reduce the reliability of step-level reasoning. We propose GeoSteer, a manifold-based framework that improves the quality of intermediate reasoning. The method consists of: (1) constructing a CoT dataset with segment-level scores, (2) training a Variational Autoencoder (VAE) model and a quality estimation model to learn a low-dimensional manifold of high-quality CoT trajectories, and (3) steering hidden states of target LLMs toward higher-quality regions in the latent space. This update in a latent space behaves like a natural-gradient adjustment in the original hidden-state space. It ensures geometrically coherent steering. We evaluate GeoSteer on the GSM8k dataset using the Qwen3 series. We measure via answer accuracy and overall reasoning performance. GeoSteer improved the exact match accuracy by up to 2.6 points. It also enhanced the pairwise win rate by 5.3 points. These results indicate that GeoSteer provides an effective and controllable mechanism for improving the quality of intermediate reasoning in LLMs.
{
"annotation_id": "db9ba933-a9de-458c-a81b-7e0ac004fc13",
"date_created": "2026-02-17T05:53:23.604000Z",
"date_modified": "2026-02-17T05:53:23.604000Z",
"file_hash": "5a8464005cc7274f0bff0102fb2cc6faca4d6d210fcafe0608c51bb96eb621b6",
"private": false,
"record": {
"abstract": "Recent advances in Large Language Models (LLMs) have improved multi-step reasoning. Most approaches rely on Chain-of-Thought (CoT) rationales. Previous studies have shown that LLMs often generate logically inconsistent reasoning steps even when their final answers are correct. These inconsistencies reduce the reliability of step-level reasoning. We propose GeoSteer, a manifold-based framework that improves the quality of intermediate reasoning. The method consists of: (1) constructing a CoT dataset with segment-level scores, (2) training a Variational Autoencoder (VAE) model and a quality estimation model to learn a low-dimensional manifold of high-quality CoT trajectories, and (3) steering hidden states of target LLMs toward higher-quality regions in the latent space. This update in a latent space behaves like a natural-gradient adjustment in the original hidden-state space. It ensures geometrically coherent steering. We evaluate GeoSteer on the GSM8k dataset using the Qwen3 series. We measure via answer accuracy and overall reasoning performance. GeoSteer improved the exact match accuracy by up to 2.6 points. It also enhanced the pairwise win rate by 5.3 points. These results indicate that GeoSteer provides an effective and controllable mechanism for improving the quality of intermediate reasoning in LLMs.",
"arxiv_id": "2601.10229",
"authors": [
"Kentaro Kazama",
"Daiki Shirafuji",
"Tatsuhiko Saito"
],
"categories": [
"cs.CL"
],
"license": "http://creativecommons.org/licenses/by-nc-nd/4.0/",
"title": "GeoSteer: Faithful Chain-of-Thought Steering via Latent Manifold Gradients",
"url": "https://arxiv.org/abs/2601.10229",
"version": "v2"
},
"schema_id": "dorsal/arxiv",
"source": {
"execution_id": "2279ce97-d3fb-4ec7-86ff-cad8703f7d5c",
"id": "arXiv Dataset",
"type": "Model",
"variant": "snapshot-2026-01-17",
"version": "0.1.0"
},
"user_id": 1000002
}