A Tunable Loss Function for Binary Classification
A Tunable Loss Function for Binary Classification
复制标题
DOI:
10.1109/isit.2019.8849796
复制
发表时间:
2019-02
期刊:
影响因子:
--
通讯作者:
Tyler Sypherd;Mario Díaz;L. Sankar;P. Kairouz
中科院分区:
文献类型:
--
作者:
Tyler Sypherd;Mario Díaz;L. Sankar;P. Kairouz
We present α-loss, α ∈ [1, ∞], a tunable loss function for binary classification that bridges log-loss (α = 1) and 0-1 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 0-1 loss. For logistic regression-based classification, we provide an upper bound on the difference between the empirical and expected risk for α-loss at the critical points of the empirical risk by exploiting its Lipschitzianity along with recent results on the landscape features of empirical risk functions. Finally, we show that α-loss with α = 2 performs better than log-loss on MNIST for logistic regression.