1.基于轻量级卷积神经网络的自适应轴承故障诊断方法,其特征在于,包括:步骤1:从轴承数据集中提取轴承图像并进行预处理,得到灰度图像,且按照预设比例将所述灰度图像划分为训练集和测试集;
步骤2:设计初始模型的缩放因子,并利用训练集对初始模型进行迭代训练,确定最优诊断精度的模型权重,并基于所述模型权重得到第一诊断模型;
步骤3:利用测试集对第一诊断模型进行测试,根据测试结果对第一诊断模型进行调整,得到目标诊断模型;
步骤4:利用所述目标诊断模型对当前机械设备的目标轴承图像进行故障诊断,得到故障类型;
步骤2之前,还包括,构建所述初始模型,具体为:建立5*5的卷积层作为所述初始模型的第一层;
分别将两个第一类型模块作为初始模型的第二层和第三层,将两个第二类型模块作为初始模型的第四层和第五层;
将全连接层作为初始模型的第六层;
将第一层到第六层进行依次连接,构建得到所述初始模型;
在第二层到第五层中,上一层的输出将作为之后所有层的输入,第二层对应的第一类型模块输出的特征矩阵作为第三层对应的第一类型模块以及第四层和第五层对应的第二类型模块的输入值;
所述第一类型模块的结构如下:
第一类型模块的第一层为3*3和5*5两种不同的卷积核,且两个卷积核并列,第二层为注意力模块,第三层为1*1的卷积;
所述第二类型模块的结构如下:
第二类型模块的第一层为1*1的卷积,第二层为3*3,5*5,7*7和9*9四种不同的卷积核,且四个卷积核并列,第三层为注意力模块,第四层为1*1的卷积;
设计初始模型的缩放因子,包括:
所述初始模型的缩放因子包括对第一类型模块和第二类型模块的宽度缩放因子β和对第一类型模块和第二类型模块的重复次数缩放因子γ;
所述初始模型的第二层到第五层的模块重复次数依次为1γ、2γ、1γ、2γ;
所述宽度缩放因子β和的重复次数缩放因子γ的取值满足如下公式:Φ
r=α
Φ
N=n
Φ
w=β
Φ
d=γ
2 2
α·n·β·γ ≈4
α≥1,n≥1,β≥1,γ≥1
其中,r表示初始模型的的分辨率,α表示所述灰度图像的尺寸缩放因子,N表示初始模型的通道扩充数值,n表示初始模型的通道缩放因子,w表示第一类型模块和第二类型模块的宽度,d表示第一类型模块和第二类型模块的深度,Φ表示复合系数;
按照上述约束确定宽度缩放因子β和重复次数缩放因子γ的可取值集合,从所述取值集合中选取在其他参数不变的情况下,初始模型的诊断精度最大值对应的目标取值,作为初始模型的缩放因子。
2.根据权利要求1所述的基于轻量级卷积神经网络的自适应轴承故障诊断方法,其特征在于,步骤1中,从轴承数据集中提取轴承图像并进行预处理,得到灰度图像,包括:获取与当前机械设备的目标轴承的相似度大于预设相似度的轴承图像;
从所述轴承图像对应的分别在驱动端、负载端、轴承端的原始震动信号任取预设数量段进行采样,得到振幅点,并将所述振幅点展开,依次重新组合,得到轴承图像对应的灰度图像。
3.根据权利要求2所述的基于轻量级卷积神经网络的自适应轴承故障诊断方法,其特征在于,所述任取预设数量段进行采样时为不重叠采样。
4.根据权利要求1所述的基于轻量级卷积神经网络的自适应轴承故障诊断方法,其特征在于,步骤2中,利用训练集对初始模型进行迭代训练,确定最优诊断精度的模型权重,并基于所述模型权重得到第一诊断模型,包括:对初始模型中卷积层中的神经元进行随机失活,每个卷积层的神经元的失活比率如下:其中,其中Di表示第i层卷积的失活比率数值,Ltotal表示所述初始模型中层卷积的层数,Dtarget表示初始模型整体的失活比率数值;
按照所述失活比率和预设激活函数来对初始模型进行设置,利用训练集对设置好的初始模型进行迭代训练,将根据每次迭代训练结果对模型权重进行更新,在迭代完成后选出最优诊断精度对应的模型权重;
将最优诊断精度对应的模型权重替换初始模型的初始权重,将替换后的初始模型作为第一诊断模型。
5.根据权利要求1所述的基于轻量级卷积神经网络的自适应轴承故障诊断方法,其特征在于,步骤3中,利用测试集对第一诊断模型进行测试,根据测试结果对第一诊断模型进行调整,得到目标诊断模型,包括:获取第一数据集对应的第一测试集和第二数据集对应的第二测试集;
分别利用所述第一测试集和第二测试集对第一诊断模型进行测试,并根据测试结果建立第一测试集对应的第一混淆矩阵和第二测试集对应的第二混淆矩阵;
基于所述第一混淆矩阵确定第一故障分类准确率,基于所述第二混淆矩阵确定第二故障分类准确率;
当所述第一故障分类准确率和第二故障分类准确率均满足预设准确率要求时,确定所述第一诊断模型为目标诊断模型;
否则,重复步骤2对第一诊断模型进行再次迭代训练,直到测试结果满足预设准确率要求。
6.根据权利要求1所述的基于轻量级卷积神经网络的自适应轴承故障诊断方法,其特征在于,步骤4中,利用所述目标诊断模型对当前机械设备的目标轴承图像进行故障诊断,得到故障类型,包括:从当前机械设备的不同角度和不同时间获取多个目标轴承图像;
将目标轴承图像按照与对轴承数据集的图像预处理相同的方式进行处理,得到目标灰度图像;
将每个目标灰度图像输入所述目标诊断模型中进行故障诊断,得到对应的故障类型;
对所有目标灰度图像对应的故障类型进行分析,确定得到的故障类型次数最多的故障类型作为所述当前机械设备的故障类型。