Search papers, labs, and topics across Lattice.
This paper addresses the challenge of estimating hidden capabilities of partners in ad-hoc teamwork (AHT) by introducing CE-CM, an approximate Bayesian method that infers task-invariant capability vectors through simulation-based sampling. The authors extend this approach with CE-CM-Div, which evaluates diverse planner rollouts to better account for human unpredictability, leading to improved capability estimates. Experimental results show that CE-CM effectively adapts to changing partner capabilities and significantly outperforms baseline methods in both simulated and human studies.
Rapidly inferring hidden partner capabilities can transform how autonomous agents collaborate in dynamic environments.
Effective collaboration with novel and diverse partners is a crucial skill for autonomous agents. Most current ad-hoc teamwork (AHT) approaches assume that agents will collaborate on a single, fixed task and that the partner's capabilities, their ability to successfully execute the desired action, are already known. In reality, a partner's true capabilities are often hidden, and human collaborators may act sub-optimally on tasks with multiple valid strategies. To address these limitations, we extend ad-hoc teamwork into a multi-task setting by re-framing it as a problem of joint planning with decentralised execution under hidden partner capabilities. We introduce CE-CM (Capability Estimation via Contextual Models), an approximate Bayesian method that infers task-invariant capability vectors. By using simulation-based sampling, the agent estimates capabilities and induces a contextual Multi-agent Markov Decision Processes for planning. This approach requires no population pre-training and refines its beliefs online from just a few tasks. To account for human unpredictability, we propose CE-CM-Div, an extension that evaluates capability hypotheses against diverse planner rollouts rather than a single optimal trajectory. Simulated experiments demonstrate that CE-CM rapidly recovers hidden capabilities, reduces infeasible action assignments, and adapts to changes over time. Furthermore, in an offline human study of 225 trajectories from 15 participants, CE-CM-Div substantially improved capability estimates over the baseline CE-CM method. Our results suggest capability-based modelling is a promising interpretable, task-agnostic representation in the studied settings, demonstrating that accounting for behavioural diversity is essential for robust human-AI teaming.