A dual-branch inverted residual network guided by mixed loss for the bearing fault diagnosis with few samples
摘要
Aiming at the problem of few-sample fault diagnosis in industrial scenarios, a dualbranch inverted residual network guided by mixed loss (DBIRN-ML) is developed in this paper. First, a multi-depth feature fusion network is established based on inverted residual modules, an attention mechanism and a backward-forward feature fusion strategy. Its paired input mechanism expands learning tasks to alleviate the few-shot limitation. Then, a contrastive loss is constructed between the two feature extraction branches, while classification losses are separately connected after each branch of the dual-branch network. The weighted sum of the above loss functions forms a mixed loss, enabling the network to synergistically extract contrastive information from sample pairs and categorical information from individual samples, thereby further enhancing the fault identification capability of DBIRN-ML under few-shot conditions. Two case studies validate the diagnosis effectiveness of the method in the limited sample situation.