CGAN-MBL for Reliability Assessment With Imbalanced Transmission Gear Data
Document Type
Article
Date of Original Version
1-1-2018
Abstract
In term of data-driven techniques for reliability assessment of transmission gears, it is necessary to collect sufficient data of each class in practice. However, the fact that highest and lowest reliability events rarely occur would result in a class imbalance problem. To resolve such an issue, we propose conditional generative adversarial network-mean-covariance balancing labeling (CGAN-MBL) to improve reliability assessment of transmission gears with insufficient and imbalanced data. First, we establish a CGAN-based model to generate credible instances with a similar distribution of original data. In the model, it is proven that by introducing a network weight initialization scheme, the ability to escape from local optimum is enhanced and global search ability of CGAN is strengthened; minibatch discriminator is designed to decrease mode collapse by distinguishing a batch of original and generated data iteratively. Second, in order to overcome the drawback of CGAN with unlabeled generated data, MBL is introduced to tag the generated data through exploring the nearest distance between generated instances and original class centers. Finally, synthetic data combined generated data with original ones train classifiers. Simulations show the effectiveness of the proposed approach for imbalance learning in real-world transmission gear data.
Publication Title, e.g., Journal
IEEE Transactions on Instrumentation and Measurement
Citation/Publisher Attribution
Li, Jie, Haibo He, and Lusi Li. "CGAN-MBL for Reliability Assessment With Imbalanced Transmission Gear Data." IEEE Transactions on Instrumentation and Measurement (2018). doi: 10.1109/TIM.2018.2876777.