atlas

Research map / Large language models / Agents, code and math

Symbolic Regression with a Learned Concept Library

NeurIPS 2024 · Large language models › Agents, code and math · approach: LLMs for optimization and heuristics · 9 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes LaSR, which augments PySR-style genetic symbolic regression with LLM-guided operators conditioned on a self-evolving library of natural-language concepts, outperforming prior SR methods on Feynman equations.

Problem

Symbolic regression searches an exponentially large discrete space of programs; purely symbolic genetic search is slow and lacks prior knowledge about which structures are plausible.

Approach

Induce a library of abstract natural-language concepts by prompting an LLM to summarize high-performing and low-performing hypotheses, evolve those concepts, and use them zero-shot to guide LLM-based mutation, crossover and initialization interleaved with standard genetic operators.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use LaSR instead of PySRSymbolic regression on Feynman Equations · Feynman Equationsexact match solve rate: 59/100 -> 72/100Helped
Increase Llama3-8B LLM-operation probability from p=1% to p=5%LaSR · Feynman Equationsexact solve: 67/100 -> 69/100Helped
Increase Llama3-8B LLM-operation probability from p=5% to p=10%LaSR · Feynman Equationsexact solve: 69/100 -> 71/100Helped
Use GPT-3.5 instead of Llama3-8B at p=1%LaSR · Feynman Equationsexact solve: 67/100 -> 72/100Helped
Remove concept evolution (ablation)LaSR · Feynman EquationsMSE solve rateHurt
Remove the concept library (ablation)LaSR · Feynman EquationsMSE solve rateHurt
Remove variable names (ablation)LaSR · Feynman EquationsMSE solve rateHurt
Add vague user-provided natural-language hints (ablation)LaSR · Feynman EquationsMSE solve rateHelped
Use LaSR instead of PySRSymbolic regression on synthetic equations · Synthetic symbolic regression datasettest set R^2: 0.070 -> 0.913Helped
Use the LaSR-discovered scaling-law skeleton instead of the Chinchilla skeletonBigBench multiple choice grade prediction · BigBench multiple choice gradeMSE loss: $0.03610 \pm 0.00268$ -> $0.03598 \pm 0.00265$No clear effect
Add number of shots to the training-steps exponent (ablation)Modified Chinchilla scaling law · BigBench multiple choice gradeMSE loss: $0.03610 \pm 0.00268$ -> $\mathbf{0.03598 \pm 0.00264}$No clear effect
Replace a scaling-law skeleton with a residual-only constant score (ablation)BigBench multiple choice grade prediction · BigBench multiple choice gradeMSE loss: $0.03598 \pm 0.00265$ -> $0.09324 \pm 0.01992$Hurt

Surprising findings

Limitations

Experimental setup

Datasets: Feynman Equations BIG-Bench

Benchmarks: Feynman Equations synthetic equations dataset BIG-Bench scaling law subset

Metrics: exact solve rate test set R^2 MSE loss

Baselines: GPlearn AFP AFP-FE DSR uDSR AI Feynman PySR

Builds on: PySR FunSearch AI Feynman

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy