用概率模型优化多模型链的置信度,提升可靠性。
Rational Tuning of LLM Cascades via Probabilistic Modeling
- 构建马尔可夫-耦合模型,预测多个LLM串联时的联合性能分布。
- 在3个以上模型的链路中,误差-成本曲线提升4.3%,小样本下达10.2%。
- 适合需要高效调参的LLM系统设计者,尤其关注低数据场景。
理解大语言模型(LLM)的可靠性已成为研究热点。由于LLM易产生幻觉且对提示设计敏感,预测单个模型性能已具挑战性;而对如级联结构这类复合系统而言,还需考虑各模型错误率之间的相互作用。本文提出一种用于序列型LLM联合性能分布的概率模型,支持通过连续优化理性调节级联中的置信度阈值。相比贝叶斯优化,所提出的参数化马尔可夫-耦合模型在 $k\≥ 3$ 的级联中平均提升误差-成本曲线下面积4.3%。在 $n \leq 30$ 的小样本情形下,性能优势扩大至10.2%,表明该框架关于错误率交互的归纳假设显著提升了样本效率。总体而言,该马尔可夫-耦合模型为调节LLM级联性能提供了合理依据,并揭示了概率方法在分析复杂LLM系统中的潜力。
原文摘要 · Abstract (English)
Understanding the reliability of large language models (LLMs) has recently garnered significant attention. Given LLMs' propensity to hallucinate, as well as their high sensitivity to prompt design, it is already challenging to predict the performance of an individual LLM. However, the problem becomes more complex for compound LLM systems such as cascades, where in addition to each model's standalone performance, we must understand how the error rates of different models interact. In this paper, we present a probabilistic model for the joint performance distribution of a sequence of LLMs, which enables a framework for rationally tuning the confidence thresholds of a LLM cascade using continuous optimization. Compared to selecting confidence thresholds using Bayesian optimization, our parametric Markov-copula model yields more favorable error-cost trade-offs, improving the area under the error-cost curve by 4.3% on average for cascades with $k\geq 3$ models. In the low-sample regime with $n \leq 30$ training examples, the performance improvement widens to 10.2%, suggesting that our framework's inductive assumptions about the interactions between the error rates of different LLMs enhance sample efficiency. Overall, our Markov-copula model provides a rational basis for tuning LLM cascade performance and points to the potential of probabilistic methods in analyzing systems of LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。