Towards imbalanced image classification : a generative adversarial network ensemble learning method

• Main Authors: Huang, Yangru, Jin, Yi, Li, Yidong, Lin, Zhiping
• Other Authors: School of Electrical and Electronic Engineering
• Format: Article
• Language: English
• Published: 2020
• Subjects: Engineering::Electrical and electronic engineering; GAN; Data Cleaning
• Online Access: https://hdl.handle.net/10356/145610
• Institution: Nanyang Technological University

Learning from minority class has been a significant and challenging task which has many potential applications. Weather classification is such a case of imbalanced label distribution. This is because in places like Beijing, some types of weather, such as rain and snow, are relatively rare compared to sunny and haze days. Existing methods are primary to classify the weather conditions relying on expensive sensors or human assistance, which however usually are expensive and time-consuming. In this paper, we propose a new ensemble framework based on the advanced generative adversarial network and an effective data cleaning way to address the class imbalance problem for weather classification. The proposed method not only generates new and reliable samples for the minority class to restore balance, but also filters those generated samples which are unreliable. Experiments show that our approach outperforms the state-of-the-art methods by a huge margin for imbalanced weather classification on several benchmark data sets.