Primal-dual forward-reflected splitting with linesearch for distributed min-max problems under local Lipschitz smoothness

Abstract

We consider a network of finitely many agents aiming to cooperatively solve a convex-concave min-max problem combining nonsmooth terms with a smooth saddle function whose partial gradients are only locally Lipschitz continuous. Without requiring prior knowledge of the local Lipschitz moduli of the partial gradients, we propose a distributed proximal-reflected-gradient algorithm which incorporates a distributed backtracking procedure to adjust the stepsizes at each iteration. The algorithm is derived by reformulating the min-max problem as a composite monotone inclusion involving the sum of a maximally monotone operator, a linearly composed maximally monotone operator, and a single-valued locally Lipschitz continuous monotone operator. To solve this inclusion, we introduce a novel primal-dual forward-reflected splitting method with linesearch and establish its convergence using a Lyapunov-based analysis. In addition, when specialised to the min-max formulation, we derive an $\mathcal{O}(1/N)$ ergodic convergence rate for a consensus-penalised gap function. Finally, applying this splitting framework with mixing matrices yields our distributed method for consensus min-max optimisation.