改进黏菌算法优化TCN−LSTM−MHSA的巷道锚杆(索)应力预测模型

TCN-LSTM-MHSA model optimized by improved slime mould algorithm for stress prediction of roadway anchor bolts (cables)

  • 摘要: 锚杆(索)应力的变化过程呈现明显的短期突变与长期时序依赖特征,而传统单一预测模型对长期趋势建模能力有限且对局部突变敏感性不足,往往难以全面捕捉上述复杂特征。针对该问题,提出一种基于改进黏菌算法(ISMA)优化时间卷积网络(TCN)−长短期记忆网络(LSTM)−多头自注意力机制(MHSA)的锚杆(索)应力预测模型。在煤矿巷道锚杆(索)应力预测问题中,模型训练过程通常涉及超参数调整、学习率选择等复杂优化任务,为提升模型的训练效率与预测精度,提出ISMA,引入邻域搜索与动态步长因子增强局部搜索能力,融合人工蜂群搜索机制提升全局搜索效率,有效增强模型跳出局部最优解的能力。TCN−LSTM−MHSA模型采用TCN提取局部时序特征,利用LSTM学习数据的长期依赖关系,通过MHSA强化对全局时序依赖的建模,从而提高模型对锚杆(索)应力的预测能力。在TCN−LSTM−MHSA模型的训练中利用ISMA对学习率进行迭代寻优,以提高模型的预测精度和速度。实验结果表明:① 与黏菌算法(SMA)、遗传算法(GA)、粒子群算法(PSO)、麻雀搜索算法(SSA)相比,ISMA优化策略在多个基准函数测试中表现出更优的收敛速度与寻优能力。② 在应力预测实验中,通过消融实验验证了TCN,LSTM,MHSA模块的必要性。③ ISMA优化TCN−LSTM−MHSA模型在MAE,RMSE及R2等指标上均优于BP,GRU等主流预测模型,具有更高的预测精度和稳定性。

     

    Abstract: The variation process of anchor bolt (cable) stress exhibits distinct short-term fluctuations and long-term temporal dependencies. However, traditional single prediction models have limited capability in modeling long-term trends and insufficient sensitivity to local fluctuations, often making it difficult to fully capture these complex features. To address this problem, an anchor bolt (cable) stress prediction model based on an Improved Slime Mould Algorithm (ISMA) optimized Temporal Convolutional Network (TCN)-Long Short-Term Memory (LSTM)-Multi-Head Self-Attention (MHSA) architecture is proposed. In the problem of anchor bolt (cable) stress prediction in coal mine roadways, model training often involves complex optimization tasks such as hyperparameter tuning and learning rate selection. To improve the training efficiency and prediction accuracy of the model, ISMA was proposed, which enhanced local search capability by introducing neighborhood search and a dynamic step-size factor. Global search efficiency was improved through integrating an Artificial Bee Colony (ABC) search mechanism, thereby effectively improving the model's ability to escape from local optima. The TCN-LSTM-MHSA model was constructed by using TCN to extract local temporal features, employing LSTM to learn long-term dependencies in the data, and strengthening global temporal modeling through MHSA, thereby enhancing the prediction capability for anchor bolt (cable) stress. During training, ISMA was used to iteratively optimize the learning rate of the TCN-LSTM-MHSA model to improve prediction accuracy and speed. Experimental results showed that: ① Compared with the Slime Mould Algorithm (SMA), Genetic Algorithm (GA), Particle Swarm Optimization (PSO), and Sparrow Search Algorithm (SSA), the ISMA optimization strategy demonstrated better convergence speed and optimization ability in multiple benchmark function tests. ② In the stress prediction experiment, ablation experiments verified the necessity of TCN, LSTM, and MHSA modules. ③ The ISMA-optimized TCN-LSTM-MHSA model outperformed mainstream prediction models such as BP and GRU in MAE, RMSE, and R2 metrics, showing higher prediction accuracy and stability.

     

/

返回文章
返回