ReCollab: Retrieval-Augmented LLMs for Cooperative Ad-hoc Teammate Modeling

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wallace, Conor, Siddique, Umer, Cao, Yongcan
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917171490193408
author Wallace, Conor
Siddique, Umer
Cao, Yongcan
author_facet Wallace, Conor
Siddique, Umer
Cao, Yongcan
contents Ad-hoc teamwork (AHT) requires agents to infer the behavior of previously unseen teammates and adapt their policy accordingly. Conventional approaches often rely on fixed probabilistic models or classifiers, which can be brittle under partial observability and limited interaction. Large language models (LLMs) offer a flexible alternative: by mapping short behavioral traces into high-level hypotheses, they can serve as world models over teammate behavior. We introduce \Collab, a language-based framework that classifies partner types using a behavior rubric derived from trajectory features, and extend it to \ReCollab, which incorporates retrieval-augmented generation (RAG) to stabilize inference with exemplar trajectories. In the cooperative Overcooked environment, \Collab effectively distinguishes teammate types, while \ReCollab consistently improves adaptation across layouts, achieving Pareto-optimal trade-offs between classification accuracy and episodic return. These findings demonstrate the potential of LLMs as behavioral world models for AHT and highlight the importance of retrieval grounding in challenging coordination settings.
format Preprint
id arxiv_https___arxiv_org_abs_2512_22129
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle ReCollab: Retrieval-Augmented LLMs for Cooperative Ad-hoc Teammate Modeling
Wallace, Conor
Siddique, Umer
Cao, Yongcan
Multiagent Systems
Artificial Intelligence
Machine Learning
Ad-hoc teamwork (AHT) requires agents to infer the behavior of previously unseen teammates and adapt their policy accordingly. Conventional approaches often rely on fixed probabilistic models or classifiers, which can be brittle under partial observability and limited interaction. Large language models (LLMs) offer a flexible alternative: by mapping short behavioral traces into high-level hypotheses, they can serve as world models over teammate behavior. We introduce \Collab, a language-based framework that classifies partner types using a behavior rubric derived from trajectory features, and extend it to \ReCollab, which incorporates retrieval-augmented generation (RAG) to stabilize inference with exemplar trajectories. In the cooperative Overcooked environment, \Collab effectively distinguishes teammate types, while \ReCollab consistently improves adaptation across layouts, achieving Pareto-optimal trade-offs between classification accuracy and episodic return. These findings demonstrate the potential of LLMs as behavioral world models for AHT and highlight the importance of retrieval grounding in challenging coordination settings.
title ReCollab: Retrieval-Augmented LLMs for Cooperative Ad-hoc Teammate Modeling
topic Multiagent Systems
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2512.22129