摘要
arXiv:2408.10878v3 通告类型:替换
摘要:来自团队体育等领域的多智能体轨迹数据经常由于各种因素而缺失。尽管已经提出了许多用于时空数据的插值方法,但在球员移动高度动态且智能体间交互不断变化的多智能体体育场景中,这些方法并不适用。为了解决这些挑战,我们提出了一种名为MIDAS(Multi-agent Imputer with Derivative-Accumulating Self-ensemble)的框架,该框架能够以高准确性和物理合理性插值多智能体轨迹。它通过基于Set Transformer的神经网络同时预测位置、速度和加速度,并通过递归积累预测的速度和加速度值生成替代估计。然后,使用可学习的加权集成将这些预测结合,生成最终的插值轨迹。在三个体育数据集上的实验表明,MIDAS在位置准确性和物理合理性方面显著优于现有基线。最后,我们展示了MIDAS的应用案例,例如估算总体距离和传球成功率,以突出其在需要完整跟踪数据的实际下游任务中的适用性。