面向小样本约束下旋转机械故障分类的新型渐进神经网络(PNN)
PNN: A Novel Progressive Neural Network for Fault Classification in Rotating Machinery under Small Dataset Constraint
摘要 Abstract
旋转机械的故障检测是一项复杂任务,特别是在小样本且数据异质性较高的场景下。传感器布置、机械配置以及结构差异等因素进一步增加了问题的复杂性。传统深度学习方法通常需要大规模同质数据集,限制了其在工业环境中数据匮乏场景下的适用性。虽然迁移学习和少量学习显示出潜力,但它们往往受限于对大量故障数据集的需求。本研究提出了一种统一框架,利用一种新型的渐进神经网络(PNN)架构来解决这些挑战。PNN通过利用之前估计的所有特征逐步估计更高阶的固定大小精炼特征,并将其附加到特征集中。每层固定的特征输出控制了PNN的复杂度,使其适合从少量数据中进行有效学习。该框架的有效性在八个数据集上得到了验证,包括六个开源数据集、一个内部故障模拟器和一个真实工业数据集。PNN在不同规模的数据集和机械类型下的故障检测中实现了最先进的性能,展示了卓越的泛化和分类能力。
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.