Learning With Multiclass AUC: Theory and Algorithms
文献类型:期刊论文
作者 | Yang, Zhiyong7; Xu, Qianqian6; Bao, Shilong4,5; Cao, Xiaochun3,5; Huang, Qingming1,2,6 |
刊名 | IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE
![]() |
出版日期 | 2022-11-01 |
卷号 | 44期号:11页码:7747-7763 |
关键词 | AUC optimization machine learning |
ISSN号 | 0162-8828 |
DOI | 10.1109/TPAMI.2021.3101125 |
英文摘要 | The Area under the ROC curve (AUC) is a well-known ranking metric for problems such as imbalanced learning and recommender systems. The vast majority of existing AUC-optimization-based machine learning methods only focus on binary-class cases, while leaving the multiclass cases unconsidered. In this paper, we start an early trial to consider the problem of learning multiclass scoring functions via optimizing multiclass AUC metrics. Our foundation is based on the M metric, which is a well-known multiclass extension of AUC. We first pay a revisit to this metric, showing that it could eliminate the imbalance issue from the minority class pairs. Motivated by this, we propose an empirical surrogate risk minimization framework to approximately optimize the M metric. Theoretically, we show that: (i) optimizing most of the popular differentiable surrogate losses suffices to reach the Bayes optimal scoring function asymptotically; (ii) the training framework enjoys an imbalance-aware generalization error bound, which pays more attention to the bottleneck samples of minority classes compared with the traditional O(root 1/N) THORN result. Practically, to deal with the low scalability of the computational operations, we propose acceleration methods for three popular surrogate loss functions, including the exponential loss, squared loss, and hinge loss, to speed up loss and gradient evaluations. Finally, experimental results on 11 real-world datasets demonstrate the effectiveness of our proposed framework. The code is now available at https://github.com/joshuaas/ Learning-with-Multiclass-AUC-Theory-and-Algorithms. |
资助项目 | National Key R&D Program of China[2018AAA0102003] ; National Natural Science Foundation of China[61620106009] ; National Natural Science Foundation of China[62025604] ; National Natural Science Foundation of China[61861166002] ; National Natural Science Foundation of China[61931008] ; National Natural Science Foundation of China[61836002] ; National Natural Science Foundation of China[61976202] ; Fundamental Research Funds for the Central Universities ; National Postdoctoral Program for Innovative Talents[BX2021298] ; Youth Innovation Promotion Association CAS ; Strategic Priority Research Program of Chinese Academy of Sciences[XDB28000000] |
WOS研究方向 | Computer Science ; Engineering |
语种 | 英语 |
WOS记录号 | WOS:000864325900036 |
出版者 | IEEE COMPUTER SOC |
源URL | [http://119.78.100.204/handle/2XEOYT63/19764] ![]() |
专题 | 中国科学院计算技术研究所期刊论文 |
通讯作者 | Xu, Qianqian; Huang, Qingming |
作者单位 | 1.Peng Cheng Lab, Shenzhen 518055, Peoples R China 2.Univ Chinese Acad Sci, Sch Comp Sci & Technol, Key Lab Big Data Min & Knowledge Management BDKM, Beijing 101408, Peoples R China 3.Sun Yat Sen Univ, Sch Cyber Sci & Technol, Shenzhen 518100, Peoples R China 4.Univ Chinese Acad Sci, Sch Cyber Secur, Beijing 100049, Peoples R China 5.Chinese Acad Sci, Inst Informat Engn, State Key Lab Informat Secur, Beijing 100093, Peoples R China 6.Chinese Acad Sci, Inst Comp Technol, Key Lab Intelligent Informat Proc, Beijing 100190, Peoples R China 7.Univ Chinese Acad Sci, Sch Comp Sci & Technol, Beijing 101408, Peoples R China |
推荐引用方式 GB/T 7714 | Yang, Zhiyong,Xu, Qianqian,Bao, Shilong,et al. Learning With Multiclass AUC: Theory and Algorithms[J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE,2022,44(11):7747-7763. |
APA | Yang, Zhiyong,Xu, Qianqian,Bao, Shilong,Cao, Xiaochun,&Huang, Qingming.(2022).Learning With Multiclass AUC: Theory and Algorithms.IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE,44(11),7747-7763. |
MLA | Yang, Zhiyong,et al."Learning With Multiclass AUC: Theory and Algorithms".IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE 44.11(2022):7747-7763. |
入库方式: OAI收割
来源:计算技术研究所
浏览0
下载0
收藏0
其他版本
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。