atlas

Research map / Large language models / Instruction and fine-tuning

CogBench: a large language model walks into a psychology lab

ICML 2024 · Large language models › Instruction and fine-tuning · approach: Fine-tuning for reasoning data · 5 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.

Introduces CogBench, a text-based benchmark of seven cognitive psychology experiments yielding ten behavioral metrics, and uses it to phenotype 35 LLMs with multilevel regression and prompt-engineering analyses.

Problem

LLM benchmarks focus almost exclusively on performance scores and overlook underlying behavioral mechanisms, so there is no holistic behavioral evaluation of LLMs.

Approach

Adapt canonical cognitive psychology experiments (probabilistic reasoning, horizon task, restless bandit, instrumental learning, two-step task, temporal discounting, BART) into text prompts and derive behavioral metrics (e.g., model-basedness, meta-cognition, directed/random exploration, optimism bias) alongside performance, then relate them to LLM features with multilevel regression.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add chain-of-thought promptingFive evaluated LLMs · Probabilistic reasoningposterior accuracy: $9.01\%$ average increaseHelped
Add take-a-step-back promptingFive evaluated LLMs · Probabilistic reasoningposterior accuracy: $3.10\%$ increaseHelped
Add chain-of-thought promptingFive evaluated LLMs · Two-step taskmodel-basedness: $64.59\%$ increaseHelped
Add take-a-step-back promptingFive evaluated LLMs · Two-step taskmodel-basedness: $118.59\%$ increaseHelped
Use RLHFLLMs · CogBench behavioral metricsL2-Norm distance from humans: $11.7\%$ average decreaseHelped
Increase number of parametersLLMs · Performance across all seven tasksstandardized performance: $\beta=0.277\pm0.39,z=14.1,p<0.001$Helped
Increase number of parametersLLMs · Two-step taskmodel-basedness: $\beta=0.481\pm0.22,z=4.2,p<0.001$Helped
Use RLHFLLMs · Restless bandit taskmeta-cognition: $\beta=0.461\pm0.15,z=5.9,p<0.001$Helped
Fine-tune on codeLLMs · Performance across all seven tasksstandardized performanceNo clear effect
Fine-tune on codeLLMs · Two-step taskmodel-basednessNo clear effect
Use an open-source modelLLMs · Balloon Analog Risk Taskrisk-taking: $\beta=-0.612 \pm 0.11,z=-11.4,p<0.001$Hurt

Surprising findings

Limitations

Experimental setup

Datasets: CogBench procedurally generated experiment prompts human behavioral data from prior cognitive psychology studies

Benchmarks: CogBench

Metrics: posterior accuracy average delivered dollars model-basedness meta-cognition (adjusted QSR) learning rate optimism bias directed exploration random exploration temporal discounting BART risk-taking

Baselines: text-davinci-003 text-davinci-002 Claude-1 Claude-2 text-bison@002 Falcon-40b MPT-30b LLaMA-2-70

Builds on: chain-of-thought prompting take-a-step-back prompting Rescorla-Wagner model UMAP

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