Generative adversarial networks driven by multi-domain information for improving the quality of generated samples in fault diagnosis.

The performance of intelligent fault diagnosis models is often hindered by the lack of available samples, a common issue in both the few-shot learning and imbalanced learning problems. While data generation has been shown to be an effective strategy for addressing this issue, existing methods tend to focus solely on the similarity of the generated samples, overlooking their diversity. In this regard, a self-reasoning training strategy that enables the participation of highly reliable generated samples in the training process is proposed. By augmenting the training samples, the strategy provides the generation model with diverse input information, effectively enhancing the diversity of the samples generated by the model. To assess the reliability of the generated samples, Pearson correlation coefficient between the generated and real samples in the amplitude–frequency domain is utilized. To this end, an amplitude–frequency domain generation model is constructed. In order to avoid issues such as gradient disappearance and convergence degradation during the generation process in the amplitude–frequency domain, a phase-frequency domain generation model as well as a time domain discriminator model are developed. While the phase-frequency domain generation model enhances the diversity of the generated samples in that domain, the time domain discriminator model ensures sample similarity by correlating the amplitude–frequency domain generation model with the phase-frequency domain generation model. Through experiments, it is demonstrated that the proposed method can effectively address the overfitting problem of generative adversarial networks with few samples, improving the diversity of the generated samples. Moreover, the improvement in fault diagnosis performance achieved by the samples generated by the proposed method further underscores its potential and superiority in practical applications. • A self-reasoning training strategy is proposed. • A generation model driven by multi-domain information is developed. • The quality (similarity and diversity) of the generated samples is enhanced. [ABSTRACT FROM AUTHOR]