Federated Compositional Deep AUC Maximization
Not on the map. Why?
Abstract (arXiv)
Federated learning has attracted increasing attention due to the promise of balancing privacy and large-scale learning; numerous approaches have been proposed. However, most existing approaches focus on problems with balanced data, and prediction performance is far from satisfactory for many real-world applications where the number of samples in different classes is highly imbalanced. To address this challenging problem, we developed a novel federated learning method for imbalanced data by directly optimizing the area under curve (AUC) score. In particular, we formulate the AUC maximization problem as a federated compositional minimax optimization problem, develop a local stochastic compositional gradient descent ascent with momentum algorithm, and provide bounds on the computational and communication complexities of our algorithm. To the best of our knowledge, this is the first work to achieve such favorable theoretical results. Finally, extensive experimental results confirm the efficacy of our method.
Key facts. Cited by 8 papers in the map (data through 7 Oct 2026). Builds on 25 of its 42 references that are in the map. First posted on arXiv 20 Apr 2023.
Builds on
25 of 42 references.
References · 25
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
- 2014Explaining and Harnessing Adversarial Examplescited by 7,287
- 2016Densely Connected Convolutional Networkscited by 6,691
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
- 2018Federated Optimization in Heterogeneous Networkscited by 2,971
- 2020Tackling the Objective Inconsistency Problem in Heterogeneous Federated Optimizationcited by 897
- 2018Local SGD Converges Fast and Communicates Littlecited by 595
- 2018Parallel Restarted SGD with Faster Convergence and Less Communication: Demystifying Why Model Averaging Works for Deep Learningcited by 277
- 2019On Gradient Descent Ascent for Nonconvex-Concave Minimax Problemscited by 266
- 2019On the Linear Speedup Analysis of Communication Efficient Momentum SGD for Distributed Non-Convex Optimizationcited by 178
- 2020A Patient-Centric Dataset of Images and Metadata for Identifying Melanomas Using Clinical Contextcited by 146
13 more
- 2014Stochastic Compositional Gradient Descent: Algorithms for Minimizing Compositions of Expected-Value Functionscited by 93
- 2019Stochastic AUC Maximization with Deep Neural Networkscited by 61
- 2022FedNest: Federated Bilevel, Minimax, and Compositional Optimizationcited by 60
- 2020A Single-Loop Smoothed Gradient Descent-Ascent Algorithm for Nonconvex-Concave Min-Max Problemscited by 57
- 2021Local Stochastic Gradient Descent Ascent: Convergence Analysis and Communication Efficiencycited by 40
- 2022Federated Minimax Optimization: Improved Convergence Analyses and Algorithmscited by 37
- 2020Communication-Efficient Distributed Stochastic AUC Maximization with Deep Neural Networkscited by 27
- 2020Single-Timescale Stochastic Nonconvex-Concave Optimization for Smooth Nonlinear TD Learningcited by 24
- 2021Federated Deep AUC Maximization for Heterogeneous Data with a Constant Communication Complexitycited by 21
- 2015Experiments on Parallel Training of Deep Neural Network using Model Averagingcited by 14
- 2021Compositional federated learning: Applications in distributionally robust averaging and meta learningcited by 12
- 2022FeDXL: Provable Federated Learning for Deep X-Risk Optimizationcited by 10
- 2019Stochastic Recursive Variance Reduction for Efficient Smooth Non-Convex Compositional Optimizationcited by 6
Built on by
Citing papers per year: 2023: 3; 2024: 1; 2025: 3; 2026: 1.
8 papers.
- 2023Serverless Federated AUPRC Optimization for Multi-Party Collaborative Imbalanced Data Miningcited by 3
- 2024Stochastic Compositional Minimax Optimization with Provable Convergence Guaranteescited by 1
- 2025Stochastic Primal-Dual Double Block-Coordinate for Two-way Partial AUC Maximizationcited by 1
- 2023Federated Multi-Sequence Stochastic Approximation with Local Hypergradient Estimationcited by 0
- 2023Achieving Linear Speedup in Decentralized Stochastic Compositional Minimax Optimizationcited by 0
- 2025Online AUC Optimization Based on Second-order Surrogate Losscited by 0
- 2025Semi-overlapping Multi-bandit Best Arm Identification for Sequential Support Network Learningcited by 0
- 2026Partial AUC Maximization from Positive-unlabeled Datacited by 0
- 2026Partial AUC Maximization from Positive-unlabeled Data25 Sep 2026
- 2025Semi-overlapping Multi-bandit Best Arm Identification for Sequential Support Network Learning31 Dec 2025
- 2025Online AUC Optimization Based on Second-order Surrogate Loss24 Oct 2025
- 2025Stochastic Primal-Dual Double Block-Coordinate for Two-way Partial AUC Maximization28 May 2025
- 2024Stochastic Compositional Minimax Optimization with Provable Convergence Guarantees22 Aug 2024
- 2023Serverless Federated AUPRC Optimization for Multi-Party Collaborative Imbalanced Data Mining6 Aug 2023
- 2023Achieving Linear Speedup in Decentralized Stochastic Compositional Minimax Optimization25 Jul 2023
- 2023Federated Multi-Sequence Stochastic Approximation with Local Hypergradient Estimation2 Jun 2023