Forecasting players in sports has grown in popularity due to the potential for a tactical advantage and the applicability of such research to multi-agent interaction systems. Team sports contain a significant social component that influences interactions between teammates and opponents. However, it still needs to be fully exploited. In this work, we hypothesize that each participant has a specific function in each action and that role-based interaction is critical for predicting players' future moves. We create RolFor, a novel end-to-end model for Role-based Forecasting. RolFor uses a new module we developed called Ordering Neural Networks (OrderNN) to permute the order of the players such that each player is assigned to a latent role. The latent role is then modeled with a RoleGCN. Thanks to its graph representation, it provides a fully learnable adjacency matrix that captures the relationships between roles and is subsequently used to forecast the players' future trajectories. Extensive experiments on a challenging NBA basketball dataset back up the importance of roles and justify our goal of modeling them using optimizable models. When an oracle provides roles, the proposed RolFor compares favorably to the current state-of-the-art (it ranks first in terms of ADE and second in terms of FDE errors). However, training the end-to-end RolFor incurs the issues of differentiability of permutation methods, which we experimentally review. Finally, this work restates differentiable ranking as a difficult open problem and its great potential in conjunction with graph-based interaction models. Project is available at: https://www.pinlab.org/aboutlatentroles
翻译:摘要:由于具有战术优势和适用于多智能体交互系统的潜在应用,体育比赛中球员的预测变得越来越受欢迎。团队运动包含影响队友和对手之间互动的重要社交因素,然而,这种因素尚未被充分利用。在这项工作中,我们假设每个参与者在每个行动中都有特定的功能,并且基于角色的互动对于预测球员未来的移动至关重要。我们创建了RolFor,一种新型的基于角色的预测模型。RolFor使用我们开发的一个新模块,被称为Ordering神经网络(OrderNN),对球员顺序进行排列,使每个球员分配到一个潜在的角色。然后使用RoleGCN对潜在角色进行建模。由于其图形表示形式,它提供了一个完全可学习的邻接矩阵,捕获角色之间的关系,并随后用于预测球员的未来轨迹。在具有挑战性的NBA篮球数据集上进行广泛实验,支持角色的重要性,并证明了我们使用可优化模型对其进行建模的目标。当预设角色时,所提出的RolFor在ADE误差方面比现有技术处于领先位置,而在FDE误差方面排名第二。然而,训练端对端RolFor引发了排列方法的可微分性问题,我们对其进行了实验检查。最后,本工作将不同iable ranking重新阐述为一个困难的开放问题及其与基于图形的交互模型的巨大潜力。该项目可在https://www.pinlab.org/aboutlatentroles中找到。