基于IWAE的不平衡数据集下轴承故障诊断研究
李梦男
李琨
吴聪
昆明理工大学 信息工程与自动化学院,昆明 650504
摘要：针对目前现有轴承故障诊断方法对不平衡数据集中的少数类诊断准确率低的问题,提出了不平衡数据集下基于重要性加权自编码器(Importance Weighted Auto-encoder,IWAE)的轴承故障诊断方法.首先通过少数类的样本数据来训练IWAE网络,将生成的样本数据加入到原始数据集中,得到平衡后的数据集;然后引入深度学习方法作为诊断网络,将平衡后的数据集直接输入诊断网络中,自适应的学习故障特征,实现故障分类.为了增强诊断网络的准确率,使用一维多尺度卷积神经网络进行故障诊断.大量的定性定量实验表明,所提出的方法在不平衡比为 1/7 时,少数类诊断的准确率已经能够达到 98.90％,均优于其他现有模型,并且拥有较好的收敛性和泛化性.
关键词：不平衡数据集重要性加权自编码一维多尺度卷积神经网络轴承故障诊断
分类号：TP277(自动化技术及设备)TH133.3(机械零件及传动装置)
论文发表日期：2023-06-15
在线出版日期：2026-09-12（本平台首次上网日期，不代表文献的发表时间）
页数：7( 569-575 )
英文信息