基于Social Transformer的井下多人轨迹预测方法
马征1
杨大山1
张天翔2
1.煤炭科学技术研究院有限公司,北京 100013;煤炭智能开采与岩层控制全国重点实验室,北京 100013;煤矿应急避险技术装备工程研究中心,北京 100013;北京市煤矿安全工程技术研究中心,北京 1000132.北京科技大学 自动化学院,北京 100083
摘要：目前煤矿井下人员轨迹预测方法中,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交互编码Social Transformer
分类号：TD67(矿山电工)
资助基金：中央高校基本科研业务费专项项目(FRF-TP-24-060A)天地科技股份有限公司科技创新创业资金专项项目(2023-TD-ZD005-005)天地科技股份有限公司科技创新创业资金专项项目(2023CG-ZB-10)
论文发表日期：2024-05-10
在线出版日期：2025-08-15（本平台首次上网日期，不代表文献的发表时间）
页数：8( 67-74 )
英文信息