arXiv:2511.08305cs.LGcs.AI2025-11AAAI被引 3

通过黎曼优化增强模型推理多样性,提升复杂任务解题准确率。

Test-time Diverse Reasoning by Riemannian Activation Steering

  • 测试时基于黎曼流形优化,动态调整激活向量以扩大推理路径差异。
  • 在多个数学基准上,生成多样性和解题准确率均优于传统采样方法。
  • 无需标注数据,适合希望提升大模型推理能力的研究者与开发者。

Best-of-N 推理通过采样多个候选解并依据标准选择最优解,提升了语言模型在复杂任务中的准确性。该策略的关键瓶颈在于输出多样性不足:尽管采样具有随机性,模型仍可能生成相似结果,重复相同错误。为解决推理路径缺乏差异的问题,本文提出一种新颖的无监督激活转向策略,在测试阶段同时优化多条推理轨迹的转向向量。在批量生成过程的每个同步锚点,通过最大化所有可能干预激活子集所张成的总体积,寻找最优转向向量。该问题被建模为在球面乘积空间上的黎曼优化问题,目标函数为对数行列式。采用经过调优学习率的黎曼块坐标下降算法求解其驻点,并将所得转向向量应用至下一同步锚点。在主流数学基准上的实证评估表明,该测试时黎曼激活转向策略在生成多样性和解题准确率方面均显著优于原始采样方法。

原文摘要 · Abstract (English)

Best-of-$N$ reasoning improves the accuracy of language models in solving complex tasks by sampling multiple candidate solutions and then selecting the best one based on some criteria. A critical bottleneck for this strategy is the output diversity limit, which occurs when the model generates similar outputs despite stochastic sampling, and hence recites the same error. To address this lack of variance in reasoning paths, we propose a novel unsupervised activation steering strategy that simultaneously optimizes the steering vectors for multiple reasoning trajectories at test time. At any synchronization anchor along the batch generation process, we find the steering vectors that maximize the total volume spanned by all possible intervened activation subsets. We demonstrate that these steering vectors can be determined by solving a Riemannian optimization problem over the product of spheres with a log-determinant objective function. We then use a Riemannian block-coordinate descent algorithm with a well-tuned learning rate to obtain a stationary point of the problem, and we apply these steering vectors until the generation process reaches the subsequent synchronization anchor. Empirical evaluations on popular mathematical benchmarks demonstrate that our test-time Riemannian activation steering strategy outperforms vanilla sampling techniques in terms of generative diversity and solution accuracy.

推理增强黎曼优化大模型

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