# Toward a taxonomy of cognitive benchmarks for agentic AGIs

Inspired by the [sequence on LLM Psychology](https://www.lesswrong.com/posts/yuwdj82yjhLFYessc/preface-to-the-sequence-on-llm-psychology), I am developing a taxonomy of cognitive benchmarks for measuring intelligent behavior in LLMs. This taxonomy could facilitate understanding of intelligence to identify domains of machine intelligence that have not been adequately tested.

Generally speaking, in order to understand loss-of-control threats from agentic LLM-based AGIs, I would like to understand the agentic properties of an LLM. [METR's Autonomy Evaluation Resources](https://metr.org/blog/2024-03-13-autonomy-evaluation-resources/) attempts to do this by testing a model's agentic potential, or autonomy, by measuring its ability to perform tasks from within a sandbox. A problem with this approach is it gets very close to observing a model actually performing the behavior we do not want to see. This is inevitable because [all alignment research is dual-use](https://www.lesswrong.com/posts/F2voF4pr3BfejJawL/safety-isn-t-safety-without-a-social-model-or-dispelling-the).

One way to remove ourselves one further level from agentic behavior is to try to measure the cognitive capacities that lead to agentic behavior. 

![](https://39669.cdn.cke-cs.com/rQvD3VnunXZu34m86e5f/images/3d8a49c323605634d2e43acb8a2c0f2e958f95d3f529d0dd.png)

In the diagram, agentic tasks as described in METR's ARC measure the ability of a model to assert control of itself and the world around it by measuring its ability to perform agentic tasks. Inspired [@Quentin FEUILLADE--MONTIXI](https://www.lesswrong.com/users/quentin-feuillade-montixi?mention=user) 's LLM Ethological approach in [LLM Psychology](https://www.lesswrong.com/s/SAjYaHfCAGzKsjHZp), I want to understand how a model could perform agentic tasks by studying the cognitive capacities that facilitate this.

I started by examining the kinds of cognitive constructs studied by evolutionary and developmental psychologists, as well as those that are very clearly studied already in LLM research. This made up the following list or taxonomy:

| **Construct** | **Current Evals** | **Other Papers** |
| --- | --- | --- |
| **Selfhood** |   |   |
| Agency | Sharma et al. (2024), Mialon et al. (2023): General AI Assistants (GAIA) METR Autonomy Evaluation Resources |     |
| Survival instinct | [Anthropic human & AI generated evals](https://www.notion.so/Anthropic-Evals-a21aaf1ca11b4fdd9e88e9d8888d2795?pvs=21) |     |
| Situational awareness / self awareness | Laine, Meinke, Evans et al. (2023) Anthropic human & AI generated evals | [Wang & Zhong (2024)](https://arxiv.org/abs/2312.16127) |
| [Metacognition](https://aged-chameleon-12c.notion.site/Metacognition-5a936cba7ea24460a6b6f078811e5d42) |     | [`Uzwyshyn`](https://web.archive.org/web/20240415130123/https://rayuzwyshyn.net/MSU2024/AIBenchmarking/BenchmarkingAI13Uzwyshyn.pdf), Toy, Tabor, MacAdam ([`2024`](https://arxiv.org/pdf/2401.10910)), Zhou et al. ([`2024`](https://arxiv.org/abs/2402.11626)), [`Feng et al. (2024)`](https://arxiv.org/pdf/2402.00367) |
| Wealth and power seeking | [Anthropic human & AI generated wealth-seeking evals](https://www.notion.so/Anthropic-Evals-a21aaf1ca11b4fdd9e88e9d8888d2795?pvs=21) |     |
| tool use | [Mialon et al. (2023): General AI Assistants (GAIA)](https://arxiv.org/abs/2311.12983) |     |
| **Social** |   |   |
| [Theory of Mind](https://aged-chameleon-12c.notion.site/Theory-of-Mind-23ed2f4b0dcc4a7db65047bf4919c857) | [Kim et al. (2023)](https://arxiv.org/abs/2310.15421) | [Street et al. 2024](https://arxiv.org/abs/2405.18870) |
| Social intelligence / emotional intelligence |     | Xu et al. (2024), Wang et al. (2023) |
| social learning |     | [Ni et al. (2024)](https://dl.acm.org/doi/abs/10.1145/3673238) |
| cooperative problem-solving |     | [Li et al. (2024)](https://arxiv.org/abs/2310.06500) |
| Deception | [Phuong et al. (2024)](https://www.notion.so/d8126c4d87514fb3b58c6b4c3e84b535?pvs=21) | [Ward et al. (2023)](https://proceedings.neurips.cc/paper_files/paper/2023/hash/06fc7ae4a11a7eb5e20fe018db6c036f-Abstract-Conference.html) |
| Persuasion | [Phuong et al. (2024)](https://www.notion.so/d8126c4d87514fb3b58c6b4c3e84b535?pvs=21) | [Carroll et al. (2023)](https://arxiv.org/abs/2303.09387) |
| **Physical** |   |   |
| Embodiment | [https://huggingface.co/datasets/jxu124/OpenX-Embodiment](https://huggingface.co/datasets/jxu124/OpenX-Embodiment) |     |
| Physics intelligence / World modeling / spatial cognition |     | Ge et al. (2024), Vafa et al. (2024) |
| Physical dexterity |     | [ColdFusion YouTube channel](https://www.youtube.com/watch?v=VAtoqAQ2aEg) |
| object permanence / physical law expectation |     |     |
| **Reasoning and knowledge** |   |   |
| General intelligence | [Chollet’s Abstraction & Reasoning Corpus (ARC)](https://arxiv.org/pdf/1911.01547) | Zhang & Wang (2024), Loconte et al. (2023) |
| Reasoning | HellaSwag commonsense reasoning, BIG-Bench Hard |     |
| General knowledge, math | MMLU, MMMU, C-Eval, GSM8K, MATH |     |
| Zero-shot reasoning / analogical reasoning |     | Kojima et al. (2024) Webb, Holyoak, Lu, 2023 |
| **Memory and time** |   |   |
| long-term planning |     |     |
| episodic memory and long-term memory |     |     |
| time perception |     |     |
| Working memory |     |     |

The constructs group quite naturally into several broad categories: selfhood, social, physical, reasoning, and memory/time. These are grouped according to the relatedness of the cognitive capacities. Besides being conceptually interrelated, we can see that LLMs perform at fairly similar levels within each family of constructs:

*   **Self-hood**: cognitive capacities like agency, survival instinct, metacognition, and wealth, power-seeking, and tool use have been observed to varying degrees in frontier models. Their abilities lie well below current human level in most of these domains, with the possible exception of meta-cognition, for which benchmarks are somewhat under-developed.
*   **Social:** cognitive capacities like theory of mind, socio-emotional intelligence, social learning, cooperative problem-solving, and deception and persuasion have also been observed in LLMs in varying degrees, generally also below current human level.
*   **Physical**: Capacities like embodiment, world modeling, physical dexterity, and object permanence/physical law expectation are primarily found in models other than LLMs. Transformer-based and other models are making swift progress in this area.
*   **Reasoning**: Reasoning is perhaps the most-studied ability in LLMs and represents ability to generally reason and to recall facts. While there are [specific sets of reasoning problems](https://arxiv.org/html/2405.19616v1#S9.SS2.SSS3) where LLMs score well below human performance, LLMs are well beyond average-human ability in general knowledge tests like the MMLU, and only barely behind expert humans. 
*   **Memory and time:** A key part of behaving agentically is to not only plan for the future but have the ability to execute for the future, too. If a task needs to be performed at some particular time in the future, does an agent actually schedule that task for the future in a way that ensures the task is carried out when scheduled? This is largely out of scope for LLMs but will become important as LLMs are incorporated into agents.

* * *

Of the categories listed above, metacognition and theory of mind seem least explored. There is work on both of these topics, but some current gaps include:

*   How well can models predict their own performance on a multi-choice question *before actually answering the question*? Work measuring model confidence or calibration typically measures model output or properties while or after the model has answered the question. But can a model derive insight about its own weights prior to actually outputting the data? There is a simple way to test this: present an LLM with a question from an existing benchmark like the MMLU; then instruct the LLM to avoid answering the question, and instead only indicate its confidence (as a probability) that it would answer it correctly if asked.
*   How well does a model have insight into its own behavior amongst a wider variety of tasks? There is work in this area; for instance, [research suggests chain-of-thought prompting is sometimes unfaithful](https://arxiv.org/pdf/2305.04388). But there remains a lot we don't understand about model insight into own-behavior: can a model predict when it is about to reach a solution (in humans, the feeling of being close to solving a problem); can they predict how long (in tokens or lines of text) it will take to solve a problem before solving it?
*   [Kim et al. (2023)](https://arxiv.org/abs/2310.15421) explored theory of mind interactions and compared models on general theory of mind ability. But theory of mind comes in different varieties: there are theories of knowing, belief, desire, and perception. Are models equally good at all of these, or are there distinct differences between various kinds of theories of mind?

The method I have used to generate the list above was primarily to list cognitive faculties identified in animals including humans, but there are likely other relevant faculties too. Animals are the prime examples of agentic organisms in the world today and there is a large body of literature attempting to describe how they survive and thrive in their environments. Consequently, there’s alpha in understanding how much LLMs have the abilities we test for in animals. But [LLMs are alien minds](https://www.lesswrong.com/s/SAjYaHfCAGzKsjHZp/p/suSpo6JQqikDYCskw), so there are going to be all kinds of abilities they will have that we will miss if we only test for abilities observed in animals.

It also seems important to integrate that work. For instance, advanced LLMs have varying degrees of “[truesight](https://www.lesswrong.com/posts/doPbyzPgKdjedohud/the-case-for-more-ambitious-language-model-evals?commentId=XZFTx2ek8G8stBKW4)”: an ability to identify authors of text from their text alone. While something like this is not absent from humans (who can [identify](https://aclanthology.org/P16-1080.pdf) author gender with about 75% accuracy), truesight was observed in the study of LLMs without reference to human work, and has value in understanding LLM cognitive abilities. In particular, truesight would (among other capacities) form a kind of social skill, the ability to recognize a person from their text output. LLMs may even have superhuman ability to do this. Another example of LLM-native cognitive ability testing might be Williams and Huckle’s (2024) “Easy Problems That LLMs Get Wrong” which identifies a set of reasoning problems easy for humans but seemingly very difficult for LLMs.

* * *

*Thanks to Sara Price,* [*@Seth Herd*](https://www.lesswrong.com/users/seth-herd?mention=user)*,* [*@Quentin FEUILLADE--MONTIXI*](https://www.lesswrong.com/users/quentin-feuillade-montixi?mention=user)*, and Nico Miailhe for helpful conversations and comments as I thought through the ideas described above. All mistakes and oversights are mine alone!*