Distribution Bias Aware Collaborative Generative Adversarial Network for Imbalanced Deep Learning in Industrial IoT

Xiaokang Zhou, Yiyong Hu, Jiayi Wu, Wei Liang, Jianhua Ma, Qun Jin

Research output: Contribution to journalArticlepeer-review

Abstract

In this study, we propose a DB-CGAN model to deal with distribution bias issues for imbalanced deep learning in industrial IoT. An integrated data augmentation framework is constructed based on introduction of a complementary classifier into the basic GAN model. Specifically, a conditional generator with random labels is designed and trained adversarially with the classifier to effectively enhance the augmentation of number of data samples in minority classes, while a weight sharing scheme is newly devised between two separated feature extractors, enabling the collaborative adversarial training among generator, discriminator, and classifier. An augmentation algorithm is then developed to facilitate intelligent anomaly detection, which can efficiently improve the classification accuracy based on the correction of distribution bias. Experiment evaluations based on two real-world imbalanced datasets demonstrate the outstanding performance of our proposed model in tackling distribution bias issues for imbalanced learning in industrial IoT applications, compared with five baseline methods.

Original languageEnglish
JournalIEEE Transactions on Industrial Informatics
DOIs
Publication statusAccepted/In press - 2022

Keywords

  • Collaboration
  • Collaborative Adversarial Training
  • Data Augmentation
  • Data models
  • Distribution Bias
  • Feature extraction
  • Generative Adversarial Network
  • Generative adversarial networks
  • Generators
  • Imbalanced Learning
  • Industrial Internet of Things
  • Industrial IoT
  • Training

ASJC Scopus subject areas

  • Control and Systems Engineering
  • Information Systems
  • Computer Science Applications
  • Electrical and Electronic Engineering

Fingerprint

Dive into the research topics of 'Distribution Bias Aware Collaborative Generative Adversarial Network for Imbalanced Deep Learning in Industrial IoT'. Together they form a unique fingerprint.

Cite this