- Main
Automated Discovery of Psychological Representations using Language Model Agents
- Lima Braga, Daniel;
- Marjieh, Raja;
- Jagadish, Akshay Kumar;
- Sucholutsky, Ilia;
- Jacoby, Nori;
- Lake, Brenden;
- Griffiths, Tom
Abstract
Similarity judgments provide a powerful behavioral signal for characterizing psychological representations. Computational techniques for translating similarity judgments into representations have been available since the early days of cognitive science. However, these algorithms typically focus on one particular kind of representation (e.g., spaces or trees), and identifying a satisfying representation often requires searching over a large space of possible structures and algorithms. Inspired by recent developments in AI, we propose an automated process for efficient representation discovery (AutoRep), in which large language model agents iteratively propose, refine, and critique code for fitting representational structures to similarity data. We show how this process can reliably recover psychological representations from both synthetic and human datasets, and how it can flexibly explore spatial, feature-based, graphical, or even neural representations. Our work demonstrates how modern agentic pipelines can substantially facilitate basic research workflows in cognitive science.