arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2309.00331cs.CVcs.RO

基于注意力机制的长短期记忆网络的人类轨迹预测

Human trajectory prediction using LSTM with Attention mechanism

Isfahan University of Technology(伊斯法罕理工大学)

Amin Manafi Soltan Ahmadi, Samaneh Hoseini Semnani

AI总结:

提出一种结合LSTM与注意力机制的人类轨迹预测模型,通过注意力分数聚焦关键输入特征,在ETH和UCY数据集上相比Social LSTM在ADE和FDE上分别提升6.2%和6.3%。

摘要:

在本文中,我们提出了一种结合长短期记忆(LSTM)网络与注意力机制的人类轨迹预测模型。为此,我们使用注意力分数来确定模型在做出预测时应关注输入数据的哪些部分。注意力分数是针对每个输入特征计算的,分数越高表示该特征在预测输出中的重要性越大。最初,这些分数是针对目标人的位置、速度及其邻近个体的位置和速度确定的。通过使用注意力分数,我们的模型可以优先处理输入数据中最相关的信息,并做出更准确的预测。我们从注意力机制中提取注意力分数,并将其整合到轨迹预测模块中,以预测人类的未来轨迹。为了实现这一点,我们引入了一个新的神经层,该层在提取注意力分数后对其进行处理,并将其与位置信息拼接起来。我们在公开可用的ETH和UCY数据集上评估了我们的方法,并使用最终位移误差(FDE)和平均位移误差(ADE)指标来衡量其性能。我们表明,我们改进的算法在预测拥挤空间中行人的未来轨迹方面优于Social LSTM。具体来说,与文献中的Social LSTM结果相比,我们的模型在ADE上提高了6.2%,在FDE上提高了6.3%。

英文摘要:

In this paper, we propose a human trajectory prediction model that combines a Long Short-Term Memory (LSTM) network with an attention mechanism. To do that, we use attention scores to determine which parts of the input data the model should focus on when making predictions. Attention scores are calculated for each input feature, with a higher score indicating the greater significance of that feature in predicting the output. Initially, these scores are determined for the target human position, velocity, and their neighboring individual's positions and velocities. By using attention scores, our model can prioritize the most relevant information in the input data and make more accurate predictions. We extract attention scores from our attention mechanism and integrate them into the trajectory prediction module to predict human future trajectories. To achieve this, we introduce a new neural layer that processes attention scores after extracting them and concatenates them with positional information. We evaluate our approach on the publicly available ETH and UCY datasets and measure its performance using the final displacement error (FDE) and average displacement error (ADE) metrics. We show that our modified algorithm performs better than the Social LSTM in predicting the future trajectory of pedestrians in crowded spaces. Specifically, our model achieves an improvement of 6.2% in ADE and 6.3% in FDE compared to the Social LSTM results in the literature.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑