To address the problem that it is difficult to label variable working condition gearbox fault samples and the significant data distribution discrepancies in practical engineering, which result in reduced accuracy of fault diagnosis models,a semi-supervised gearbox fault diagnosis method based on masked contrastive learning is proposed. Firstly, a random mask was used to hide part of the information in the unlabeled dataset, generating two different masked instances for each unlabeled sample. Secondly, a dynamic convolutional neural network was employed to dynamically weight and aggregate the masked instances, enabling discriminative feature modeling of different masked instances. Then, a contrastive learning framework was constructed with the optimization goal of maximizing the similarity between features of different masked instances. By enhancing the consistency of feature representations of masked instance pairs, the model's dependency on labels was reduced. Finally, during the fine-tuning phase, a domain-conditioned feature correction strategy was introduced to generate target domain feature corrections. By aligning source domain features and target domain corrected features according to the metric of minimizing domain feature distribution discrepancies, the method explicitly reduces the domain distribution differences caused by varying working conditions. Validation on a variable working condition gearbox fault dataset demonstrates the effectiveness of the proposed method.
| 科 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 |