A Novelty Adversarial Loss for Classifying Unbalanced Anomaly Images

Wenqi Huang,Qunsheng Zeng,Peng Li,Ailing Jia,Jie Sona,Mingli Song
DOI: https://doi.org/10.1109/cisp-bmei56279.2022.9979891
2022-01-01
Abstract:The class imbalance problem has lately witnessed substantial progress attributed to the powerful feature representation capability of deep convolutional neural networks (CNNs). However, most of existing adversarial methods are not proficient in learning the distribution of abnormal samples because their loss function is only designed for learning the distribution of normal samples. In this paper, inspired by the existing adversarial methods, we increase an encoder loss to obtain a discriminative margin for abnormal samples based on existing adversarial loss. We also collect a professional shockproof hammer dataset called SHD. Our method is implemented on the SHD dataset and public CIFAR-10 dataset, which needs a large number of normal samples and only a few abnormal samples. Experimental results demonstrate that our approach is capable of achieving an efficient classification score with a higher area under the curve (AUC) results compared to the existing common methods on the above two datasets.
What problem does this paper attempt to address?