基于神经常微分方程的机械故障诊断方法

Mechanical fault diagnosis method based on neural ordinary differential equations

  • 摘要: 针对传统的深度学习故障诊断方法中存在架构可解释性差以及盲目堆叠层数导致的参数增加和内存消耗等问题,将神经常微分方程(neural ordinary differential equations, NODE)引入到机械故障诊断中。搭建面向机械故障诊断的神经常微分方程网络架构,利用神经网络参数化隐藏状态的导数代替指定隐藏层的离散序列。通过构建故障数据与故障类型的非线性关系,使用常微分方程求解器(ODE solver)完成对不同故障类别的分类任务,形成一种端对端的故障诊断模式。将该方法应用到机械故障诊断领域,搭建特定的神经常微分方程网络模型,通过故障数据的输入完成对不同故障类别的分类任务。将该模型应用到航空发动机主轴轴承故障诊断中,并与残差网络模型的故障诊断方法进行对比。试验结果表明,在确保准确率不降低的情况下,该方法不仅减少了内存消耗,而且将模型参数数量减少了将近五倍。

     

    Abstract: Based on the problems of poor interpretability, as well as parameter increase and memory consumption caused by blind stacking layers in traditional fault diagnosis method based on deep learning, Neural ordinary differential equation (NODE) is introduced into mechanical fault diagnosis, the network structure of NODE for machinery fault diagnosis is constructed. In the constructed structure, the derivatives of the parameterized hidden states of the neural network are used to replace the discrete sequences of the specified hidden layers. By constructing a nonlinear relationship between fault data and fault types, an ordinary differential equation solver (ODE solver) is used to complete the classification of different fault types, and an end-to-end fault diagnosis model is formed. The proposed method is applied to mechanical fault diagnosis to build a specific NODE network model, and the classification task of different fault categories is accomplished through the input of fault data. The constructed model is applied to the fault diagnosis of spindle bearing in the aircraft engine, and compared with the fault diagnosis method based on residual network model. The experimental results show that the constructed model and residual network model have satisfactory accuracy. However, the constructed model not only reduces the memory consumption, but also reduces the number of model parameters by almost five times.

     

/

返回文章
返回