arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.09359cs.LGcs.AIcs.NEcs.SC
收藏

基于学习型概念库的符号回归

Symbolic Regression with a Learned Concept Library

UT Austin(德克萨斯大学奥斯汀分校);Foundry Technologies(Foundry科技公司);MIT(麻省理工学院);University of Cambridge(剑桥大学)

Arya Grayeli, Atharva Sehgal, Omar Costilla-Reyes, Miles Cranmer, Swarat Chaudhuri

AI总结:

提出 LaSR 符号回归方法,通过 LLM 零样本查询构建并演化抽象概念库,结合进化搜索提升假设发现,在 Feynman 等基准上超过现有方法并发现新的 LLM 缩放定律。

摘要:

我们提出了一种用于符号回归(SR)的新方法,该任务旨在寻找能够最佳解释数据集的紧凑程序化假设。该问题通常通过遗传算法解决;我们表明,可以通过归纳一个抽象文本概念库来增强此类方法。我们的算法名为 LaSR,使用对大型语言模型(LLM)的零样本查询来发现并演化出现在已知高性能假设中的概念。我们结合标准演化步骤和以已发现概念为条件、由 LLM 引导的步骤(通过零样本 LLM 查询获得)来发现新假设。假设一旦被发现,就会用于新一轮的概念抽象和演化。我们在常用 SR 基准 Feynman 方程以及一组合成任务上验证了 LaSR。在这些基准上,LaSR 显著优于多种基于深度学习和演化算法的最先进 SR 方法。此外,我们表明 LaSR 可用于发现一条新颖且强大的 LLM 缩放定律。

英文摘要:

We present a novel method for symbolic regression (SR), the task of searching for compact programmatic hypotheses that best explain a dataset. The problem is commonly solved using genetic algorithms; we show that we can enhance such methods by inducing a library of abstract textual concepts. Our algorithm, called LaSR, uses zero-shot queries to a large language model (LLM) to discover and evolve concepts occurring in known high-performing hypotheses. We discover new hypotheses using a mix of standard evolutionary steps and LLM-guided steps (obtained through zero-shot LLM queries) conditioned on discovered concepts. Once discovered, hypotheses are used in a new round of concept abstraction and evolution. We validate LaSR on the Feynman equations, a popular SR benchmark, as well as a set of synthetic tasks. On these benchmarks, LaSR substantially outperforms a variety of state-of-the-art SR approaches based on deep learning and evolutionary algorithms. Moreover, we show that LaSR can be used to discover a novel and powerful scaling law for LLMs.

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

↑