arXiv:2507.23167cs.CLcs.AI2025-07

用模型内部状态学信心,让多个大模型回答更靠谱

LENS: Learning Ensemble Confidence from Neural States for Multi-LLM Answer Integration

  • 通过分析模型隐藏层状态预测信心值
  • 多任务测试中显著优于传统集成方法
  • 轻量无修改,适合实际部署场景

大型语言模型在各类任务中表现优异,各模型在不同领域各有专长。有效融合多个模型的预测结果对提升系统鲁棒性和性能至关重要。然而,现有集成方法多采用投票或逻辑值拼接等简单策略,忽略了模型在不同语境下的信心差异。本文提出LENS(从神经状态学习集成信心),通过分析各模型的内部表示来学习信心估计。为每个LLM训练一个轻量级线性信心预测器,输入为逐层隐藏状态和归一化概率,实现基于上下文可靠性的动态加权。该方法无需修改模型参数,计算开销可忽略。在多项选择题和布尔型问答任务上的实验表明,LENS显著优于传统集成方法。研究结果表明,内部表示蕴含有价值的信心信号,可有效用于集成学习。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have demonstrated impressive performance across various tasks, with different models excelling in distinct domains and specific abilities. Effectively combining the predictions of multiple LLMs is crucial for enhancing system robustness and performance. However, existing ensemble methods often rely on simple techniques like voting or logits ensembling, which overlook the varying confidence and reliability of models in different contexts. In this work, we propose LENS (Learning ENsemble confidence from Neural States), a novel approach that learns to estimate model confidence by analyzing internal representations. For each LLM, we train a lightweight linear confidence predictor that leverages layer-wise hidden states and normalized probabilities as inputs. This allows for more nuanced weighting of model predictions based on their context-dependent reliability. Our method does not require modifying the model parameters and requires negligible additional computation. Experimental results on multiple-choice and boolean question-answering tasks demonstrate that LENS outperforms traditional ensemble methods by a substantial margin. Our findings suggest that internal representations provide valuable signals for determining model confidence and can be effectively leveraged for ensemble learning.

大模型集成信心估计轻量推理

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