Fault Diagnosis of Power-Shift Systems in Agricultural Continuously Variable Transmissions Using Generative Adversarial Networks

The power-shift system employed in agricultural multi-range continuously variable transmissions (CVTs) features a complex structure and control logic, presenting significant challenges to the reliability of agricultural machinery. To enable timely detection of faults, constructing an intelligent fault diagnosis classifier to monitor the system’s health status is essential. Typically, fault samples utilized for classifier development originate from ideal bench tests, characterized by uniform patterns and limited diversity, thereby hindering the algorithm’s generalization capability. This study addresses this issue by proposing a generative adversarial network (GAN) model, integrated with a triple loss function and a novel generator architecture, to augment the fault dataset under laboratory conditions. The generator architecture comprises a variational autoencoder module and an oil pressure point attention mechanism, enabling the generation of diverse and fluctuating virtual samples. Building on this augmented dataset, a fault classifier based on one-dimensional ConvNeXt was developed. Experimental results indicate that the classifier achieves an accuracy of 99.73%. While classifier accuracy decreases with increasing noise levels, the GAN-generated dataset provides more comprehensive training, resulting in an accuracy approximately 3% higher than that achieved using the original dataset.