PADiff: Predictive and Adaptive Diffusion Policies for Ad Hoc Teamwork

Hohei Chan; Xinzhi Zhang; Antao Xiang; Weinan Zhang; Mengchen Zhao

AAAI 2026technical0 citations

PADiff: Predictive and Adaptive Diffusion Policies for Ad Hoc Teamwork

Hohei Chan, Xinzhi Zhang, Antao Xiang, Weinan Zhang, Mengchen Zhao

Abstract

Ad hoc teamwork (AHT) requires agents to collaborate with previously unseen teammates, which is crucial for many real-world applications. The core challenge of AHT is to develop an ego agent that can predict and adapt to unknown teammates on the fly. Conventional RL-based approaches optimize a single expected return, which often causes policies to collapse into a single dominant behavior, thus failing to capture the multimodal cooperation patterns inherent in AHT. In this work, we introduce PADiff, a diffusion-based approach that captures agent

BibTeX

@inproceedings{aaai2026_padiffpredictive,
  title = {PADiff: Predictive and Adaptive Diffusion Policies for Ad Hoc Teamwork},
  author = {Hohei Chan and Xinzhi Zhang and Antao Xiang and Weinan Zhang and Mengchen Zhao},
  booktitle = {AAAI 2026},
  year = {2026}
}