面向类不均衡数据的多任务博弈概率分类向量机

作者：潘海洋,李丙新,郑近德,等 日期：2024-03-26/span> 浏览：1005 查看PDF文档

摘要:在工程实际中获取的故障样本往往会呈现不均衡特点,同时传统的分类模型也会存在局限性。针对这些问题,基于稀疏贝叶斯理论、模糊隶属度等理论,提出了一种多任务博弈概率分类向量机(MGPCVM)分类方法。首先,在MGPCVM的目标函数中,设计了博弈因子,将不同类样本质心间的博弈信息赋予每个样本特定的样本质心敏感值,以解决传统分类器对不平衡数据集分类表现较差的问题;然后,在贝叶斯框架理论下,采用截断高斯先验分布的方法,使样本参数的正负与对应的标签信息相一致,且使样本质心敏感值产生了稀疏估计;最后,将MGPCVM方法应用于两种不同实验平台采集的滚动轴承实验数据处理,进行了故障诊断有效性验证。研究结果表明:在不同的不平衡比(IR)下,MGPCVM方法的准确率均保持在95%以上,相对于支持向量机(SVM)、概率分类向量机(PCVM)等方法提升了4%~8%;与典型向量式分类方法相比,MGPCVM方法可以在不平衡数据条件下表现出优越的分类性能,适用于实际工况中数据失衡的分类问题。

关键词:滚动轴承;故障诊断;多任务博弈概率分类向量机;支持向量机;概率分类向量机;不均衡比;故障分类模型

中图分类号:TH133.33文献标识码:A文章编号:1001-4551(2024)03-0430-08


潘海洋,李丙新,郑近德,等.面向类不均衡数据的多任务博弈概率分类向量机［J］.机电工程,2024,41(3):430-437.

PAN Haiyang, LI Bingxin, ZHENG Jinde, et al. Multitask game probabilistic classification vector machine for class-imbalanced data［J］.Journal of Mechanical & Electrical Engineering, 2024,41(3):430-437.