让小模型提前放弃回答难题,省钱又降错。
Cost-Saving LLM Cascades with Early Abstention
- 小模型也能主动不答难题,提前拦截复杂问题。
- 平均成本降13%,错误率降5%,仅多放弃4.1%的问题。
- 适合金融医疗等容错率低的高风险场景。
LLM级联通过小型语言模型处理大部分查询,仅在困难问题上启用昂贵的大模型,显著降低成本且不影响性能。但在金融、医疗等风险敏感领域,避免模型出错尤为重要。由于大模型仍会犯错,系统应允许对难题完全拒答。这引发设计疑问:拒答是否仅限于最后一步,还是也应在早期模型中引入?由于小模型与大模型的错误模式相关,让早期模型提前拒答可预判后续昂贵模型的拒答行为,从而避免运行它们,降低推理成本和延迟。我们研究了这种‘早期拒答’在级联中的效果,在六个基准测试(GSM8K、MedMCQA、MMLU、TriviaQA、TruthfulQA 和 XSum)上平均测试损失降低2.2%。这得益于更有效的拒答策略:整体拒答率上升4.1%,但成本减少13.0%,错误率下降5.0%。结果表明,利用不同模型间错误模式的相关性,可提升带拒答机制的LLM系统的性能。
原文摘要 · Abstract (English)
LLM cascades deploy small LLMs to answer most queries, limiting the use of large and expensive LLMs to difficult queries. This approach can significantly reduce costs without impacting performance. However, risk-sensitive domains such as finance or medicine place an additional premium on avoiding model errors. Since even the most expensive models are susceptible to making mistakes, applications in these domains benefit from allowing LLM systems to completely abstain from answering difficult queries. Introducing abstention poses a design question for LLM cascades: should abstention only be allowed at the final model or also at earlier models? Since the error patterns of small and large models are correlated, allowing earlier models to abstain may reduce inference costs and latency by anticipating abstention decisions by expensive and slow models, thus avoiding the need to run these models. We investigate the benefits of such "early abstention" in LLM cascades and find that it reduces overall test loss by 2.2% on average across six benchmarks (GSM8K, MedMCQA, MMLU, TriviaQA, TruthfulQA, and XSum). These gains result from a more effective use of abstention, trading a 4.1% average increase in the overall abstention rate for a 13.0% reduction in cost and a 5.0% reduction in error rate. Our findings demonstrate the possibility of leveraging correlations between the error patterns of different language models to drive performance improvements for LLM systems with abstention.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。