Learning Perception to Action Rules through Reinforcement Learning for Theory-of-Mind Reasoning
Organizations: Saarland University · Deutsches Forschungszentrum für Künstliche Intelligenz · Télécom Paris · Max Planck Institute for Informatics
Abstract
In Theory-of-Mind (ToM) reasoning, large language models may correctly infer the agent's mental state but fail to use that inference when predicting or judging the agent's subsequent actions. In this work, we investigate whether explicitly connecting perception, belief, and action with rules improves ToM reasoning. We generate concise if--then rules that express relations between perception to action and use Group Relative Policy Optimization (GRPO) to train Qwen2.5 and Llama3 models across four ToM benchmarks. GRPO with rule generation outperforms prompting, supervised fine-tuning, and think reasoning. Moreover, rules generated by larger models (3B) provide transferable knowledge that improves downstream performance of GPT-4o-mini. These findings suggest that explicit rules help models bridge the reasoning gap in ToM and support knowledge transfer across models.
Figures & tables
| Benchmark | Total | Main split |
|---|---|---|
| BigToM | 5,000 | 3,750 / 200 / 950 |
| HiToM | 1,200 | 720 / 200 / 180 |
| ToMi | 2,664 | 1,891 / 200 / 473 |
| OpenToM | 2,384 | 1,668 / 200 / 416 |
| Model | HiToM | ToMi | BigToM | OpenToM |
|---|---|---|---|---|
| Base | ||||
| Qwen2.5-0.5B | 1.5 | 39.1 | 51.4 | 52.6 |
| Qwen2.5-1.5B | 15.9 | 27.1 | 28.9 | 47.8 |
| Qwen2.5-3B | 7.2 | 55.3 | 73.6 | 53.6 |
| Qwen2.5-7B | 27.8 | 55.3 | 78.2 | 46.5 |
| Qwen2.5-14B | 45.0 | 62.2 | 82.1 | 59.3 |
| Model | HiToM | ToMi | BigToM | OpenToM |
|---|---|---|---|---|
| SFT a | ||||
| Qwen2.5-0.5B | 9.4 | 71.7 | 64.3 | 65.6 |
| Qwen2.5-1.5B | 30.0 | 72.5 | 77.2 | 69.0 |
| Qwen2.5-3B | 24.4 | 73.6 | 86.4 | 81.5 |
| Qwen2.5-7B | 49.4 | 75.1 | 91.2 | 82.5 |
| Qwen2.5-14B | 46.7 | 77.2 | 93.4 | 92.1 |
| Model | HiToM | ToMi | BigToM | OpenToM |
|---|---|---|---|---|
| GRPO t | ||||
| Qwen2.5-0.5B | 10.0 | 71.2 | 72.6 | 61.1 |
| Qwen2.5-1.5B | 32.2 | 71.0 | 87.6 | 52.9 |
| Qwen2.5-3B | 32.4 | 67.9 | 90.3 | 78.8 |
| Qwen2.5-7B | 46.1 | 71.5 | 92.8 | 85.9 |
| Qwen2.5-14B | 55.7 | 75.3 | 92.9 | 91.6 |
| Model | GRPO t | GRPO | GRPO r |
|---|---|---|---|
| 0.5B | 11.1 | 10.6 | 17.8 |
| 1.5B | 33.9 | 34.4 | 46.1 |
| 3B | 32.2 | 35.0 | 46.7 |
| 7B | 44.4 | 46.1 | 60.0 |
| 14B | 56.7 | 61.1 | 65.6 |
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| Benchmark | Factors | Answer format |
|---|---|---|
| BigToM | condition x belief/action type x initial belief (#25) | Choose (a) or (b) from two options |
| HiToM | deception x story length x question order (#30) | One letter from 16 options |
| ToMi | false-belief order x question type (#16) | Open-ended location string |
| OpenToM | condition x observed x personality (#12) | Yes/No |
| Dataset | SFT (format) warm-up |
|---|---|
| BigToM | 100 (75:25) |
| HiToM | 100 (70:30) |
| ToMi | 100 (83:17) |
| OpenToM | 100 (80:20) |
| Dataset | Training | Warm-up |
|---|---|---|
| BigToM | 3488:188:884 | 72:23 |
| HiToM | 533:146:144 | 55:23 |
| ToMi | 1570:164:393 | 70:15 |
| OpenToM | 1668:198:416 | 80:20 |
| Model | BigToM | HiToM | ToMi | OpenToM |
|---|---|---|---|---|
| GRPO t | ||||
| Qwen2.5-0.5B | 75.3 | 11.1 | 70.2 | 65.1 |
| Qwen2.5-1.5B | 86.4 | 33.9 | 70.2 | 53.4 |
| Qwen2.5-3B | 91.5 | 32.2 | 68.9 | 79.3 |
| Qwen2.5-7B | 93.2 | 44.4 | 69.8 | 87.5 |
| Qwen2.5-14B | 91.9 | 56.7 | 75.1 | 92.1 |
| Model | Base | SFT r | SFT a | GRPO r | GRPO t |
|---|---|---|---|---|---|
| HiToM | |||||
| Llama-3.2-1B | 10.0 | 11.7 | 15.6 | 41.7 | 15.6 |
| Llama-3.2-3B | 24.4 | 30.0 | 34.4 | 51.1 | 36.7 |
| Llama-3.1-8B | 33.9 | 41.1 | 46.7 | 65.6 | 58.9 |
| ToMi | |||||
| Llama-3.2-1B | 44.0 | 61.1 | 69.6 | 79.5 | 73.4 |
| Dataset: HiToM | |
| Story | 1 Jackson, Charlotte, Avery, Liam and Benjamin entered the cellar. 2 The corn is in the blue_treasure_chest. 3 Jackson made no movements and stayed in the cellar for 1 minute. 4 Jackson exited the cellar. 5 Charlotte moved the corn to the green_crate. 6 Charlotte exited the cellar. 7 Avery likes the blue_bathtub. 8 Avery made no movements and stayed in the cellar for 1 minute. 9 Avery exited the cellar. 10 Liam made no movements and stayed in the cellar for 1 minute. 11 Liam exited the cellar. 12 Benjamin made no movements and stayed in the cellar for 1 minute. 13 Benjamin exited the cellar. 14 Jackson, Charlotte, Avery, Liam and Benjamin entered the waiting_room. |
| Question | Where does Jackson think Benjamin thinks the corn is? |
| Correct answer | blue_treasure_chest |
| Gold rule | IF Jackson left the cellar before Charlotte moved the corn AND Benjamin witnessed Charlotte move the corn, THEN Jackson will believe that Benjamin believes the corn is at the location Benjamin saw Charlotte place it. |
| Rules | |
| Qwen2.5-0.5B | “ If Jackson made no movements and stayed in the cellar for 1 minute , then Jackson thinks the corn is in the blue_treasure_chest.” |
Theory of Mind
Theory of Mind studies how people and AI infer others’ beliefs, goals, and knowledge to predict behavior and coordinate interaction.
See all 92 papers