arXiv:2504.16760cs.AI2025-04被引 3

用小模型从大模型隐状态中提取判断结果,省下大量算力。

Lightweight Latent Verifiers for Efficient Meta-Generation Strategies

  • 从大模型隐状态直接提取正确性信号,不依赖额外大模型。
  • 相比传统方法,计算成本降低至几十分之一,仍保持高准确率。
  • 适合资源有限的场景,尤其适合小模型做复杂推理任务。

验证器是辅助模型,用于评估基础大语言模型(LLM)生成输出的正确性,在解决复杂推理问题时至关重要。传统验证器本身也是大型语言模型,常与基础模型规模相当甚至更大,导致计算开销高昂。本文提出轻量级验证方法 LiLaVe,能从基础 LLM 的隐藏状态中可靠提取正确性信号。其核心优势在于仅需传统方法极小的计算预算即可运行。为验证实用性,我们将其与 best-of-n、自一致性等主流元生成策略结合;同时设计了基于 LiLaVe 的新方法,如条件自修正和条件多数投票,在小模型上显著提升生成任务的准确率与效率。本工作证明了从 LLM 隐状态中挖掘信息的可行性,为资源高效、可扩展的推理应用开辟新路径。

原文摘要 · Abstract (English)

Verifiers are auxiliary models that assess the correctness of outputs generated by base large language models (LLMs). They play a crucial role in many strategies for solving reasoning-intensive problems with LLMs. Typically, verifiers are LLMs themselves, often as large (or larger) than the base model they support, making them computationally expensive. In this work, we introduce a novel lightweight verification approach, LiLaVe, which reliably extracts correctness signals from the hidden states of the base LLM. A key advantage of LiLaVe is its ability to operate with only a small fraction of the computational budget required by traditional LLM-based verifiers. To demonstrate its practicality, we couple LiLaVe with popular meta-generation strategies, like best-of-n or self-consistency. Moreover, we design novel LiLaVe-based approaches, like conditional self-correction or conditional majority voting, that significantly improve both accuracy and efficiency in generation tasks with smaller LLMs. Our work demonstrates the fruitfulness of extracting latent information from the hidden states of LLMs, and opens the door to scalable and resource-efficient solutions for reasoning-intensive applications.

轻量验证大模型推理隐状态利用

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