用概念评分提升LLM程序搜索效率,解决抽象推理难题
ConceptSearch: Towards Efficient Program Search Using LLMs for Abstraction and Reasoning Corpus (ARC)
- 用概念理解替代像素比对,指导LLM生成更优程序
- 相比直接提示GPT-4,解题成功率显著提升
- 概念评分使搜索效率提高30%,适合做推理任务研究
抽象与推理语料库(ARC)对人工智能构成重大挑战,要求模型具备广泛的泛化能力和少样本学习能力,而当前深度学习方法(包括大语言模型)仍难以应对。尽管大语言模型在程序合成方面表现优异,但直接应用于ARC时效果有限。为此,我们提出ConceptSearch——一种新型函数搜索算法,利用大语言模型生成程序,并采用基于概念的评分机制高效引导搜索过程。不同于简单的像素级度量(如汉明距离),ConceptSearch评估程序捕捉输入输出示例中隐含转换概念的能力。我们对比了三种评分函数:汉明距离、基于CNN的评分函数和基于LLM的自然语言评分函数。实验表明,ConceptSearch显著优于直接使用GPT-4提示的方法。此外,新提出的概念评分机制在达到正确解所需的迭代次数上,相较汉明距离最高提升30%的效率。这些结果表明,将大语言模型驱动的程序搜索与概念引导相结合,可有效应对类似ARC的复杂泛化问题。
原文摘要 · Abstract (English)
The Abstraction and Reasoning Corpus (ARC) poses a significant challenge to artificial intelligence, demanding broad generalization and few-shot learning capabilities that remain elusive for current deep learning methods, including large language models (LLMs). While LLMs excel in program synthesis, their direct application to ARC yields limited success. To address this, we introduce ConceptSearch, a novel function-search algorithm that leverages LLMs for program generation and employs a concept-based scoring method to guide the search efficiently. Unlike simplistic pixel-based metrics like Hamming distance, ConceptSearch evaluates programs on their ability to capture the underlying transformation concept reflected in the input-output examples. We explore three scoring functions: Hamming distance, a CNN-based scoring function, and an LLM-based natural language scoring function. Experimental results demonstrate the effectiveness of ConceptSearch, achieving a significant performance improvement over direct prompting with GPT-4. Moreover, our novel concept-based scoring exhibits up to 30% greater efficiency compared to Hamming distance, measured in terms of the number of iterations required to reach the correct solution. These findings highlight the potential of LLM-driven program search when integrated with concept-based guidance for tackling challenging generalization problems like ARC.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。