中文

Xtreme Margin:一种用于二分类问题的可调损失函数

机器学习 2022-11-02 v1 人工智能 计算机视觉与模式识别

摘要

损失函数驱动机器学习算法的优化。损失函数的选择会对模型的训练以及模型学习数据的方式产生显著影响。二分类是机器学习问题的主要支柱之一,应用于从医学影像到故障检测等诸多领域。二分类最常用的替代损失函数包括二元交叉熵和铰链损失函数,它们构成了我们研究的重点。在本文中,我们概述了一种新颖的损失函数——Xtreme Margin 损失函数。与二元交叉熵和铰链损失函数不同,该损失函数通过可调超参数 λ1\lambda_1λ2\lambda_2 为研究者和从业者提供了训练过程的灵活性,从最大化精确率和 AUC 分数到针对特定类别最大化条件准确率,即改变它们的值将改变模型的训练。

关键词

引用

@article{arxiv.2211.00176,
  title  = {Xtreme Margin: A Tunable Loss Function for Binary Classification Problems},
  author = {Rayan Wali},
  journal= {arXiv preprint arXiv:2211.00176},
  year   = {2022}
}

备注

10 pages