MAML-S3M: Selective state space meta-learning for cross-condition few-shot bearing fault diagnosis

Abstract

Deep learning achieves widespread success in fault diagnosis. However, its effectiveness is hindered in practical industrial environments due to complex operating conditions and data sparsity. This article proposes a novel model-agnostic meta-learning framework based on a selective state space model (MAML-S3M) to address the challenge of cross-condition few-shot bearing fault diagnosis. The framework introduces three core innovations. First, the continuously stacked selective state space module dynamically adjusts its receptive field, enabling precise feature extraction under varying conditions. Second, the channel attention mechanism is combined with the selective state space model to capture multi-scale features, thereby enhancing the feature extraction capability of the model. Third, the introduction of an explicit information discarding strategy during meta-task optimization refines the meta-learning process, thereby yielding optimal parameters. Extensive experiments on bearing datasets across different operating conditions demonstrate that the proposed MAML-S3M achieves superior diagnostic accuracy, with an average accuracy of 99.18% across six cross-condition tasks, outperforming state-of-the-art methods such as generalized model-agnostic meta-learning (GMAML) by at least 1.1%. The improvements are particularly helpful in scenarios with complex operating conditions and scarce samples, maintaining over 94% accuracy even in the challenging “10-way 1-shot” setting. We have made the paper’s results publicly available on GitHub. The link is as follows: https://github.com/12138250/MAML-S3M.