Instruct: Identify the main and secondary category of Arxiv papers based on the titles and abstracts
Query: THOMAS: Trajectory Heatmap Output with learned Multi-Agent Sampling   In this paper, we propose THOMAS, a joint multi-agent trajectory prediction
framework allowing for an efficient and consistent prediction of multi-agent
multi-modal trajectories. We present a unified model architecture for
simultaneous agent future heatmap estimation, in which we leverage hierarchical
and sparse image generation for fast and memory-efficient inference. We propose
a learnable trajectory recombination model that takes as input a set of
predicted trajectories for each agent and outputs its consistent reordered
recombination. This recombination module is able to realign the initially
independent modalities so that they do no collide and are coherent with each
other. We report our results on the Interaction multi-agent prediction
challenge and rank $1^{st}$ on the online test leaderboard.

22
The categories of this paper are: chao-dyn
23
The categories of this paper are: solv-int
4
The categories of this paper are: cond-mat
21
The categories of this paper are: adap-org
34
The categories of this paper are: chem-ph
16
The categories of this paper are: stat
15
The categories of this paper are: nucl-ex
11
The categories of this paper are: hep-lat
9
The categories of this paper are: nlin
14
The categories of this paper are: math-ph
30
The categories of this paper are: q-alg
17
The categories of this paper are: q-fin
7
The categories of this paper are: hep-th
33
The categories of this paper are: plasm-ph
31
The categories of this paper are: ao-sci
10
The categories of this paper are: q-bio