To address the issues of incomplete feature extraction, poor stability, and limited generalization in traditional fault diagnosis models, a model based on a multi-scale convolutional neural networks (MCNN), bidirectional gated recurrent units (BiGRU), and multi-head self-attention mechanism (MSA) was proposed. The model was designed to achieve comprehensive feature extraction from both spatial and temporal perspectives. It took raw vibration signals as input, and multi-scale features were extracted through convolution kernels of different sizes. A multi-head self-attention mechanism was used to dynamically adjust output weights, disregarding redundant information and weighting the extracted features for fusion. Then the fused features were input into a BiGRU network, which utilized a bidirectional information fusion mechanism to explore information from both past and future directions, capturing dependencies between different parts of the input sequence. Finally, Softmax was employed for classification. Experimental validation was conducted using three bearing fault datasets, and the results show that the proposed model has excellent performance metrics on different datasets and showcases good generalization and feasibility.
| 科 Family | 属数 Number of genus | 种数 Number of species | 占总种数比例 Percentage of total species (%) | 属 Genus | 种数 Number of species | 占总种数比例 Percentage of total species (%) |
|---|---|---|---|---|---|---|
| 鹅膏菌科Amanitaceae | 2 | 11 | 5.26 | 鹅膏菌属 Amanita | 10 | 4.78 |
| 小菇科 Mycenaceae | 2 | 12 | 5.74 | 丝盖伞属 Inocybe | 5 | 2.39 |
| 多孔菌科 Polyporaceae | 8 | 14 | 6.70 | 蜡蘑属 Laccaria | 5 | 2.39 |
| 红菇科 Russulaceae | 3 | 23 | 11.00 | 小皮伞属 Marasmius | 6 | 2.87 |
| 小菇属 Mycena | 11 | 5.26 | ||||
| 光柄菇属 Pluteus | 5 | 2.39 | ||||
| 红菇属 Russula | 17 | 8.13 | ||||
| 栓菌属 Trametes | 5 | 2.39 |