A Robust Loss Function for Classification with Imbalanced Datasets.

Yidan Wang,Liming Yang
DOI: https://doi.org/10.1016/j.neucom.2018.11.024
IF: 6
2018-01-01
Neurocomputing
Abstract:Based on minimizing misclassification cost, a new robust loss function is designed in this paper to deal with the imbalanced classification problem under noise environment. It is nonconvex but maintains Fisher consistency. Applying the proposed loss function into support vector machine (SVM), a robust SVM framework is presented which results in a Bayes optimal classifier. However, nonconvexity makes the model difficult to optimize. We develop an alternative iterative algorithm to solve the proposed model. What's more, we analyze the robustness of the proposed model theoretically from a re-weighted SVM viewpoint and the obtained optimal solution is consistent with Bayesian optimal decision rule. Further-more, numerical experiments are carried out on databases that are drawn from UCI Machine Learning Repository and a practical application. With two different types of noise environments, one with label noise and one with feature noise, experiment results show that on these two databases the proposed method achieves better generalization results compared to other SVM methods. (C) 2018 Elsevier B.V. All rights reserved.
What problem does this paper attempt to address?