• 论文
主办单位:煤炭科学研究总院有限公司、中国煤炭学会学术期刊工作委员会
基于Social Transformer的井下多人轨迹预测方法
  • 49
  • 作者

    马征杨大山张天翔

  • 单位

    煤炭科学技术研究院有限公司 煤炭智能开采与岩层控制全国重点实验室煤矿应急避险技术装备工程研究中心 北京市煤矿安全工程技术研究中心 北京科技大学自动化学院

  • 摘要
    目前煤矿井下人员轨迹预测方法中,Transformer与循环神经网络(RNN)、长短期记忆(LSTM)网络相比,在处理数据时不仅计算量小,同时有效解决了梯度消失导致的长时依赖问题。但当环境中涉及多人同时运动时,Transformer对于场景中所有人员未来轨迹的预测会出现较大偏差。并且目前在井下多人轨迹预测领域尚未出现一种同时采用Transformer并考虑个体之间相互影响的模型。针对上述问题,提出一种基于Social Transformer的井下多人轨迹预测方法。首先对井下每一个人员独立建模,获取人员历史轨迹信息,通过Transformer编码器进行特征提取,接着由全连接层将特征进行更好的表示,然后通过基于图卷积的交互层相互连接,该交互层允许空间上接近的网络彼此共享信息,计算预测对象在受到周围邻居影响时对周围邻居分配的注意力,从而提取其邻居的运动模式,继而更新特征矩阵,最后新的特征矩阵由Transformer解码器进行解码,输出对于未来时刻的人员位置信息预测。实验结果表明,Social Transformer的平均位移误差相较于Transformer降低了45.8%,且与其他主流轨迹预测方法LSTM,S-GAN,Trajectron++和Social-STGCNN相比分别降低了67.1%,35.9%,30.1%和10.9%,有效克服了煤矿井下多人场景中由于人员间互相影响导致预测轨迹失准的问题,提升了预测精度。
  • 关键词

    电子围栏井下多人轨迹预测Transformer交互编码SocialTransformer

  • 文章目录
    0 引言
    1 方法整体结构
    2 Transformer网络模型
    2.1 Transformer网络模型结构
    2.2 自注意力机制
    3 Social Transformer网络模型
    3.1 时序建模
    3.2 周围人员寻找
    3.3 信息交互编码
    3.4 未来轨迹预测
    4 实验与结果分析
    4.1 数据集与评价指标
    4.2 模型训练
    4.3 定量实验
    4.4 消融实验
    4.4.1 预测序列长度对轨迹预测效果影响
    4.4.2 历史数据缺失对轨迹预测效果影响
    4.5 定性实验
    5 结论
相关问题

主办单位:煤炭科学研究总院有限公司 中国煤炭学会学术期刊工作委员会

©版权所有2015 煤炭科学研究总院有限公司 地址:北京市朝阳区和平里青年沟东路煤炭大厦 邮编:100013
京ICP备05086979号-16  技术支持:云智互联