arXiv:2601.10229cs.CL2026-01被引 3

通过潜空间梯度引导,让大模型推理更连贯可信

GeoSteer: Faithful Chain-of-Thought Steering via Latent Manifold Gradients

  • 基于潜流形梯度,动态调整模型中间推理状态
  • 在GSM8k上提升准确率0.9点,推理质量提升4.5点
  • 适合需要可解释、高可靠性推理的场景

大语言模型(LLM)在链式思考(CoT)推理方面取得显著进展,但常出现答案正确却推理步骤逻辑不一致的问题,影响推理可靠性。本文提出GeoSteer,一种基于流形的框架,以提升中间推理质量。方法包括:(1) 构建带步骤评分的CoT数据集;(2) 训练变分自编码器(VAE)与质量评估模型,学习高质量CoT轨迹的低维潜空间;(3) 沿学习到的流形梯度,引导目标LLM的隐藏状态向高质量区域迁移。该过程实现几何上连贯的推理引导。在GSM8k数据集上使用Qwen3系列模型进行评估,结果表明:相比原始LLM,GeoSteer使准确率提升0.9点,推理质量平均提升4.5点。这证明其为改进大模型中间推理质量提供了有效且可控的新机制。

原文摘要 · Abstract (English)

Recent advances in Large Language Models (LLMs) have demonstrated remarkable progress in their reasoning capabilities, such as Chain-of-Thought (CoT). Most approaches rely on 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 the reasoning process. We propose GeoSteer, a manifold-based framework that improves the quality of intermediate reasoning. The method consists of: (1) constructing a CoT dataset with step-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 last step enables steering of the hidden states by following gradients along the learned manifold. It facilitates geometrically coherent steering. Evaluation experiments were conducted on the GSM8k dataset using the Qwen3 series. We evaluated performance using two metrics: answer accuracy and overall reasoning quality. GeoSteer improved the accuracy by 0.9 points and enhanced the reasoning quality by 4.5 points on average, compared with those of original LLMs. These results indicate that GeoSteer improves an effective and controllable mechanism for improving the quality of intermediate reasoning in LLMs.

链式思考推理优化潜空间

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。