Skip to yearly menu bar Skip to main content


Poster

A Federated Stochastic Multi-level Compositional Minimax Algorithm for Deep AUC Maximization

Xinwen Zhang · Ali Payani · Myungjin Lee · Richard Souvenir · Hongchang Gao


Abstract:

AUC maximization is an effective approach to address the imbalanced data classification problem in federated learning. In the past few years, a couple of federated AUC maximization approaches have been developed based on the minimax optimization. However, directly solving a minimax optimization problem to maximize the AUC score cannot achieve satisfactory performance. To address this issue, we propose to maximize AUC via optimizing a federated multi-level compositional minimax problem. In particular, we develop a novel federated multi-level compositional minimax algorithm with rigorous theoretical guarantees to solve this new learning paradigm in both algorithmic design and theoretical analysis. To the best of our knowledge, this is the first work studying the multi-level minimax optimization problem. Additionally, the extensive empirical evaluation confirms the efficacy of our proposed approach.

Live content is unavailable. Log in and register to view live content