To address the challenge of low fault diagnosis accuracy in traditional neural networks with few labeled samples, a method based on contrastive learning and convolution transformer network is proposed.
First, raw monitoring data are transformed into similar sample pairs by data augmentation. These similar sample pairs are then mapped to a deep feature space by a feature extractor. A transformer network is utilized to design cross-prediction tasks for both local and global comparisons, facilitating the clustering of data with the same fault type by comparing the intrinsic similarity between the same batches of data. Finally, the downstream classification network is trained with few labeled samples to improve the diagnostic performance of the proposed model.
The effectiveness of the proposed method is validated using a self-built reducer test rig. The results show that accuracy of the proposed method reaches 98.38% with few labeled samples, showing significant advantages over existing methods.
The research results can provide the key technology for fault diagnosis of industrial equipment with few labeled samples, contributing to the advancement of intelligent manufacturing.
| 科 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 |