Python环境下基于深度学习的旋转机械故障诊断及其权重可视化

随着神经网络所要完成的任务越来越智能化,其内部的结构也变得越来越复杂,神经网络学到的解题方法也越来越难以被人类所理解。神经网络的内部单元就好比一个"黑箱",虽然这种处理方法能够在某种程度上达到非常不错的效果,但其原因却非常模糊不清,例如卷积核在学习提取特征的过程中,究竟学习到的是什么?在卷积神经网络的架构设计中应该设计多少层网络、每层又该包含多少的卷积核数,以及如何选择激活函数,且为什么某些选择会优于另一些选择?这些问题的答案不仅有利于提升我们对CNN的科学理解,而且还能提升它们的实用性。随着神经网络越来越多地参与解决现实世界中的问题,解决这个"黑箱"问题也变得越发重要。

然而人们对这些网络工作原理的理解仍然滞后于其发展,至今没有人能够从数学原理上对该模型予以解释。研究者们都是通过不断的试错和反复的调参过程才得到较为满意的结果。这对理论研究者,尤其是数学家来说当然是不可接受的。此外,目前实现CNN的方法需要大量训练数据,而且不同的模型设计对结果有着很大的影响。从理论上更深入地理解卷积神经网络能够减轻对数据驱动设计的依赖。尽管已有实证研究调查了所实现的网络的运行方式,但到目前为止,这些结果很大程度上还局限在内部处理过程的可视化上,即理解CNN中不同层中发生的情况。

神经网络的特征层可视化是一个新兴的科学研究领域,虽然无法完全"打开"这个"黑箱",但是仍然出现了很多探索这个"黑箱"的尝试工作。神经网络的可视化研究,具体来说是对神经网络所提取的特征的可视化研究,不仅有助于理解网络内部结构的工作机制,而且对神经网络在其他应用上的研究也具有指导意义,避免了盲目的调参或试错。神经网络的训练涉及大量的反复试错,直到找到合适的参数为止,这个过程既耗时又容易出错。另外一个训练好的神经网络相当于含百万参数的复杂非线性函数,很难指出其优缺点。神经网络用在其他领域如工业、政府机构及医疗机构等同样会因为其内部的"黑盒"特性而很难让人们理解。试想一名医生若用深度神经网络的方法来诊断一位病人,除非该医生能够理解神经网络的工作原理他才有可能相信这种自动诊断方法。更好地理解这些网络能够帮助研究者们找到新的指导训练网络的方法,对常用网络中的缺陷进行检测并对其进行改善,使得训练的网络层数更深、提取的特征更加有效。

鉴于此,本项目采用卷积神经网络对滚动轴承进行故障诊断并进行特征可视化,采用的数据集为西储大学轴承数据集,Python版本为3.9.0,所用部分模块如下:

复制代码
import numpy as np
import tensorflow as tf
from tensorflow.keras.applications.vgg19 import VGG19
from tensorflow.keras.preprocessing import image
from tensorflow.keras.applications.vgg19 import preprocess_input
from tensorflow.keras.models import Model, load_model

重要模块版本如下:

复制代码
tensorflow=2.8.0
keras=2.8.0
sklearn=1.0.2

部分出图如下:

完整代码可由知乎学术咨询获得:

Python环境下基于深度学习的旋转机械故障诊断及其权重可视化

工学博士,担任《Mechanical System and Signal Processing》审稿专家,担任《中国电机工程学报》优秀审稿专家,《控制与决策》,《系统工程与电子技术》,《电力系统保护与控制》,《宇航学报》等EI期刊审稿专家。

擅长领域:现代信号处理,机器学习,深度学习,数字孪生,时间序列分析,设备缺陷检测、设备异常检测、设备智能故障诊断与健康管理PHM等。

相关推荐
Dm_dotnet1 小时前
公益站Agent Router注册送200刀额度竟然是真的
人工智能
算家计算2 小时前
7B参数拿下30个世界第一!Hunyuan-MT-7B本地部署教程:腾讯混元开源业界首个翻译集成模型
人工智能·开源
用户2519162427112 小时前
Python之语言特点
python
机器之心2 小时前
LLM开源2.0大洗牌:60个出局,39个上桌,AI Coding疯魔,TensorFlow已死
人工智能·openai
刘立军2 小时前
使用pyHugeGraph查询HugeGraph图数据
python·graphql
Juchecar3 小时前
交叉熵:深度学习中最常用的损失函数
人工智能
林木森ai3 小时前
爆款AI动物运动会视频,用Coze(扣子)一键搞定全流程(附保姆级拆解)
人工智能·aigc
聚客AI4 小时前
🙋‍♀️Transformer训练与推理全流程:从输入处理到输出生成
人工智能·算法·llm
BeerBear5 小时前
【保姆级教程-从0开始开发MCP服务器】一、MCP学习压根没有你想象得那么难!.md
人工智能·mcp
小气小憩5 小时前
“暗战”百度搜索页:Monica悬浮球被“围剿”,一场AI Agent与传统巨头的流量攻防战
前端·人工智能