Machine fault diagnosis with small sample based on variational information constrained generative adversarial network |
| |
Affiliation: | 1. School of Civil Aviation, Northwestern Polytechnical University, 710072 Xi’an, China;2. COMAC Flight Test Center, 201207 Shanghai, China |
| |
Abstract: | In actual engineering scenarios, limited fault data leads to insufficient model training and over-fitting, which negatively affects the diagnostic performance of intelligent diagnostic models. To solve the problem, this paper proposes a variational information constrained generative adversarial network (VICGAN) for effective machine fault diagnosis. Firstly, by incorporating the encoder into the discriminator to map the deep features, an improved generative adversarial network with stronger data synthesis capability is established. Secondly, to promote the stable training of the model and guarantee better convergence, a variational information constraint technique is utilized, which constrains the input signals and deep features of the discriminator using the information bottleneck method. In addition, a representation matching module is added to impose restrictions on the generator, avoiding the mode collapse problem and boosting the sample diversity. Two rolling bearing datasets are utilized to verify the effectiveness and stability of the presented network, which demonstrates that the presented network has an admirable ability in processing fault diagnosis with few samples, and performs better than state-of-the-art approaches. |
| |
Keywords: | Fault diagnosis Generative adversarial network Rolling bearing Variational information constraint Small sample |
本文献已被 ScienceDirect 等数据库收录! |
|