基于不平衡数据与集成学习的柴油机故障诊断研究
Research on Diesel Engine Fault Diagnosis Based on Imbalanced Data and Ensemble Learning
李伟真 1商蕾 1汪敏 1邱天1
作者信息
- 1. 武汉理工大学船海与能源动力工程学院 武汉 430063
- 折叠
摘要
文中将集成学习理论与深度置信网络相结合,采用Adaboost-M2算法作为集成学习生成方法,使用麻雀搜索算法确定不同终止适应度下的不同初始化超参数分布的深度置信网络模型.将上述不同的深度置信网络模型作为基分类器进行集成学习,得到集成Ada-DBN模型.并通过实验分析了分类器数量对集成学习后的模型诊断性能的影响.结果表明:集成Ada-DBN模型不仅能够保证在平衡数据下的诊断能力,还能提高在不平衡数据下的泛化能力,是一种适用于实际柴油机故障诊断的有效方法.
Abstract
Combining the ensemble learning theory with the deep confidence network,Adaboost-M2 al-gorithm was adopted as the ensemble learning generation method,and sparrow search algorithm was used to determine the deep confidence network models with different initial hyperparametric distribu-tions under different termination fitness.The above-mentioned different deep confidence network models were used as base classifiers for ensemble learning,and the integrated Ada-DBN model was obtained.The influence of the number of classifiers on the model diagnosis performance after ensem-ble learning was analyzed through experiments.The results show that the integrated Ada-DBN model can not only ensure the diagnosis ability under balanced data,but also improve the generalization abili-ty under unbalanced data,and it is an effective method for practical diesel engine fault diagnosis.
关键词
船舶柴油机/不平衡数据/故障诊断/深度置信网络/集成学习Key words
marine diesel engine/imbalanced data/fault diagnosis/DBN/ensemble learning引用本文复制引用
基金项目
国家自然科学基金重点项目(U1709215)
工业和信息化部高技术船舶项目(MC-201917-C09)
出版年
2024