Abstract Fault diagnosis plays a crucial role in maintaining mechanical equipment reliability. Deep Neural Networks (DNNs) have exhibited superior performance in fault diagnosis under closed sample space assumptions. However, the deployment of neural network models in practical industrial environments faces significant challenges due to the emergence of out-of-distribution (OOD) data, particularly novel fault categories that deviate substantially from the initial training assumptions. To address these limitations, we propose a robust fault diagnosis framework incorporating OOD sample detection and adaptive learning capabilities. The framework utilizes a novel Mixture of Experts model (MoEFormer) as the feature extraction mechanism, which enables fine-grained feature representation while enhancing computational efficiency. Furthermore, we introduce a Gaussian Density Peak Clustering (GDPC) algorithm for OOD data identification and autonomous model retraining. This integrated framework enables real-time adaptation and online diagnosis in industrial scenarios where novel fault categories may emerge. Experimental validation was conducted using the bearing fault dataset from the University of Ottawa and the gear fault dataset from Xi'an Jiaotong University. The results demonstrate that our proposed method outperforms existing state-of-the-art approaches in identifying unknown out-of-distribution faults and achieves superior diagnostic performance. The adaptively updated model attained a diagnostic accuracy of 96.0%, representing a significant improvement of approximately 30 percentage points compared to non-adaptive methods.
Read full abstract