一种用于鲁棒分类的可调损失函数:校准、优化地形与泛化
机器学习
2022-12-22 v6 机器学习
摘要
我们引入了一种称为 -loss 的可调损失函数,由 参数化,在分类的机器学习设定下,它插值于指数损失()、对数损失()与 0-1 损失()之间。在理论上,我们阐明了 -loss 与 Arimoto 条件熵之间的基本联系,验证了 -loss 的分类校准性以通过 Rademacher 复杂度泛化技术证明渐近最优性,并基于一种称为严格局部拟凸性的概念来定量刻画 -loss 的优化地形。在实践中,我们使用卷积神经网络在基准图像数据集上进行了类别不平衡、鲁棒性与分类实验。我们的主要实践结论是,某些任务可能受益于将 -loss 从对数损失()处调离,为此我们为实践者提供了简单的启发式方法。特别地,调节 超参数可轻易地提供对标签翻转的更优模型鲁棒性()以及对不平衡类别的敏感性()。
引用
@article{arxiv.1906.02314,
title = {A Tunable Loss Function for Robust Classification: Calibration, Landscape, and Generalization},
author = {Tyler Sypherd and Mario Diaz and John Kevin Cava and Gautam Dasarathy and Peter Kairouz and Lalitha Sankar},
journal= {arXiv preprint arXiv:1906.02314},
year = {2022}
}
备注
Published at the Transactions on Information Theory