Semi-Supervised Generative Models for Multi-Agent Trajectories
Research output: Contributions to collected editions/works › Article in conference proceedings › Research › peer-review
Authors
Analyzing the spatiotemporal behavior of multiple agents is of great interest to many communities. Existing probabilistic models in this realm are formalized either in an unsupervised framework, where the latent space is described by discrete or continuous variables, or in a supervised framework, where weakly preserved labels add explicit information to continuous latent representations. To overcome inherent limitations, we propose a novel objective function for processing multi-agent trajectories based on semi-supervised variational autoencoders, where equivariance and interaction of agents are captured via customized graph networks. The resulting architecture disentangles discrete and continuous latent effects and provides a natural solution for injecting expensive domain knowledge into interactive sequential systems. Empirically, our model not only outperforms various state-of-the-art baselines in trajectory forecasting, but also learns to effectively leverage unsupervised multi-agent sequences for classification tasks on interactive real-world sports datasets.
Original language | English |
---|---|
Title of host publication | Advances in Neural Information Processing Systems 35 : 36th Conference on Neural Information Processing Systems (NeurIPS 2022) |
Editors | S. Koyejo, S. Mohamed, A. Agarwal, D. Belgrave, K. Cho, A. Oh |
Number of pages | 15 |
Volume | 48 |
Place of Publication | Red Hook |
Publisher | Curran Associates |
Publication date | 2022 |
Pages | 37267-37281 |
ISBN (print) | 978-1-7138-7108-8 |
ISBN (electronic) | 978-1-7138-7312-9 |
Publication status | Published - 2022 |
Event | 36th Conference on Neural Information Processing Systems - NeurIPS 2022 - New Orleans Convention Center, New Orleans, United States Duration: 28.11.2022 → 09.12.2022 Conference number: 36 https://nips.cc/Conferences/2022 |
Bibliographical note
Publisher Copyright:
© 2022 Neural information processing systems foundation. All rights reserved.
- Informatics
- Business informatics