Papers
arxiv:2608.05224

Small Foundation Models of Human Cognition and Behaviour

Published on Aug 5
· Submitted by
Nick Oh
on Aug 10
Authors:

Abstract

Large language models fine-tuned on human behavioural data have emerged as general-purpose cognitive proxies, but the scale this requires, and whether these models process task structure or exploit statistical shortcuts, remain open questions. We train fourteen models from 135M to 14B parameters across four architecture families on Psych-101, a dataset of 10.7 million trial-level choices from 160 experiments. In-distribution, scale barely matters. The models fall within a narrow band, as though against a ceiling, and 0.6B to 1B parameters suffice to match a 70B baseline on held-out participants. Out-of-distribution, that band opens into a markedly steeper scaling gradient, with larger models clearly advantaged in generalisation to novel task structure. To determine what information these models use, we run two diagnostics. We progressively strip four prompt channels -- task instructions, experimental stimuli, outcome feedback, and choice history -- across 27 experiments, and permute trial order. Masking the content of stimuli and feedback destroys 75.7% of learned information and pushes models below chance, demonstrating that choice history alone does not account for performance. Permutation reveals invariance on tasks with independent trials but sensitivity where trial order is determined by prior responses. Small cognitively fine-tuned models therefore show promise as noise ceiling estimators for psychological experiments, though their scope remains bounded by the paradigms seen in training.

Community

Paper author Paper submitter

The paper fine-tunes small pretrained LLMs with LoRA to predict human choices on the Psych-101 benchmark, extending Centaur-style SFT (Binz et al., 2025) down to sub-billion-parameter models.

How many LoRA adapters are available?

·

There are approximately 117 adapters in total. We swept r = 4, 8, 16, 32, and 64 across 14 pre-trained, non-instruct base models from four model families: five Qwen3-Base models (0.6B, 1.7B, 4B, 8B, and 14B), three Llama-3 models (1B, 3B, and 8B), four SmolLM models (SmolLM2 at 135M, 360M, and 1.7B, and SmolLM3 at 3B), and two OLMo models (OLMo-2 at 1B and OLMo-3 at 7B).

But for Qwen3-1.7B, we swept only r = 8 and 16, while for Qwen3-14B, we swept r = 4, 16, and 64.

This is an automated message from the Librarian Bot. I found the following papers similar to this paper.

The following papers were recommended by the Semantic Scholar API

Please give a thumbs up to this comment if you found it helpful!

If you want recommendations for any Paper on Hugging Face checkout this Space

You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend

Sign up or log in to comment

Models citing this paper 117

Browse 117 models citing this paper

Datasets citing this paper 1

Spaces citing this paper 1

Collections including this paper 2