计算机工程与科学

作者：孙庞博, 符琦, 陈安华, 蒋云霞

单位：湖南科技大学计算机科学与工程学院，湖南 湘潭 411201

关键词：滚动轴承,故障诊断,变分自编码器,半监督学习,LightGBM,小样本数据,

基金：国家重点研发计划(2018YFB1702602)；国家自然科学基金(61402167,6177219)；湖南省教育厅科技重点项目(19A174)；湖南省自然科学基金(2018JJ2139)

滚动轴承是旋转机械内常出现问题的重要部件，其故障情况复杂且难以诊断。基于小样本故障数据学习环境，针对小样本学习在提取真实特征值与目标特征值时有较大差异且泛化能力较弱的问题，提出一种采用半监督变分自编码器与LightGBM分类模型相结合的小样本学习模型LSVAE,并利用基于高斯过程的贝叶斯优化改进算法对LightGBM的超参数进行了优化处理，有效地解决了小样本学习性能不稳定，提取特征能力弱，过拟合等问题，并在凯斯西储大学发布的轴承实验数据集上进行了对比实验，结果表明LSVAE模型在面向小样本数据空间时有着更优的诊断准确率。

来源：2021年第9期

《计算机工程与科学》期刊编辑部

查看计算机工程与科学杂志2021年第9期