改进生成对抗网络解决样本不平衡的故障诊断方法
DOI:
CSTR:
作者:
作者单位:

1.南京信息工程大学自动化学院;2.无锡学院自动化学院;3.金陵科技学院智能科学与控制工程学院

作者简介:

通讯作者:

中图分类号:

基金项目:


Research on Improved Generative Adversarial Networks for Fault Diagnosis with Imbalanced Samples
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对工业设备故障诊断中存在的样本不平衡问题,提出一种自适应采样生成对抗网络ADWGAN-GP(Adaptive Density Wasserstein GAN with Gradient Penalty)模型。该模型集成自适应采样模块和改进的WGAN-GP结构,通过最远点采样、局部区域查询和多尺度特征提取来增强对少数类别特征的学习能力,并引入密度损失函数优化生成样本的分布均匀性。实验采用凯斯西储大学和帕德博恩大学轴承数据集,在多种样本不平衡比例(10:1、5:1、2.5:1)下进行验证。结果表明:该模型不仅能够有效平衡各类别样本数量,而且在维持样本多样性的同时确保了数据质量,使得在最大不平衡比例(10:1)条件下的诊断准确率达到96.21%。通过混淆矩阵分析验证,模型在各故障类别上均表现出稳定的分类效果,证实了该方法在平衡样本分布方面的显著优势。

    Abstract:

    To address the sample imbalance problem in industrial equipment fault diagnosis, an adaptive sampling generative adversarial network ADWGAN-GP (Adaptive Density Wasserstein GAN with Gradient Penalty) model. The model integrates the adaptive sampling module and the improved WGAN-GP structure, enhances the learning ability of features of a few categories through farthest point sampling, local region querying and multi-scale feature extraction, and introduces a density loss function to optimise the distribution uniformity of the generated samples. The experiments are validated with Case Western Reserve University and University of Paderborn bearing datasets under various sample imbalance ratios (10:1, 5:1, 2.5:1). The results show that the model not only effectively balances the number of samples in each category, but also ensures data quality while maintaining sample diversity, resulting in a diagnostic accuracy of 96.21% under the condition of maximum imbalance ratio (10:1). Verified by confusion matrix analysis, the model shows stable classification effects on all fault categories, confirming the significant advantage of the method in balancing the sample distribution.

    参考文献
    相似文献
    引证文献
引用本文
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-12-21
  • 最后修改日期:2025-03-02
  • 录用日期:2024-12-27
  • 在线发布日期:
  • 出版日期:
文章二维码