Author Login Editor-in-Chief Peer Review Editor Work Office Work

Computer Engineering

   

Elevator safety risk prediction based on domain adaptation and attention mechanism

  

  • Published:2024-04-15

基于领域自适应与注意力机制的电梯安全风险预测

Abstract: As one of the special equipment, the operation safety risk prediction of elevators is crucial. At present, most of the research on elevators is based on elevator component data, and the prediction method will have problems such as low prediction accuracy and poor generalization ability in the case of changing application scenarios. Therefore, a method of elevator safety risk prediction based on domain adaptation and attention mechanism is proposed. This method is based on adversarial domain adaptive network, and uses the attention mechanism to optimize the feature extraction ability of the network. The method includes three parts: feature extractor, label classifier and domain classifier, the input data is the elevator safety risk factor containing both source domain and target domain data, the feature extractor optimized by the attention mechanism adaptively extracts and retains the common key features between the source domain and the target domain, and then the key features are input to the label classifier and the domain classifier at the same time, the transfer learning from the source domain to the target domain is realized through domain adaptation, and the elevator operation status is output through the label classifier. The experimental results show that the prediction accuracy of the proposed method can reach 86.9% when it is transferred to the target domain application scenario, which is 2.6 percentage points higher than that before optimization, and 9.5%, 8.3%, 3.7% and 1.2% higher than that of LSTM-AE, CNN-LSTM, TrAdaBoost.R2 and DSAN, respectively, which can effectively predict elevator safety risks.

摘要: 电梯作为特种设备之一,其运行安全风险预测至关重要。当前对于电梯相关的研究多基于电梯部件数据,并且预测方法在变换应用场景的情况下会出现预测精度低,泛化能力差等问题。为此提出一种基于领域自适应与注意力机制的电梯安全风险预测方法。该方法基于对抗领域自适应网络,并且使用注意力机制优化网络的特征提取能力。方法包括特征提取器,标签分类器和领域分类器三部分,输入数据为同时包含源域与目标域数据的电梯安全风险因素,经由注意力机制优化的特征提取器,自适应提取并保留源域和目标域之间的公共关键特征,然后将关键特征同时输入至标签分类器和领域分类器,通过领域自适应实现由源域至目标域的迁移学习,通过标签分类器输出电梯运行状态。实验结果表明,所提出的方法在迁移至目标域应用场景的情况下,预测精度可以达到86.9%,相较于优化前提高了2.6个百分点,与LSTM-AE、CNN-LSTM、TrAdaBoost.R2、DSAN比较分别高出9.5、8.3、3.7以及1.2个百分点,能够有效对电梯安全风险进行预测。