Efficient Optimization of F-Measure with Cost-Sensitive SVM

Abstract EN

F -measure is one of the most commonly used performance metrics in classification, particularly when the classes are highly imbalanced.

Direct optimization of this measure is often challenging, since no closed form solution exists.

Current algorithms design the classifiers by using the approximations to the F-measure.

These algorithms are not efficient and do not scale well to the large datasets.

To fill the gap, in this paper, we propose a novel algorithm, which can efficiently optimize F-measure with cost-sensitive SVM.

First of all, we present an explicit transformation from the optimization of F-measure to cost-sensitive SVM.

Then we adopt bundle method to solve the inner optimization.

For the problem where the existing bundle method may have the fluctuations in the primal objective during iterations, an additional line search procedure is involved, which can alleviate the fluctuations problem and make our algorithm more efficient.

Empirical studies on the large-scale datasets demonstrate that our algorithm can provide significant speedups over current state-of-the-art F-measure based learners, while obtaining better (or comparable) precise solutions.