Continuously Indexed Domain Generalization for Fault Diagnosis under Continuously Varying Working Conditions

Deep learning-based fault diagnosis often suffers from significant accuracy degradation under varying working conditions. While domain generalization methods can ensure consistency in fault prediction when the unavailable test data is out-of-domain, existing research typically involves discretizing continuous operating conditions, which fails to fully utilize the continuity of common factors such as speed and load. This oversight limits improvements in generalization capability. To address this, we propose a fault diagnosis framework suitable for continuously varying working conditions, utilizing continuously indexed domain generalization methods. This framework builds on wavelet transform and convolutional neural networks, incorporating Continuous Domain Mix (CDMix) data augmentation and Continuous Mutual Information Minimization (CMIM) loss constraints to achieve cross-domain generalization. Specifically, CDMix determines mixing probabilities by measuring the distance between domain labels of samples, enhancing the reliability of the generated out-of-domain data. Meanwhile, CMIM estimates the mutual information between features and continuous domain labels using kernel methods, then minimizes it to guide the network in extracting fault features that are invariant to continuous working conditions. The proposed methods improve generalization capability and cross-domain fault classification accuracy of fault diagnosis models. Extensive experiments on gearbox and bearing fault datasets validate the effectiveness of the proposed CDMix-CMIM framework, demonstrating significant superiority over existing methods and good robustness.