English

A Tunable Loss Function for Binary Classification

Machine Learning 2019-03-21 v2 Information Theory math.IT Machine Learning

Abstract

We present α\alpha-loss, α[1,]\alpha \in [1,\infty], a tunable loss function for binary classification that bridges log-loss (α=1\alpha=1) and 00-11 loss (α=\alpha = \infty). We prove that α\alpha-loss has an equivalent margin-based form and is classification-calibrated, two desirable properties for a good surrogate loss function for the ideal yet intractable 00-11 loss. For logistic regression-based classification, we provide an upper bound on the difference between the empirical and expected risk at the empirical risk minimizers for α\alpha-loss by exploiting its Lipschitzianity along with recent results on the landscape features of empirical risk functions. Finally, we show that α\alpha-loss with α=2\alpha = 2 performs better than log-loss on MNIST for logistic regression.

Cite

@article{arxiv.1902.04639,
  title  = {A Tunable Loss Function for Binary Classification},
  author = {Tyler Sypherd and Mario Diaz and Lalitha Sankar and Peter Kairouz},
  journal= {arXiv preprint arXiv:1902.04639},
  year   = {2019}
}

Comments

9 pages, 1 figure, ISIT 2019