我们的服务符合学术规范和道德
专业 高端让您使用时没有后顾之忧

工矿自动化2024年第05期:基于Social Transformer的井下多人轨迹预测方法

来源:期刊VIP网 时间:


作者:马征;杨大山;张天翔;

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

摘要:目前煤矿井下人员轨迹预测方法中,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

基金资助:中央高校基本科研业务费专项项目(FRF-TP-24-060A); 天地科技股份有限公司科技创新创业资金专项项目(2023-TDZD005-005,2023CG-ZB-10)