arXiv:2504.19391cs.LG2025-04被引 2

双向信心评估提升大模型推理效率,减少低效调用。

Bi-directional Model Cascading with Proxy Confidence

  • 用代理模型和隐藏状态分析同时评估大小模型信心
  • 在多选数据集上降低对昂贵大模型的调用比例
  • 适合追求高效推理的LLM部署场景

模型级联最近被成功应用于大语言模型,通过按规模从小到大顺序使用模型,仅在小模型置信度不足时才调用大型昂贵模型,从而提升推理效率。现有方法因无法访问大模型而仅依赖小模型的有限置信度估计,但大模型置信度其实很重要。为此,我们提出一种双向级联方法,通过代理模型同时考虑小模型与大模型的置信度。为此,我们利用隐藏状态分析增强小模型的后调用置信度表示,进而实现更优的校准;同时引入一个极小的代理模型,估算大模型的前调用置信度。在多个挑战性多选数据集上的实验表明,该方法相比标准级联基线显著减少了对高成本模型的调用次数。

原文摘要 · Abstract (English)

Model Cascading, recently applied successfully to LLMs, is a simple but powerful technique that improves the efficiency of inference by selectively applying models of varying sizes. Models are used in sequence from smallest to largest, only deferring samples to large, costly models when smaller models are not sufficiently confident. Existing approaches to deferral use only limited small model confidence estimates because of the inaccessibility of the large model, although large model confidence is known to be important. We therefore propose a bi-directional approach to deferral that considers the confidence of small and large models in the cascade simultaneously through the use of a proxy for the large model. This requires a richer representation of model confidence to enable comparative calibration: we use an analysis of hidden states to improve post-invocation confidence of the small model, which in itself improves cascading results over prior approaches. We then combine this with a tiny proxy model to estimate pre-invocation confidence of the large model. We examine the proposed cascading system over challenging, multiple-choice datasets, finding improvements over standard cascading baselines reflected in reductions in deferrals to more costly models.

模型级联推理优化置信度评估

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