Asymptotically Unbiased Instance-wise Regularized Partial AUC Optimization: Theory and Algorithm
Huiyang Shao, Qianqian Xu, Zhiyong Yang, Shilong Bao, Qingming Huang
Abstract
The Partial Area Under the ROC Curve (PAUC), typically including One-way Partial AUC (OPAUC) and Two-way Partial AUC (TPAUC), measures the average performance of a binary classifier within a specific false positive rate and/or true positive rate interval, which is a widely adopted measure when decision constraints must be considered. Consequently, PAUC optimization has naturally attracted increasing attention in the machine learning community within the last few years. Nonetheless, most of the existing methods could only optimize PAUC approximately, leading to inevitable biases that are not controllable. Fortunately, a recent work presents an unbiased formulation of the PAUC optimization problem via distributional robust optimization. However, it is based on the pair-wise formulation of AUC, which suffers from the limited scalability w.r.t. sample size and a slow convergence rate, especially for TPAUC. To address this issue, we present a simpler reformulation of the problem in an asymptotically unbiased and instance-wise manner. For both OPAUC and TPAUC, we come to a nonconvex strongly concave minimax regularized problem of instance-wise functions. On top of this, we employ an efficient solver enjoys a linear per-iteration computational complexity w.r.t. the sample size and a time-complexity of to reach a stationary point. Furthermore, we find that the minimax reformulation also facilitates the theoretical analysis of generalization error as a byproduct. Compared with the existing results, we present new error bounds that are much easier to prove and could deal with hypotheses with real-valued outputs. Finally, extensive experiments on several benchmark datasets demonstrate the effectiveness of our method.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 539b35e6-0de2-4ed3-888e-33ded8099068Cited by top-tier papers3
- AUCSeg: AUC-oriented Pixel-level Long-tail Semantic SegmentationBoyu Han, Qianqian Xu, Zhiyong Yang, Shilong Bao et al.NeurIPS 2024 · 26 citations
- Lower-Left Partial AUC: An Effective and Efficient Optimization Metric for RecommendationWentao Shi, Chenxu Wang, Fuli Feng, Yang Zhang et al.WWW 2024 · 13 citations
- Weighted ROC Curve in Cost Space: Extending AUC to Cost-Sensitive LearningHuiyang Shao, Qianqian Xu, Zhiyong Yang, Peisong Wen et al.NeurIPS 2023 · 10 citations
Builds on7
- Stochastic AUC Maximization with Deep Neural NetworksMingrui Liu, Zhuoning Yuan, Yiming Ying, Tianbao YangICLR 2020 · 118 citations
- AUC-oriented Graph Neural Network for Fraud DetectionMengda Huang, Yang Liu, Xiang Ao, Kuan Li et al.WWW 2022 · 114 citations
- Communication-Efficient Distributed Stochastic AUC Maximization with Deep Neural NetworksZhishuai Guo, Mingrui Liu, Zhuoning Yuan, Li Shen et al.ICML 2020 · 45 citations
- Compositional Training for End-to-End Deep AUC MaximizationZhuoning Yuan, Zhishuai Guo, Nitesh V. Chawla, Tianbao YangICLR 2022 · 34 citations
- When All We Need is a Piece of the Pie: A Generic Framework for Optimizing Two-way Partial AUCZhiyong Yang, Qianqian Xu, Shilong Bao, Yuan He et al.ICML 2021 · 33 citations
Related papers
- When AUC meets DRO: Optimizing Partial AUC for Deep Learning with Non-Convex Convergence GuaranteeDixian Zhu, Gang Li, Bokun Wang, Xiaodong Wu et al.ICML 2022 · 42 citations
- DRAUC: An Instance-wise Distributionally Robust AUC Optimization FrameworkSiran Dai, Qianqian Xu, Zhiyong Yang, Xiaochun Cao et al.NeurIPS 2023 · 5 citations
- Semi-Supervised Learning for Maximizing the Partial AUCTomoharu Iwata, Akinori Fujino, Naonori UedaAAAI 2020 · 11 citations
- Large-scale Optimization of Partial AUC in a Range of False Positive RatesYao Yao, Qihang Lin, Tianbao YangNeurIPS 2022 · 24 citations
- Minimax AUC Fairness: Efficient Algorithm with Provable ConvergenceZhenhuan Yang, Yan Lok Ko, Kush R. Varshney, Yiming YingAAAI 2023 · 22 citations
