arXiv cs.AI· Caroline Wang, Rolando Fernandez, Zelal Su Mustafaoglu, Montek Singh Kundan, Johnny Liu, Aditya Madhan, Zhihan Wang, Lingyun Xiao, Di Yang Shi, Jiaxun Cui, Arrasy Rahman, Peter Stone·· 3 天前AI 评分23
JaxAHT：基于 JAX 的 Ad Hoc Teamwork 开源库
JaxAHT: A JAX-Based Library for Ad Hoc Teamwork
AI 导读
研究者发布 JaxAHT，首个基于 JAX 的开源 Ad Hoc Teamwork（AHT）库，统一支持队友生成、ego agent 训练与 unseen teammate 评估，相比现有 PyTorch 实现取得约 95x 墙钟加速。
正文 · AI 翻译
作者：Caroline Wang、Rolando Fernandez、Zelal Su Mustafaoglu、Montek Singh Kundan、Johnny Liu、Aditya Madhan、Zhihan Wang、Lingyun Xiao、Di Yang Shi、Jiaxun Cui、Arrasy Rahman、Peter Stone
摘要：随着自主智能体日益广泛部署，它们必须能够与先前未见过的智能体（例如人类）进行协调。这一挑战已被形式化为 Ad Hoc Teamwork（AHT），并吸引了大量研究。然而，AHT 研究生命周期高昂的计算成本、缺乏标准化的基准实现，以及缺少多样化且经过验证的评估队友套件，都阻碍了进展。本工作提出了 JaxAHT，这是首个基于 JAX 的开源库，旨在加速并标准化 AHT 研究生命周期。借助 JAX 的硬件加速和大规模并行化能力，JaxAHT 为队友生成、自我智能体训练以及针对未见队友的评估提供了统一框架，相比现有的 PyTorch 对应实现实现了约 95 倍的挂钟时间加速。除该库之外，我们还贡献了一套多样化的评估队友，涵盖 Level-Based Foraging、Overcooked 和 Hanabi。使用 JaxAHT，我们对 AHT 算法进行了大规模、计算量受控的基准研究，发现没有任何算法始终表现最佳，智能体建模主要在基于角色的场景中带来收益，并且最佳响应回报被从零开始的强化学习低估了。
| 主题： | 人工智能（cs.AI） |
| ACM 分类： | I.2; I.2.11; I.2.1; H.5.3; D.2.13 |
| 引用为： | arXiv:2609.13716 [cs.AI] |
| （或此版本使用 arXiv:2609.13716v2 [cs.AI]） | |
| https://doi.org/10.48550/arXiv.2609.13716 由 DataCite 颁发的 arXiv DOI |
提交历史
来自：Caroline Wang [查看电子邮件]
[v1]
2026年9月12日星期六 05:01:27 UTC（5,012 KB）
[v2]
2026年10月11日星期日 18:20:23 UTC（4,595 KB）
来源：arXiv cs.AI · arxiv.org