【技术实现步骤摘要】
一种故障诊断模型训练方法及装置
[0001]本申请涉及故障诊断
,更具体地,涉及一种故障诊断模型训练方法及装置。
技术介绍
[0002]轴承作为机械传动的关键零件,被广泛应用于各种机械设备中,其健康状况对于机械设备的安全性与稳定性具有重要影响。然而设备在高速、重载等恶劣环境下长期工作时,轴承将会不可避免地发生退化,产生裂纹、磨损等。一旦发生故障将直接影响整个设备的正常运行,轻则给企业造成经济损失,重则引发事故,威胁生命安全。因此,为保证机械设备的正常运行,对轴承的健康状况进行监测以及时排除安全隐患具有重大的工程意义。
[0003]目前,对于滚动轴承的故障诊断方法主要分为基于解析模型和基于数据驱动两方面。基于解析模型的方法需要对故障诊断问题进行解析化表达,对于复杂度较高的系统建模难度大,且建立的模型在其他系统上的普适性较低,实际推广使用具有一定局限性。而基于数据驱动的故障诊断方法因其特征提取能力不足,难以挖掘提取故障数据中更深层次的微小特征,从而限制了诊断准确率的提升。
[0004]随着互联网、物联网等快速兴起与普及,当前社会数据的增长速度比以往任何时期都要迅猛。大数据给深度神经网络提供了充足的训练“原料”,给基于数据驱动的机械智能故障诊断的深入研究和应用提供了新的机遇,当前基于深度学习的故障诊断方法因能有效对故障信息进行表征而被广泛应用于故障诊断领域。但在实际工业活动中,工作人员在缺乏专业知识情况下易赋予故障模式错误的标签分类。因而在真实的工业数据集中,标注错误(即标签噪声)问题是不可避免的。< ...
【技术保护点】
【技术特征摘要】
1.一种故障诊断模型训练方法,其特征在于,包括:接收输入的第一训练数据集,所述第一训练数据集中的所有训练样本均具有标签,其中一部分训练样本具有正确标签,另一部分训练样本具有错误标签;循环执行如下步骤,直到当前模型训练轮次达到模型训练最大轮次:将所述第一训练数据集中的所有训练样本转化为对应的第一特征编码向量;对所述第一特征编码向量进行标签分类,获得所述训练样本属于每一类健康状态的概率;依据所述第一特征编码向量获取所述训练样本的注意力权重;依据所有训练样本的概率、所述注意力权重以及所述训练样本的标签计算当前轮次训练的噪声注意力损失函数;依据所述噪声注意力损失函数进行模型参数的更新。2.根据权利要求1所述的故障诊断模型训练方法,其特征在于,还包括:利用所述第一训练数据集进行对比学习,获得对比损失函数;依据所述噪声注意力损失函数和所述对比损失函数获得当前轮次训练的综合损失函数;并且,依据所述综合损失函数进行模型参数的更新。3.根据权利要求2所述的故障诊断模型训练方法,其特征在于,利用所述第一训练数据集进行对比学习,获得对比损失函数,具体包括:对所述第一训练数据集中的训练样本进行两种不同的数据增强,形成数据增强样本集;对所述数据增强样本集内的任意两个数据增强样本组成的样本对进行对比学习,以获得所述对比损失函数。4.根据权利要求3所述的故障诊断模型训练方法,其特征在于,对所述数据增强样本集内的任意两个数据增强样本组成的样本对进行对比学习,获得所述对比损失函数,具体包括:对于任意样本对,首先,将所述样本对中的两个数据增强样本分别转化为所述第一特征编码向量和第二特征编码向量;然后,分别将所述第一特征编码向量和所述第二特征编码向量映射为空间表示向量;最后计算两个空间表示向量之间的相似度;利用所有样本对的相似度计算样本间互信息,依据所有样本间互信息计算对比损失函数。5.根据权利要求1或2所述的故障诊断模型训练方法,其特征在于,计算所述噪声注意力损失函数之前,还包括:判断当前模型训练轮次是否达到标签纠正启用轮次,所述标签纠正启用轮次小于所述模型训练最大轮次;若是,则对所述标签进行标签纠正,形成纠正后标签,所有具有纠正后标签的训练样本形成第二训练数据集;依据所有训练样本的概率、注意力权重以及与所述训练样本对应的纠正后标签计算当前轮次训练的噪声注意力损失函数;
并且,将所述第一训练数据集更新为所述第二训练数据集,利用所述第二训练数据集进行后续轮次的训练。6.一种故障诊断模型训练装置,其特征在于,包括训练数据接收模块、第一转化模块、分类模块、权重获得模块、第一损失函数计算模块以及参数更新模块;所述训练数据接收模块用于接收输入的第一训练数据集,所述第一训练数据集中的...
【专利技术属性】
技术研发人员:王明君,金泽中,孙东,郑华丽,叶春明,
申请(专利权)人:浙江中烟工业有限责任公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。