A Tunable Loss Function for Binary Classification
Machine Learning
2019-03-21 v2 Information Theory
math.IT
Machine Learning
Abstract
We present -loss, , a tunable loss function for binary classification that bridges log-loss () and - loss (). We prove that -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 - 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 -loss by exploiting its Lipschitzianity along with recent results on the landscape features of empirical risk functions. Finally, we show that -loss with 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