Feature augmentation for imbalanced classification with conditional mixture WGANs
作者:
Highlights:
• We propose a framework of data augmentation for addressing imbalanced classification without regards to the modality of data.
• Further, we propose a conditional mixture WGANs (cMWGANs), which can stabilize the training process and address the mode collapse of GANs, thus generating label preserved and diverse features.
• In order to test the augmented ability of the method, we evaluate the method on three uneven benchmark datasets. The empirical results show that our model can achieve significant improvements compared to the model without feature augmentation and other typical models.
摘要
•We propose a framework of data augmentation for addressing imbalanced classification without regards to the modality of data.•Further, we propose a conditional mixture WGANs (cMWGANs), which can stabilize the training process and address the mode collapse of GANs, thus generating label preserved and diverse features.•In order to test the augmented ability of the method, we evaluate the method on three uneven benchmark datasets. The empirical results show that our model can achieve significant improvements compared to the model without feature augmentation and other typical models.
论文关键词:Imbalanced classification,Feature augmentation,Generative adversarial nets,Wasserstein distance
论文评审过程:Received 4 September 2018, Revised 11 January 2019, Accepted 17 March 2019, Available online 26 March 2019, Version of Record 5 April 2019.
论文官网地址:https://doi.org/10.1016/j.image.2019.03.010