#Paper Reading# AUCµ: A Performance Metric for Multi-Class Machine Learning Models

论文题目: AUCµ: A Performance Metric for Multi-Class Machine Learning Models
论文地址: http://proceedings.mlr.press/v97/kleiman19a.html
论文发表于: PLMR 2019

论文大体内容:
本文主要提出了在多分类下的AUC计算方法——AUCµ,具有与AUC同样的特性。

Motivation:
二分类问题往往会使用AUC进行eval,而多分类上往往会使用softmax。本文提出了AUCµ是为了能够在多分类任务上也能使用AUC方法。

Contribution:
在多分类上也可以使用类似AUC的指标去衡量;


1. AUCµ主要计算公式如下:




2. 代码解读[1]。
①代码中的S是论文中去掉系数(1/ni/nj)的部分;
②class_i_j_indices存储的是class为i和j的全部元素;
③dist存储加weight后的分数;
④sort dist;
⑤O(n)计算auc;
所以复杂度是 按class排序O(nlogn) + 计算S(i, j)部分O(K(n+nlog(2n/K)))) + 累加O(K^2) = O(Knlogn)


参考资料: 
[1] 实现代码:https://github.com/microsoft/LightGBM/pull/2567/files


以上均为个人见解,因本人水平有限,如发现有所错漏,敬请指出,谢谢!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值