arXiv:2511.11518cs.CL2025-11AAAI被引 5

用树搜索让大模型生成更符合人类偏好,不改模型也能精准调优。

W2S-AlignTree: Weak-to-Strong Inference-Time Alignment for Large Language Models via Monte Carlo Tree Search

  • 通过蒙特卡洛树搜索结合弱模型信号,动态引导强模型生成。
  • 在摘要任务中使Llama3-8B性能从1.89提升至2.19,相对提高15.9%。
  • 无需训练、可插即用,适合需要灵活控制输出的场景。

大型语言模型虽能力强大,但其输出常因弱监督不足而偏离人类偏好,且难以实现细粒度控制。传统训练时对齐方法如基于人类反馈的强化学习(RLHF)依赖昂贵专家标注,且推理阶段缺乏动态调控能力。为此,我们提出W2S-AlignTree,首个将蒙特卡洛树搜索(MCTS)与弱到强泛化范式结合的即插即用型推理时对齐框架。该方法将对齐问题建模为生成搜索树中的最优启发式搜索,利用弱模型的实时、逐步信号作为对齐代理,并引入熵感知探索机制,在不修改强模型参数的前提下实现细粒度生成引导。该方法在高维生成空间中动态平衡探索与利用。在情感控制生成、摘要和指令遵循任务上的实验表明,W2S-AlignTree持续优于强基线。特别地,在摘要任务中,其将Llama3-8B性能从1.89提升至2.19,相对提升15.9%。

原文摘要 · Abstract (English)

Large Language Models (LLMs) demonstrate impressive capabilities, yet their outputs often suffer from misalignment with human preferences due to the inadequacy of weak supervision and a lack of fine-grained control. Training-time alignment methods like Reinforcement Learning from Human Feedback (RLHF) face prohibitive costs in expert supervision and inherent scalability limitations, offering limited dynamic control during inference. Consequently, there is an urgent need for scalable and adaptable alignment mechanisms. To address this, we propose W2S-AlignTree, a pioneering plug-and-play inference-time alignment framework that synergistically combines Monte Carlo Tree Search (MCTS) with the Weak-to-Strong Generalization paradigm for the first time. W2S-AlignTree formulates LLM alignment as an optimal heuristic search problem within a generative search tree. By leveraging weak model's real-time, step-level signals as alignment proxies and introducing an Entropy-Aware exploration mechanism, W2S-AlignTree enables fine-grained guidance during strong model's generation without modifying its parameters. The approach dynamically balances exploration and exploitation in high-dimensional generation search trees. Experiments across controlled sentiment generation, summarization, and instruction-following show that W2S-AlignTree consistently outperforms strong baselines. Notably, W2S-AlignTree raises the performance of Llama3-8B from 1.89 to 2.19, a relative improvement of 15.9 on the summarization task.

大模型对齐树搜索推理优化

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