参数优化SAE方法及在轴承故障诊断的应用

杜灿谊¹，林祖胜²，张绍辉²

1. 广东技术师范大学汽车与交通工程学院2. 厦门理工学院机械与汽车工程学院

摘 要：稀疏自动编码(Sparse Auto Encoder, SAE)通过寻找一组"超完备"基向量用于挖掘输入数据的内在结构与模式,使得高层输出能够更好的表达输入样本的类别信息,其良好的降维性能受到广泛关注并逐渐应用在机械设备故障诊断中。然而,SAE模型中隐含层特征数直接影响高层输出对低层输入模式的表达效果,简单的设置隐含层特征数难以取得理想的识别效果,针对该问题,利用萤火虫寻优算法的优点,确定各个隐含层的最优特征数,从而确定最优的SAE模型。轴承仿真及故障状态识别实验证明,隐含层特征数确定之后的稀疏自动编码模型在不同测试样本数目下均能取得比浅层结构及随机参数SAE模型更好的识别效果,得到更高的识别正确率。

关键词：深度学习;稀疏自动编码;轴承故障诊断;萤火虫;