TY - JOUR
T1 - Generalization classification regularization generative adversarial network for machinery fault diagnostics under data imbalance
AU - Lin, Cuiying
AU - Kong, Yun
AU - Huang, Guoyu
AU - Han, Qinkai
AU - Dong, Mingming
AU - Liu, Hui
AU - Chu, Fulei
N1 - Publisher Copyright:
© 2024 Elsevier Ltd
PY - 2025/4
Y1 - 2025/4
N2 - Recently, generative adversarial networks (GANs) have emerged as powerful solutions for fault diagnosis under data imbalance. However, the challenges of the poor generalization ability of GANs, difficulty in capturing diverse feature representations and reliance on data preprocessing still limit effective applications of GANs-based fault diagnosis methods. To tackle these challenges, an innovative generalization classification regularization generative adversarial network (GCRGAN) is proposed, aimed at machinery fault diagnostics under data imbalance. Firstly, a new generator consisting of a random sampling module, a generating module and a generalization module is designed to improve the generated sample quality under limited data. Meanwhile, original vibration signals serve as the generator input to capture diverse feature representation and accelerate the model convergence. Subsequently, an innovative regularization loss is incorporated into the discriminator loss function, which can improve the generalization and stabilize the learning process under limited data. Then, a classifier module is developed to enforce the generator to generate high-quality samples. Finally, case studies of machinery fault diagnostics under extremely data imbalance have been conducted to demonstrate our proposed approach. Experiment validation results have demonstrated that our GCRGAN approach generates high-quality samples and yields superior diagnostic results under data imbalance, compared with mainstream data augmentation methods.
AB - Recently, generative adversarial networks (GANs) have emerged as powerful solutions for fault diagnosis under data imbalance. However, the challenges of the poor generalization ability of GANs, difficulty in capturing diverse feature representations and reliance on data preprocessing still limit effective applications of GANs-based fault diagnosis methods. To tackle these challenges, an innovative generalization classification regularization generative adversarial network (GCRGAN) is proposed, aimed at machinery fault diagnostics under data imbalance. Firstly, a new generator consisting of a random sampling module, a generating module and a generalization module is designed to improve the generated sample quality under limited data. Meanwhile, original vibration signals serve as the generator input to capture diverse feature representation and accelerate the model convergence. Subsequently, an innovative regularization loss is incorporated into the discriminator loss function, which can improve the generalization and stabilize the learning process under limited data. Then, a classifier module is developed to enforce the generator to generate high-quality samples. Finally, case studies of machinery fault diagnostics under extremely data imbalance have been conducted to demonstrate our proposed approach. Experiment validation results have demonstrated that our GCRGAN approach generates high-quality samples and yields superior diagnostic results under data imbalance, compared with mainstream data augmentation methods.
KW - Data augmentation
KW - Fault diagnosis
KW - Generative adversarial network
KW - Imbalanced data
KW - Limited data
UR - http://www.scopus.com/inward/record.url?scp=85213281197&partnerID=8YFLogxK
U2 - 10.1016/j.ress.2024.110791
DO - 10.1016/j.ress.2024.110791
M3 - Article
AN - SCOPUS:85213281197
SN - 0951-8320
VL - 256
JO - Reliability Engineering and System Safety
JF - Reliability Engineering and System Safety
M1 - 110791
ER -