baller2vec:用于多主体时空建模的多实体变压器

baller2vec 这是论文的存储库： 和。 。 arXiv。 2021。 左：每个时间步t上baller2vec的输入是一组无序的特征向量，其中包含有关NBA球员在场上的身份和位置的信息。 右： baller2vec通过使用新型的自注意蒙版张量将标准的Transformer推广到多实体设置。 然后将遮罩重新整形为矩阵，以与典型的Transformer实现兼容。 通过专门学习预测球的轨迹， baller2vec能够推断出特质球员的属性。 此外， baller2vec嵌入空间中的最近邻居是合理的doppelgängers。 图片提供者： ， ， ， ， 和 。 此外， baller2vec几个注意头似乎执行与篮球相关的不同功能，例如预期传球。 生成GIF的代码改编自。 在这里，受过训练的baller2vec模型可以同时预测场上所有球员的轨迹，同时使用历史和当前背景来预