arXiv:2503.18263cs.LG2025-03被引 6

小数据下精准识别旋转机械故障的新神经网络

PNN: A Novel Progressive Neural Network for Fault Classification in Rotating Machinery under Small Dataset Constraint

  • 逐层融合特征,固定输出尺寸控制复杂度
  • 八组数据测试均达顶尖性能,泛化能力强
  • 适合工业场景中小样本故障诊断任务

旋转机械故障检测在小规模、异构数据集场景下尤为复杂,传感器位置差异、设备配置和结构变化加剧了挑战。传统深度学习需大量同质数据,难以适用于数据稀缺的工业环境。尽管迁移学习和少样本学习有潜力,但通常依赖大量故障数据。本文提出一种统一框架,采用新型渐进式神经网络(PNN)架构,通过逐层利用先前估计的特征来推断高阶特征,并将其加入特征集,每层输出固定尺寸特征,有效控制模型复杂度,适合小数据学习。在八个数据集上验证,包括六个开源数据集、一个内部故障模拟器数据集和一个真实工业数据集,PNN在不同数据规模和设备类型下均实现领先性能,展现出优异的泛化与分类能力。

原文摘要 · Abstract (English)

Fault detection in rotating machinery is a complex task, particularly in small and heterogeneous dataset scenarios. Variability in sensor placement, machinery configurations, and structural differences further increase the complexity of the problem. Conventional deep learning approaches often demand large, homogeneous datasets, limiting their applicability in data-scarce industrial environments. While transfer learning and few-shot learning have shown potential, however, they are often constrained by the need for extensive fault datasets. This research introduces a unified framework leveraging a novel progressive neural network (PNN) architecture designed to address these challenges. The PNN sequentially estimates the fixed-size refined features of the higher order with the help of all previously estimated features and appends them to the feature set. This fixed-size feature output at each layer controls the complexity of the PNN and makes it suitable for effective learning from small datasets. The framework's effectiveness is validated on eight datasets, including six open-source datasets, one in-house fault simulator, and one real-world industrial dataset. The PNN achieves state-of-the-art performance in fault detection across varying dataset sizes and machinery types, highlighting superior generalization and classification capabilities.

故障诊断小样本学习神经网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。