Multi-scale style generative and adversarial contrastive networks for single domain generalization fault diagnosis

Jun Wang, He Ren, Changqing Shen, Weiguo Huang and Zhongkui Zhu

Reliability Engineering and System Safety, 2024, vol. 243, issue C

Abstract: Domain generalization methods can effectively identify machinery faults under unseen new target working conditions. Nevertheless, most of them rely on data from multiple source domains that are available for model training. However, it is laborious difficult to collect complete monitoring data of machinery under multiple working conditions. Confronting the scenario that only one working condition is available, this paper proposes a novel single domain generalization model, termed multi-scale style generative and adversarial contrastive networks (MSG-ACN), which learns diagnosis knowledge from the single working condition and generalizes it to new working conditions. The main idea of the MSG-ACN model is to generate diverse samples in an extended domain via a domain generation module, and extract domain-invariant features from the source and extended domains via a diagnosis task module. A multi-scale style generation strategy is established to ensure that the generated samples contain abundant state information with the aids of multi-scale convolutional kernels and style learning. Furthermore, an adversarial contrastive learning strategy is designed to promote the learning of class-wise domain-invariant representations while maintaining the diversity of the generated samples. Extensive generalization diagnosis experiments on two datasets verify the superiority of the proposed method over the state-of-the-art fault diagnosis methods.

Keywords: Single domain generalization; Fault diagnosis; Domain-invariant representation; Contrastive learning; Style learning; Adversarial training (search for similar items in EconPapers)
Date: 2024
References: View references in EconPapers View complete reference list from CitEc
Citations: View citations in EconPapers (9)

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0951832023007937
Full text for ScienceDirect subscribers only

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:eee:reensy:v:243:y:2024:i:c:s0951832023007937

DOI: 10.1016/j.ress.2023.109879

Access Statistics for this article

Reliability Engineering and System Safety is currently edited by Carlos Guedes Soares

More articles in Reliability Engineering and System Safety from Elsevier
Bibliographic data for series maintained by Catherine Liu ().