基于注意力时间卷积网络和双向门控循环单元的轴承故障诊断  

Bearing Fault Diagnosis Based on Attention Temporal Convolutional Network and Bidirectional Gated Recurrent Unit

作　　者：张璐莹 侯立群[1] ZHANG Luying;HOU Liqun(Department of Automation,North China Electric Power University,Baoding 071003,China)

机构地区：[1]华北电力大学自动化系,河北保定071003

出　　处：《电力科学与工程》2023年第6期62-70,共9页Electric Power Science and Engineering

基　　金：河北省自然科学基金(F2016502104)。

摘　　要：针对传统特征提取方法依赖人工经验以及传统神经网络未充分利用时间序列信息的问题,首先通过时间卷积网络的空洞因果卷积、随机丢弃层和残差结构跨时间步提取不同振动信号的特征;然后引入注意力机制获取关键信息,实现特征优化选择;再利用双向门控循环单元捕捉长期依赖关系;最后通过归一化指数函数进行故障分类。实验结果表明,在不同训练样本比例下,该方法的识别精度高于一维卷积神经网络、双向长短期记忆网络、双向循环神经网络;用该方法能够有效识别轴承故障类型,且模型的泛化能力较强。In order to solve the problem that traditional feature extraction methods rely on artificial experience and traditional neural networks do not make full use of time series information,firstly,the features of different vibration signals are extracted across time steps through the dilated causal convolution,random discard layer and residual structure of the temporal convolutional network;then,the attention mechanism is introduced to obtain key information and optimize feature selection;bidirectional gated recurrent units are used to capture long-term dependencies;finally,the normalized exponential function is used to classify faults.The experimental results show that the recognition accuracy of this method is higher than that of one-dimensional convolutional neural network,bidirectional long short-term memory network and bidirectional recurrent neural network under different training sample proportions;this method can identify bearing fault types effectively,and the model has strong generalization ability.

关 键 词：轴承 故障诊断 时间卷积网络 注意力机制 门控循环单元 

分 类 号：TH133.3[机械工程—机械制造及自动化] TH17