arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2310.03094cs.CLcs.AIcs.LG
收藏

大型语言模型级联与混合思维表示用于经济高效推理

Large Language Model Cascades with Mixture of Thoughts Representations for Cost-efficient Reasoning

George Mason University(乔治梅森大学);Microsoft(微软);Virginia Tech(弗吉尼亚理工大学)

Murong Yue, Jie Zhao, Min Zhang, Liang Du, Ziyu Yao

AI总结:

本文提出LLM级联框架,利用较弱模型的答案一致性判断问题难度,混合思维链与程序思维采样,在六个基准上以40%成本达到强模型性能。

摘要:

大型语言模型(LLM)如GPT-4在各种任务中表现出色,但这种强大的性能往往伴随着使用付费API服务的高昂费用。本文中,我们研究构建LLM级联以节省使用LLM的成本,特别是用于执行推理(如数学、因果)任务。我们的级联流程遵循这样的直觉:较简单的问题可以由较弱但更经济的LLM解决,而只有具有挑战性的问题才需要更强且更昂贵的LLM。为实现这一决策,我们将较弱LLM的“答案一致性”视为问题难度的信号,并提出了几种答案采样和一致性检查的方法,包括一种利用两种思维表示(即思维链和程序思维)混合的方法。通过在六个推理基准数据集上的实验,以GPT-3.5-turbo和GPT-4分别作为较弱和较强的LLM,我们证明了所提出的LLM级联可以实现与仅使用较强LLM相当的性能,但仅需其40%的成本。

英文摘要:

Large language models (LLMs) such as GPT-4 have exhibited remarkable performance in a variety of tasks, but this strong performance often comes with the high expense of using paid API services. In this paper, we are motivated to study building an LLM cascade to save the cost of using LLMs, particularly for performing reasoning (e.g., mathematical, causal) tasks. Our cascade pipeline follows the intuition that simpler questions can be addressed by a weaker but more affordable LLM, whereas only the challenging questions necessitate the stronger and more expensive LLM. To realize this decision-making, we consider the "answer consistency" of the weaker LLM as a signal of the question difficulty and propose several methods for the answer sampling and consistency checking, including one leveraging a mixture of two thought representations (i.e., Chain-of-Thought and Program-of-Thought). Through experiments on six reasoning benchmark datasets, with GPT-3.5-turbo and GPT-4 being the weaker and stronger LLMs, respectively, we demonstrate that our proposed LLM cascades can achieve performance comparable to using solely the stronger LLM but require only 40% of its cost.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑