CogFunSearch

FunSearch adapted to write short, readable programs that model how animals choose.

Read the paper

The loop

  1. Proposes

    Python programs that predict an animal's next choice from its past choices and rewards

  2. Judged by

    Scoring each program on how well it predicts recorded rat choices

  3. Steers

    The best-scoring programs are fed back into the prompt, as in FunSearch

Headline result

In a bioRxiv preprint, on rat choices in a 2-armed drifting bandit task (data from Miller et al. 2021), CogFunSearch reliably found programs that beat the best existing cognitive model the authors compared against, both when improving supplied programs and when starting from scratch.

What this does not prove

A simple unconstrained neural network predicted significantly better than both the classic models and the discovered programs, so interpretability is gained at some cost in accuracy. Preprint, one task family.

How it generalizes the harness

The evolutionary loop finds a model of behavior, and the discovered program is a readable scientific artifact usable without the LLM.

Classification

Classification of CogFunSearch
What it proposesProgram
Generative modelPretrained LLM, frozen (inherited from FunSearch)
Algorithmic authorityHeld-out predictive score of each program on behavioral data
FeedbackScore
What gets updatedPrompt context and population of candidates
When it runsAt inference time
ValidityEmpirical (measured)
EvidenceExperiments reported
LineageApplication