Adaptive variational sampling-embedded domain generalization network for fault diagnosis with intra-inter-domain class imbalance

Xiao Zhang, Weiguo Huang, Jun Wang, Zhongkui Zhu, Changqing Shen, Kai Chen, Xingli Zhong and Li He

Reliability Engineering and System Safety, 2025, vol. 256, issue C

Abstract: Domain generalization (DG)-based methods have manifested promising potential in fault diagnosis when dealing with unseen target domain. However, these methods hold the class balance assumptions within and across domains, which contradict actual industrial scenarios, where the varying difficulty of collecting fault data from various classes under the same working condition leads to intra-domain class imbalance. Moreover, changing working conditions may alter the probability of occurrence of different faults, leading to inter-domain class imbalance. For the above concerns, we propose an adaptive variational sampling-embedded domain generalization network (AVS-DGNet) for fault diagnosis with intra-inter-domain class imbalance. In this method, we first apply domain adversarial strategy to map source data to a shared feature space. Sharpness-aware minimization (SAM) is leveraged to alleviate the over-fitting problem owing to the limited data scale of minority classes. Subsequently, a generator is constructed to estimate shared latent variable of each minority class regardless of source domains, and diverse feature generation is realized by sampling from them to mitigate class imbalance. Furthermore, considering different class-wise feature generation, we design adaptive reconstruction weights (ARWs) to control the generation distribution ranges. The experimental results on three datasets verify the effectiveness and superiority of the proposed AVS-DGNet.

Keywords: Fault diagnosis; Domain generalization; Class imbalance; Adversarial training; Variational sampling (search for similar items in EconPapers)
Date: 2025
References: Add references at CitEc
Citations:

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0951832024007786
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:256:y:2025:i:c:s0951832024007786

DOI: 10.1016/j.ress.2024.110707

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 ().