optimization

A stochastic smoothing framework for nonconvex-nonconcave minEmax problems with applications to Wasserstein distributionally robust optimization

arXiv:2502.17602

summary

The paper introduces a stochastic smoothing proximal gradient algorithm for solving nonconvex‑nonconcave minimization‑expectation‑maximization (minEmax) problems, providing convergence guarantees and demonstrating its performance on distributionally robust and adversarial learning tasks.

Abstract

We study a class of stochastic nonsmooth optimization problems in which an outer variable minimizes the expectation of a pointwise maximum. This minimization--expectation--maximization (minEmax) problem arises in Wasserstein distributionally robust optimization and adversarially robust training, and it cannot in general be reformulated as a finite-dimensional minimax problem when the underlying distribution is not empirical. We propose a stochastic smoothing proximal gradient method based on log-mean-exp smoothing of the value function. Under compactness and Lipschitz-type assumptions, we present nonasymptotic analysis in terms of Goldstein stationarity and show that every almost-sure cluster point generated by our method is a Clarke stationary point; by Clarke regularity, such a point is also directional stationary for the original problem. Numerical experiments on newsvendor, robust regression, and adversarially robust learning problems show that the proposed method is competitive with existing baselines.

Topics & keywords

#stochastic optimization#nonconvex-nonconcave#minimax#distributionally robust optimization#smoothing methodslog-mean-exp smoothingGoldstein stationarityClarke stationary pointproximal gradientWasserstein DRO
A stochastic smoothing framework for nonconvex-nonconcave minEmax problems with applications to Wasserstein distributionally robust optimization · wovepaper