atlas

Research map / Large language models / Language, safety and interpretability

Turning large language models into cognitive models

ICLR 2024 · Large language models › Language, safety and interpretability · approach: Reasoning mechanisms and failure modes · 21 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.

Finetunes a linear readout on LLaMA-65B embeddings of text-described psychology experiments (CENTaUR), showing it outperforms domain-specific cognitive models in decision-making and generalizes to a held-out task.

Problem

Large language models often show unhuman-like behavior, so the paper asks whether they can be turned into accurate models of human cognition and decision-making.

Approach

Extract hidden embeddings from a frozen LLaMA-65B on text prompts describing psychological choice tasks, then fit a regularized logistic regression (linear layer) on top to predict human choices, yielding CENTaUR; extend with random effects for individual differences and joint multi-task training for hold-out generalization.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Fine-tune a linear layer on LLaMA embeddingsLLaMA 65B · choices13knegative log-likelihood: 96248.5 -> 48002.3Helped
Use CENTaUR instead of BEASTchoices13k · choices13knegative log-likelihood: 49448.1 -> 48002.3Helped
Fine-tune a linear layer on LLaMA embeddingsLLaMA 65B · horizon tasknegative log-likelihood: 46211.4 -> 25968.6Helped
Use CENTaUR instead of the hybrid modelhorizon task · horizon tasknegative log-likelihood: 29042.5 -> 25968.6Helped
Fine-tune a linear layer on LLaMA embeddingsLLaMA 65B · choices13kregret: M = 1.85, SE =0.01 -> 1.35 (SE =0.01)Helped
Fine-tune a linear layer on LLaMA embeddingsLLaMA 65B · horizon taskregret: M = 7.21, SE =0.02 -> M = 2.38, SE =0.01Helped
Add random effects for each participant and embedding dimension (ablation)CENTaUR · horizon task, participant-levelnegative log-likelihood: 25968.6 -> 23929.5Helped
Fine-tune on choices13k and horizon taskLLaMA 65B · experiential-symbolic task (hold-out)negative log-likelihood: 6307.9 -> 4521.1Helped
Fine-tune on choices13k and horizon taskLLaMA 65B · experiential-symbolic task (hold-out)negative log-likelihood: 5977.7 -> 4521.1Helped
Fine-tune on psychological decision-task dataLLaMA 65B · horizon task, equal-information and unequal-information conditionshuman-like exploratory choice patternsHelped
Fine-tune on choices13k and horizon taskLLaMA 65B · experiential-symbolic task (hold-out)description-option preferenceHelped

Surprising findings

Limitations

Experimental setup

Datasets: choices13k horizon task experiential-symbolic task

Benchmarks: choices13k horizon task experiential-symbolic task

Metrics: negative log-likelihood regret choice curves indifference points

Baselines: LLaMA-65B without finetuning BEAST hybrid model (horizon task) random guessing

Builds on: LLaMA BEAST PyTorch

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