A Structural SVM Based Approach for Binary Classification under Class Imbalance

Abstract EN

Class imbalance situations, where one class is rare compared to the other, arise frequently in machine learning applications.

It is well known that the usual misclassification error is not suitable in such settings.

A wide range of performance measures such as AM and QM have been proposed for this problem.

However, due to computational difficulties, few learning techniques have been developed to directly optimize for AM or QM metric.

To fill the gap, in this paper, we present a general structural SVM framework for directly optimizing AM and QM.

We define the loss functions oriented to AM and QM, respectively, and adopt the cutting plane algorithm to solve the outer optimization.

For the inner problem of finding the most violated constraint, we propose two efficient algorithms for the AM and QM problem.

Empirical studies on the various imbalanced datasets justify the effectiveness of the proposed approach.