Stochastic AUC Maximization with Deep Neural Networks
Not on the map. Why?
Abstract (arXiv)
Stochastic AUC maximization has garnered an increasing interest due to better fit to imbalanced data classification. However, existing works are limited to stochastic AUC maximization with a linear predictive model, which restricts its predictive power when dealing with extremely complex data. In this paper, we consider stochastic AUC maximization problem with a deep neural network as the predictive model. Building on the saddle point reformulation of a surrogated loss of AUC, the problem can be cast into a it non-convex concave min-max problem. The main contribution made in this paper is to make stochastic AUC maximization more practical for deep neural networks and big data with theoretical insights as well. In particular, we propose to explore Polyak-Lojasiewicz (PL) condition that has been proved and observed in deep learning, which enables us to develop new stochastic algorithms with even faster convergence rate and more practical step size scheme. An AdaGrad-style algorithm is also analyzed under the PL condition with adaptive convergence rate. Our experimental results demonstrate the effectiveness of the proposed algorithms.
Key facts. Cited by 61 papers in the map (data through 13 Oct 2026). First posted on arXiv 28 Aug 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 12; 2022: 21; 2023: 16; 2024: 6; 2025: 5; 2026: 1.
61 papers.
Sort:
By area:
- 2022AUC Maximization in the Era of Big Data and AI: A Surveycited by 100
- 2021Stochastic Optimization of Areas Under Precision-Recall Curves with Provable Convergencecited by 35
- 2021Learning with Multiclass AUC: Theory and Algorithmscited by 35
- 2021Randomized Stochastic Variance-Reduced Methods for Multi-Task Stochastic Bilevel Optimizationcited by 33
- 2022OpenAUC: Towards AUC-Oriented Open-Set Recognitioncited by 33
- 2022When AUC meets DRO: Optimizing Partial AUC for Deep Learning with Non-Convex Convergence Guaranteecited by 29
- 2022Differentially Private SGDA for Minimax Problemscited by 23
- 2021Federated Deep AUC Maximization for Heterogeneous Data with a Constant Communication Complexitycited by 21
- 2021Simple Stochastic and Online Gradient Descent Algorithms for Pairwise Learningcited by 20
- 2022Near-Optimal Algorithms for Making the Gradient Small in Stochastic Minimax Optimizationcited by 20
- 2024AUCSeg: AUC-oriented Pixel-level Long-tail Semantic Segmentationcited by 18
- 2021Near Optimal Stochastic Algorithms for Finite-Sum Unbalanced Convex-Concave Minimax Optimizationcited by 16
38 more
- 2022Minimax AUC Fairness: Efficient Algorithm with Provable Convergencecited by 16
- 2022Optimizing Two-way Partial AUC with an End-to-end Frameworkcited by 15
- 2023LibAUC: A Deep Learning Library for X-Risk Optimizationcited by 15
- 2022Stability and Generalization for Markov Chain Stochastic Gradient Methodscited by 14
- 2021Unified Convergence Analysis for Adaptive Optimization with Moving Average Estimatorcited by 13
- 2025OpenworldAUC: Towards Unified Evaluation and Optimization for Open-world Prompt Tuningcited by 12
- 2022Multi-block Min-max Bilevel Optimization with Applications in Multi-task Deep AUC Maximizationcited by 11
- 2022Decentralized Stochastic Gradient Descent Ascent for Finite-Sum Minimax Problemscited by 11
- 2023Weakly Supervised AUC Optimization: A Unified Partial AUC Approachcited by 11
- 2023Solving a Class of Non-Convex Minimax Optimization in Federated Learningcited by 10
- 2024Fundamental Benefit of Alternating Updates in Minimax Optimizationcited by 9
- 2024An Accelerated Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothnesscited by 9
- 2022Algorithmic Foundations of Empirical X-risk Minimizationcited by 8
- 2022INTERACT: Achieving Low Sample and Communication Complexities in Decentralized Bilevel Learning over Networkscited by 8
- 2022SGDA with shuffling: faster convergence for nonconvex-PL minimax optimizationcited by 8
- 2023Federated Compositional Deep AUC Maximizationcited by 8
- 2021Manifold Learning for Knowledge Discovery and Intelligent Inverse Design of Photonic Nanostructures: Breaking the Geometric Complexitycited by 6
- 2021Deep AUC Maximization for Medical Image Classification: Challenges and Opportunitiescited by 6
- 2022An Efficient Stochastic Algorithm for Decentralized Nonconvex-Strongly-Concave Minimax Optimizationcited by 6
- 2022AUC Maximization for Low-Resource Named Entity Recognitioncited by 6
- 2022Asymptotically Unbiased Instance-wise Regularized Partial AUC Optimization: Theory and Algorithmcited by 5
- 2023Faster Stochastic Algorithms for Minimax Optimization under Polyak--Lojasiewicz Conditionscited by 5
- 2023Stochastic Smoothed Gradient Descent Ascent for Federated Minimax Optimizationcited by 4
- 2024Single-Loop Stochastic Algorithms for Difference of Max-Structured Weakly Convex Functionscited by 4
- 2022AdAUC: End-to-end Adversarial AUC Optimization Against Long-tail Problemscited by 3
- 2023Efficient Fraud Detection Using Deep Boosting Decision Treescited by 3
- 2023Provable Multi-instance Deep AUC Maximization with Stochastic Poolingcited by 3
- 2023Serverless Federated AUPRC Optimization for Multi-Party Collaborative Imbalanced Data Miningcited by 3
- 2023DRAUC: An Instance-wise Distributionally Robust AUC Optimization Frameworkcited by 3
- 2021CDMA: A Practical Cross-Device Federated Learning Algorithm for General Minimax Problemscited by 2
- 2022A Penalty-Based Method for Communication-Efficient Decentralized Bilevel Programmingcited by 2
- 2023Enhancing Personalized Ranking With Differentiable Group AUC Optimizationcited by 2
- 2024Accelerated Stochastic Min-Max Optimization Based on Bias-corrected Momentumcited by 2
- 2025Adaptive Algorithms with Sharp Convergence Rates for Stochastic Hierarchical Optimizationcited by 2
- 2025Federated Stochastic Minimax Optimization under Heavy-Tailed Noisescited by 2
- 2021Eating Garlic Prevents COVID-19 Infection: Detecting Misinformation on the Arabic Content of Twittercited by 1
- 2023A Log-linear Gradient Descent Algorithm for Unbalanced Binary Classification using the All Pairs Squared Hinge Losscited by 1
- 2023Ranking Regularization for Critical Rare Classes: Minimizing False Positives at a High True Positive Ratecited by 1
50 most cited of 61 All 61 (JSON)
- 2026AUC Maximization from Biased Positive-unlabeled Data with Confidence10 Sep 2026
- 2025Closing the Approximation Gap of Partial AUC Optimization: A Tale of Two Formulations1 Dec 2025
- 2025Federated Stochastic Minimax Optimization under Heavy-Tailed Noises6 Nov 2025
- 2025Adaptive Algorithms with Sharp Convergence Rates for Stochastic Hierarchical Optimization18 Sep 2025
- 2025OpenworldAUC: Towards Unified Evaluation and Optimization for Open-world Prompt Tuning8 May 2025
- 2025On the Convergence of Adam-Type Algorithm for Bilevel Optimization under Unbounded Smoothness5 Mar 2025
- 2024Enhancing Topic Interpretability for Neural Topic Modeling through Topic-wise Contrastive Learning23 Dec 2024
- 2024AUCSeg: AUC-oriented Pixel-level Long-tail Semantic Segmentation30 Sep 2024
- 2024An Accelerated Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothness28 Sep 2024
- 2024Accelerated Stochastic Min-Max Optimization Based on Bias-corrected Momentum18 Jun 2024
- 2024Single-Loop Stochastic Algorithms for Difference of Max-Structured Weakly Convex Functions28 May 2024
- 2024Fundamental Benefit of Alternating Updates in Minimax Optimization16 Feb 2024
38 more
- 2023DRAUC: An Instance-wise Distributionally Robust AUC Optimization Framework6 Nov 2023
- 2023Stochastic Smoothed Gradient Descent Ascent for Federated Minimax Optimization2 Nov 2023
- 2023Solving a Class of Non-Convex Minimax Optimization in Federated Learning5 Oct 2023
- 2023Serverless Federated AUPRC Optimization for Multi-Party Collaborative Imbalanced Data Mining6 Aug 2023
- 2023Faster Stochastic Algorithms for Minimax Optimization under Polyak--Lojasiewicz Conditions29 Jul 2023
- 2023LibAUC: A Deep Learning Library for X-Risk Optimization5 Jun 2023
- 2023Does it pay to optimize AUC?2 Jun 2023
- 2023AUC Optimization from Multiple Unlabeled Datasets25 May 2023
- 2023Weakly Supervised AUC Optimization: A Unified Partial AUC Approach23 May 2023
- 2023Provable Multi-instance Deep AUC Maximization with Stochastic Pooling14 May 2023
- 2023Can Decentralized Stochastic Minimax Optimization Algorithms Converge Linearly for Finite-Sum Nonconvex-Nonconcave Problems?24 Apr 2023
- 2023Federated Compositional Deep AUC Maximization20 Apr 2023
- 2023Enhancing Personalized Ranking With Differentiable Group AUC Optimization17 Apr 2023
- 2023Ranking Regularization for Critical Rare Classes: Minimizing False Positives at a High True Positive Rate31 Mar 2023
- 2023A Log-linear Gradient Descent Algorithm for Unbalanced Binary Classification using the All Pairs Squared Hinge Loss21 Feb 2023
- 2023Efficient Fraud Detection Using Deep Boosting Decision Trees12 Feb 2023
- 2022AUC Maximization for Low-Resource Named Entity Recognition9 Dec 2022
- 2022Decentralized Stochastic Gradient Descent Ascent for Finite-Sum Minimax Problems6 Dec 2022
- 2022An Efficient Stochastic Algorithm for Decentralized Nonconvex-Strongly-Concave Minimax Optimization5 Dec 2022
- 2022A Penalty-Based Method for Communication-Efficient Decentralized Bilevel Programming8 Nov 2022
- 2022OpenAUC: Towards AUC-Oriented Open-Set Recognition22 Oct 2022
- 2022Fairness via Adversarial Attribute Neighbourhood Robust Learning12 Oct 2022
- 2022SGDA with shuffling: faster convergence for nonconvex-PL minimax optimization12 Oct 2022
- 2022Asymptotically Unbiased Instance-wise Regularized Partial AUC Optimization: Theory and Algorithm8 Oct 2022
- 2022SAGDA: Achieving O(ε-2) Communication Complexity in Federated Min-Max Learning2 Oct 2022
- 2022Stability and Generalization for Markov Chain Stochastic Gradient Methods16 Sep 2022
- 2022Minimax AUC Fairness: Efficient Algorithm with Provable Convergence22 Aug 2022
- 2022Near-Optimal Algorithms for Making the Gradient Small in Stochastic Minimax Optimization11 Aug 2022
- 2022INTERACT: Achieving Low Sample and Communication Complexities in Decentralized Bilevel Learning over Networks27 Jul 2022
- 2022Balanced Self-Paced Learning for AUC Maximization8 Jul 2022
- 2022AdAUC: End-to-end Adversarial AUC Optimization Against Long-tail Problems24 Jun 2022
- 2022Optimizing Two-way Partial AUC with an End-to-end Framework23 Jun 2022
- 2022Algorithmic Foundations of Empirical X-risk Minimization1 Jun 2022
- 2022Multi-block Min-max Bilevel Optimization with Applications in Multi-task Deep AUC Maximization1 Jun 2022
- 2022AUC Maximization in the Era of Big Data and AI: A Survey28 Mar 2022
- 2022When AUC meets DRO: Optimizing Partial AUC for Deep Learning with Non-Convex Convergence Guarantee1 Mar 2022
- 2022Differentially Private SGDA for Minimax Problems22 Jan 2022
- 2021Simple Stochastic and Online Gradient Descent Algorithms for Pairwise Learning23 Nov 2021
50 newest of 61 All 61 (JSON)
- 2022AUC Maximization in the Era of Big Data and AI: A Surveycited by 100
- 2021Stochastic Optimization of Areas Under Precision-Recall Curves with Provable Convergencecited by 35
- 2021Learning with Multiclass AUC: Theory and Algorithmscited by 35
- 2021Randomized Stochastic Variance-Reduced Methods for Multi-Task Stochastic Bilevel Optimizationcited by 33
- 2022OpenAUC: Towards AUC-Oriented Open-Set Recognitioncited by 33
- 2022When AUC meets DRO: Optimizing Partial AUC for Deep Learning with Non-Convex Convergence Guaranteecited by 29
- 2022Differentially Private SGDA for Minimax Problemscited by 23
- 2021Federated Deep AUC Maximization for Heterogeneous Data with a Constant Communication Complexitycited by 21
- 2021Simple Stochastic and Online Gradient Descent Algorithms for Pairwise Learningcited by 20
- 2022Near-Optimal Algorithms for Making the Gradient Small in Stochastic Minimax Optimizationcited by 20
- 2024AUCSeg: AUC-oriented Pixel-level Long-tail Semantic Segmentationcited by 18
- 2021Near Optimal Stochastic Algorithms for Finite-Sum Unbalanced Convex-Concave Minimax Optimizationcited by 16
12 most cited of 60 All 60 (JSON)
- 2026AUC Maximization from Biased Positive-unlabeled Data with Confidence10 Sep 2026
- 2025Closing the Approximation Gap of Partial AUC Optimization: A Tale of Two Formulations1 Dec 2025
- 2025Federated Stochastic Minimax Optimization under Heavy-Tailed Noises6 Nov 2025
- 2025Adaptive Algorithms with Sharp Convergence Rates for Stochastic Hierarchical Optimization18 Sep 2025
- 2025On the Convergence of Adam-Type Algorithm for Bilevel Optimization under Unbounded Smoothness5 Mar 2025
- 2024Enhancing Topic Interpretability for Neural Topic Modeling through Topic-wise Contrastive Learning23 Dec 2024
- 2024AUCSeg: AUC-oriented Pixel-level Long-tail Semantic Segmentation30 Sep 2024
- 2024An Accelerated Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothness28 Sep 2024
- 2024Accelerated Stochastic Min-Max Optimization Based on Bias-corrected Momentum18 Jun 2024
- 2024Single-Loop Stochastic Algorithms for Difference of Max-Structured Weakly Convex Functions28 May 2024
- 2024Fundamental Benefit of Alternating Updates in Minimax Optimization16 Feb 2024
- 2023DRAUC: An Instance-wise Distributionally Robust AUC Optimization Framework6 Nov 2023
12 newest of 60 All 60 (JSON)