工矿自动化2024，Vol.50Issue(5)：67-74,8.DOI:10.13272/j.issn.1671-251x.2023110084
基于Social Transformer的井下多人轨迹预测方法
Multi-personnel underground trajectory prediction method based on Social Transformer
摘要
Abstract
Currently,in the prediction methods of underground personnel trajectories in coal mines,Transformer not only has lower computational complexity compared to recurrent neural network(RNN)and long short-term memory(LSTM),but also effectively solves the problem of long-term dependence caused by gradient disappearance when processing data.But when multi personnel are moving simultaneously in the environment,the Transformer's prediction of the future trajectories of all personnel in the scene will have a significant deviation.And currently,there is no model in the field of underground multi personnel trajectory prediction that simultaneously uses Transformer and considers the mutual influence between individuals.In order to solve the above problems,a multi personnel underground trajectory prediction method based on Social Transformer is proposed.Firstly,each individual is independently modeled to obtain their historical trajectory information.Feature extraction is performed using a Transformer encoder,followed by a fully connected layer to better represent the features.Secondly,an interactive layer based on graph convolution is used to connect each other,allowing spatially close networks to share information with each other.This layer calculates the attention that the predicted object allocates to its neighbors when influenced by them,extracts their motion patterns,and updates the feature matrix.Finally,the new feature matrix are decoded by the Transformer decoder to output predictions of future position information.The experimental results show that the average displacement error of Social Transformer is reduced by 45.8%compared to Transformer.Compared with other mainstream trajectory prediction methods such as LSTM,S-GAN,Trajectoron++,and S-STGCNN,the prediction errors are reduced by 67.1%,35.9%,30.1%,and 10.9%,respectively.This can effectively overcome the problem of inaccurate prediction trajectories caused by mutual influence among personnel in the underground multi personnel scenario of coal mines and improve prediction precision.关键词
电子围栏/井下多人轨迹预测/Transformer/交互编码/Social TransformerKey words
electronic fence/underground multi personnel trajectory prediction/Transformer/interaction coding/Social Transformer分类
矿业与冶金引用本文复制引用
马征,杨大山,张天翔..基于Social Transformer的井下多人轨迹预测方法[J].工矿自动化,2024,50(5):67-74,8.基金项目
中央高校基本科研业务费专项项目(FRF-TP-24-060A) （FRF-TP-24-060A）
天地科技股份有限公司科技创新创业资金专项项目(2023-TD-ZD005-005,2023CG-ZB-10). （2023-TD-ZD005-005,2023CG-ZB-10）
