Python环境下基于门控双注意力机制的滚动轴承剩余使用寿命RUL预测(Tensorflow模块)

机械设备的寿命是其从开始工作持续运行直至故障出现的整个时间段,以滚动轴承为例,其寿命为开始转动直到滚动体或是内外圈等元件出现首次出现故障前。目前主流的滚动轴承RUL预测分类方法包含两种:一是基于物理模型的RUL预测方法,二是基于数据驱动的RUL预测方法。

基于物理模型的RUL预测方法是利用物理学模型来描述基于失效机制的物理现象,并在一定的假设条件下建立滚动轴承RUL预测的公式模型。但要想建立精确的滚动轴承退化物理学模型,需要对复杂的轴承结构有深入的了解,需要相关人员有较深厚的知识背景,费时费力,且模型泛化能力弱。数学模型表达退化趋势的前提是构建合理的健康指数。

基于数据驱动的RUL预测方法可分为三个步骤:一是数据获取,在数据采集的过程中通常采用的是振动信号或是声学信号;二是健康指标构建,构建健康指标来监测旋转机械的退化,主要包括均方根、变分模态分解、数学形态谱熵和主成分分析等方法;三是剩余使用寿命预测方法,基于数据驱动的轴承剩余使用寿命预测方法又可分为基于统计模型的轴承剩余使用寿命预测方法和基于人工智能模型的轴承剩余使用寿命预测方法。

该代码为Python环境下基于门控双注意力机制的滚动轴承剩余使用寿命RUL预测(采用Tensorflow模块),所用模块如下:

复制代码
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
import numpy as np
import matplotlib.pyplot as plt
import scipy.io as scio

Tensorflow模块版本如下:

复制代码
tensorflow=2.8.0

所使用的数据为公开数据,试验台如下:

部分代码如下:

复制代码
# Builds GDAU cell structure
def gdaumodel(gdau_input,gdau_state,n_input, n_hidden, n_output):
    weights = initialize_weights(n_input, n_hidden, n_output)
    r = tf.sigmoid(tf.matmul(weights['ur'], gdau_input) + tf.matmul(weights['wr'], gdau_state) + weights['br']) # The output of the reset gate
    z = tf.sigmoid(tf.matmul(weights['uz'], gdau_input) + tf.matmul(weights['wz'], gdau_state) + weights['bz']) # The output of the update gate
    candidate = tf.tanh(tf.matmul(weights['uh'], gdau_input) + tf.matmul(weights['wh'], tf.multiply(r,gdau_state)) + weights['bh']) # The output of the candidate state
    at = tf.matmul(weights['v'],tf.tanh(tf.matmul(weights['ua1'],gdau_input) + tf.matmul(weights['wa1'],gdau_state)))
    ut = tf.nn.softmax(at, axis=0)
    h = tf.tanh(tf.multiply(ut,(tf.matmul(weights['ua2'],gdau_input) + tf.matmul(weights['wa2'],gdau_state)))) # The output of the attention gate 1
    a1 = tf.sigmoid(tf.matmul(weights['uaa1'],r) + tf.matmul(weights['waa1'],z) + weights['baa1'])
    a2 = tf.tanh(tf.matmul(weights['uaa2'], r) + tf.matmul(weights['waa2'], z) + weights['baa2'])
    a = tf.multiply(a1, a2) # The output of the attention gate 2
    ht = tf.multiply((1-z), gdau_state) + tf.multiply(z ,candidate)/2 + tf.multiply(z, h)/2 + a # The output of hidden layer
    output = tf.matmul(weights['w'], ht) + weights['b'] # The output of output layer
    return output, ht

出图如下:

工学博士,担任《Mechanical System and Signal Processing》审稿专家,担任
《中国电机工程学报》优秀审稿专家,《控制与决策》,《系统工程与电子技术》,《电力系统保护与控制》,《宇航学报》等EI期刊审稿专家。擅长领域:现代信号处理,机器学习,深度学习,数字孪生,时间序列分析,设备缺陷检测、设备异常检测、设备智能故障诊断与健康管理PHM等。

相关推荐
旦莫5 分钟前
Python 教程:我们可以给 Python 文件起中文名吗?
开发语言·python
__Benco7 分钟前
OpenHarmony - 小型系统内核(LiteOS-A)(十),魔法键使用方法,用户态异常信息说明
人工智能·harmonyos
豌豆花下猫22 分钟前
Python 潮流周刊#99:如何在生产环境中运行 Python?(摘要)
后端·python·ai
小杨40425 分钟前
python入门系列二十(peewee)
人工智能·python·pycharm
弧襪26 分钟前
FlaskRestfulAPI接口的初步认识
python·flaskrestfulapi
IT古董27 分钟前
【漫话机器学习系列】225.张量(Tensors)
人工智能
船长@Quant28 分钟前
文档构建:Sphinx全面使用指南 — 进阶篇
python·markdown·sphinx·文档构建
深圳市快瞳科技有限公司28 分钟前
当OCR遇上“幻觉”:如何让AI更靠谱地“看懂”文字?
人工智能·ai·ocr
每天都要写算法(努力版)29 分钟前
【神经网络与深度学习】训练集与验证集的功能解析与差异探究
人工智能·深度学习·神经网络