Partner Modelling Emerges in Recurrent Agents (But Only When It Matters)

Tom Griffiths (Princeton University) · Ruaridh Mon-Williams (The University of Edinburgh) · Max Taylor-Davies (University of Edinburgh) · Elizabeth Mieczkowski (Princeton University) · Natalia Vélez (Princeton University) · Neil Bramley (University of Edinburgh) · Yanwei Wang (Computer Science and Artificial Intelligence Laboratory, Electrical Engineering & Computer Science) · Christopher G Lucas (University of Edinburgh)
behavioural analysiscollaborationcollaborative teamsdiverse partnersenvironmental conditionsgeneralisationinductive biasesinternal hidden statesmodel-free rnnopen-ended cooperative interactionpartner modellingprobing techniquesrapid adaptationsocial pressurestructured internal representationstask allocation

Humans are remarkably adept at collaboration, able to infer the strengths and weaknesses of new partners in order to work successfully towards shared goals. To build AI systems with this capability, we must first understand its building blocks: does such flexibility require explicit, dedicated mechanisms for modelling others—or can it emerge spontaneously from the pressures of open-ended cooperative interaction? To investigate this question, we train simple model-free RNN agents to collaborate with a population of diverse partners. Using the 'Overcooked-AI' environment, we collect data from thousands of collaborative teams, and analyse agents' internal hidden states. Despite a lack of additional architectural features, inductive biases, or auxiliary objectives, the agents nevertheless develop structured internal representations of their partners' task abilities, enabling rapid adaptation and generalisation to novel collaborators. We investigated these internal models through probing techniques, and large-scale behavioural analysis. Notably, we find that structured partner modelling emerges when agents can influence partner behaviour by controlling task allocation. Our results show that partner modelling can arise spontaneously in model-free agents—but only under environmental conditions that impose the right kind of social pressure.