AUC Maximization under Positive Distribution Shift
Atsutoshi Kumagai, Tomoharu Iwata, Hiroshi Takahashi, Taishi Nishiyama, Yasuhiro Fujiwara
Abstract
Maximizing the area under the receiver operating characteristic curve (AUC) is a common approach to imbalanced binary classification problems. Existing AUC maximization methods usually assume that training and test distributions are identical. However, this assumption is often violated in practice due to a positive distribution shift , where the negative-conditional density does not change but the positive-conditional density can vary. This shift often occurs in imbalanced classifi-cation since positive data are often more diverse or time-varying than negative data. To deal with this shift, we theoretically show that the AUC on the test distribution can be expressed by using the positive and marginal training densities and the marginal test density. Based on this result, we can maximize the AUC on the test distribution by using positive and unlabeled data in the training distribution and unlabeled data in the test distribution. The proposed method requires only positive labels in the training distribution as supervision. Moreover, the derived AUC has a simple form and thus is easy to implement. The effectiveness of the proposed method is experimentally shown with six real-world datasets.
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.
Cited by top-tier papers2
- Positive-unlabeled AUC Maximization under Covariate ShiftAtsutoshi Kumagai, Tomoharu Iwata, Hiroshi Takahashi, Taishi Nishiyama et al.ICML 2025
- Preserving AUC Fairness in Learning with Noisy Protected GroupsMingyang Wu, Li Lin, Wenbin Zhang, Xin Wang et al.ICML 2025
Builds on9
- Kitsune: An Ensemble of Autoencoders for Online Network Intrusion DetectionYisroel Mirsky, Tomer Doitshman, Yuval Elovici, Asaf ShabtaiNDSS 2018 · 945 citations
- Large-scale Robust Deep AUC Maximization: A New Surrogate Loss and Empirical Studies on Medical Image ClassificationZhuoning Yuan, Yan Yan, Milan Sonka, Tianbao YangICCV 2021 · 147 citations
- Stochastic AUC Maximization with Deep Neural NetworksMingrui Liu, Zhuoning Yuan, Yiming Ying, Tianbao YangICLR 2020 · 118 citations
- Mixture Proportion Estimation and PU Learning: A Modern ApproachSaurabh Garg, Yifan Wu, Alexander J. Smola, Sivaraman Balakrishnan et al.NeurIPS 2021 · 79 citations
- Learning from Positive and Unlabeled Data with Arbitrary Positive ShiftZayd Hammoudeh, Daniel LowdNeurIPS 2020 · 53 citations
Related papers
- Semi-Supervised Learning for Maximizing the Partial AUCTomoharu Iwata, Akinori Fujino, Naonori UedaAAAI 2020 · 11 citations
- DRAUC: An Instance-wise Distributionally Robust AUC Optimization FrameworkSiran Dai, Qianqian Xu, Zhiyong Yang, Xiaochun Cao et al.NeurIPS 2023 · 5 citations
- Posterior Re-calibration for Imbalanced DatasetsJunjiao Tian, Yen-Cheng Liu, Nathaniel Glaser, Yen-Chang Hsu et al.NeurIPS 2020 · 86 citations
- Federated Compositional Deep AUC MaximizationXinwen Zhang, Yihan Zhang, Tianbao Yang, Richard Souvenir et al.NeurIPS 2023 · 17 citations
- Estimating the Arc Length of the Optimal ROC Curve and Lower Bounding the Maximal AUCSong LiuNeurIPS 2022
