通过重排输入顺序,让大模型在对称任务中更少遗漏信息,准确率最高提升99%。
Rank It, Then Ask It: Input Reranking for Maximizing the Performance of LLMs on Symmetric Tasks
- 利用辅助模型评估元素重要性和位置影响,动态重排输入顺序
- 在真实与合成数据上,准确率最高接近理论最优上限的99%
- 适合需要处理无序数据集合的问答、聚合查询等任务
大语言模型(LLMs)已成为跨领域实用工具。本文研究其在对称任务中的应用,即针对无序元素集合(如数据库表)提出聚合查询。当元素数量多时,LLM易忽略部分元素,导致回答不准确。由于输入为有序序列而实际问题无序,我们提出输入重排策略:通过优化输入顺序,提升模型表现,且不依赖具体查询形式。关键在于估计每个元素的相关性与每个位置的重要性,我们设计算法并借助辅助LLM高效实现。在多个合成与真实数据集上实验验证,该方法使模型准确率最高提升至理论最优上限的99%。
原文摘要 · Abstract (English)
Large language models (LLMs) have quickly emerged as practical and versatile tools that provide new solutions for a wide range of domains. In this paper, we consider the application of LLMs on symmetric tasks where a query is asked on an (unordered) bag of elements. Examples of such tasks include answering aggregate queries on a database table. In general, when the bag contains a large number of elements, LLMs tend to overlook some elements, leading to challenges in generating accurate responses to the query. LLMs receive their inputs as ordered sequences. However, in this problem, we leverage the fact that the symmetric input is not ordered, and reordering should not affect the LLM's response. Observing that LLMs are less likely to miss elements at certain positions of the input, we introduce the problem of LLM input reranking: to find a ranking of the input that maximizes the LLM's accuracy for the given query without making explicit assumptions about the query. Finding the optimal ranking requires identifying (i) the relevance of each input element for answering the query and (ii) the importance of each rank position for the LLM's attention. We develop algorithms for estimating these values efficiently utilizing a helper LLM. We conduct comprehensive experiments on different synthetic and real datasets to validate our proposal and to evaluate the effectiveness of our proposed algorithms. Our experiments confirm that our reranking approach improves the accuracy of the LLMs on symmetric tasks by up to $99\%$ proximity to the optimum upper bound.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。