Generalization Analysis of Biased Stochastic Gradient Methods for Minimax Problems
Minimax optimization underlies adversarial training, GANs, and AUC maximization, yet biased stochastic gradient descent-ascent methods—zeroth-order and clipped variants—have had no generalization theory. This paper develops the first stability framework for biased SGDA in convex-concave smooth problems, replacing the standard bounded-gradient assumption with a bounded second-moment condition. It derives generalized Lipschitz-type stability bounds and proves that zeroth-order and clipped SGDA can achieve the minimax-optimal $O(1/\sqrt{n})$ excess risk.
The paper makes a timely theoretical contribution by extending algorithmic stability analysis to biased minimax learners. Theorem 2 provides a flexible stability template under generalized Lipschitz-type conditions on gradient estimators and bias (Eq. 4.6–4.8), and the applications recover optimal $O(1/\sqrt{n})$ excess risk for zeroth-order and clipped SGDA while relaxing the uniform Lipschitz assumption to a bounded second-moment condition. The tradeoff between bias and stability is cleanly quantified: "the bias has larger impacts (as compared to gradient estimates) on the stability" (Remark 2, Section 4.2). The generalization gap bound in Theorem 1, "$\triangle^{w}(A_{\mathbf{w}}(S),A_{\mathbf{v}}(S)) - \triangle^{w}_{S}(A_{\mathbf{w}}(S),A_{\mathbf{v}}(S)) \leq L\epsilon^{2}/2 + \sqrt{2\mathbb{E}[G_{S,A}^{2}]}\epsilon$", is principled but leaves an algorithm-dependent $G_{S,A}$ term to be controlled case by case.
A standout element is the surrogate-loss construction for zeroth-order SGDA (Lemma 4, Section 5.2). By defining $\tilde{f}_{1},\tilde{f}_{2}$ as smoothed versions of $f$, the zeroth-order gradient becomes unbiased for the surrogate, allowing the bias term $\bar{B}$ to be set to zero in the general bound. The authors note: "the second statement shows that they are unbiased gradient estimates of the surrogate functions. Then, the bias defined in Eq. (4.5) is 0. This shows the benefit of surrogate functions in the stability analyses." (Section 5.2). Additionally, for standard SGDA the framework strictly improves on prior work: compared to Lei et al. [30], "we relax the bounded gradient assumption to the bounded second moment assumption (Assumption 2)" (Remark 3, Section 5.1), and the resulting stability bound matches the existing rate under a milder hypothesis.
The analysis is limited to convex-concave problems, whereas the motivating settings—GANs, adversarial training, and black-box robustness—are typically nonconvex. The authors concede: "As we only consider the convex-concave problem, it is interesting to extend our analysis to strongly-convex-strongly-concave ... and nonconvex-concave problems" (Section 6). A more specific concern is the Clipped-SGDA regime: to achieve optimal rates the clipping thresholds must scale as $\tau_{1},\tau_{2} \asymp L^{1/2}GT^{1/2}$ (Remark 9, Section 5.3), which grow with the number of iterations and can render clipping inactive for typical gradient norms bounded by $G$. Furthermore, the generalization gap in Theorem 1 depends on $\sqrt{\mathbb{E}[G_{S,A}^{2}]}$, a data- and algorithm-dependent constant that is interpretable for the studied methods but less transparent than a universal Lipschitz bound for general readers.
Within the convex-concave setting, the theoretical evidence supports the claims well. The bounds for zeroth-order SGDA compare favorably against prior minimization-focused results (Table 2), removing dimension-dependent requirements on the smoothing parameter $\mu$ while preserving the $O(1/\sqrt{n})$ rate. The authors fairly state that "In comparison, we relax the Lipschitzness assumption in Lei et al. [30] to the bounded second moment assumption" (Remark 3, Section 5.1). The empirical evaluation is limited but consistent: on AUC maximization with small LIBSVM datasets, stability paths of all three algorithms are "comparable" and the generalization errors align qualitatively with theory (Appendix A). However, the experiments are a sanity check on narrow linear scoring models rather than a broad empirical study.
Reproducibility is partially satisfied. The authors provide a GitHub repository link and specify public LIBSVM datasets, hyperparameters $\eta \in \{0.01,0.03,0.05\}$, and quantities such as $\mu_{1}=\mu_{2}=0.0001$ and $K=d$ for zeroth-order SGDA and chosen $\tau$ for Clipped-SGDA (Appendix A). They define the neighboring dataset by replacing the last example and measure $\Delta_{t}$ exactly as in the theoretical definition. However, the paper omits runtime, hardware, and seeding details, and verifying the generalized Lipschitz conditions (Eq. 4.6–4.8) for any new algorithm requires tedious case-by-case derivations of constants $A,\bar{A},B,\bar{B},\widetilde{G}$, which limits plug-and-play use of the framework.
Minimax optimization refers to a class of optimization problems with some variables to minimize and other variables to maximize. Biased stochastic gradient methods have shown practical success in solving minimax problems to either improve robustness, enhance communication efficiency, or decrease computational costs. These successes motivate a lot of theoretical works to study the convergence of biased stochastic gradient methods, while their generalization analysis remains untouched. In this paper, we present the first framework to study the stability and generalization of biased stochastic gradient methods for minimax problems. We establish a connection between stability and generalization for minimax problems by relaxing the existing bounded gradient assumption to a bounded second moment condition. We then introduce a generalized Lipschitz-type condition on bias and gradient estimators, and derive a general stability bound to clarify the connection among bias, gradient estimators, and stability. We apply our general analysis to Zeroth-order and Clipped stochastic gradient descent ascent (SGDA), and derive stability bounds that match those of SGDA under appropriate smoothing/clipping parameters. We combine stability and convergence analyses together, and derive optimal excess risk bounds of order $1/\sqrt{n}$, where $n$ is the sample size.
Pick a starting point or write your own. Challenges run in the background, so you can keep reading while the AI investigates.
No challenges yet. Disagree with the review? Ask the AI to revisit a specific claim.