ICLR2025

Local Steps Speed Up Local GD for Heterogeneous Distributed Logistic Regression

Michael Crawshaw, Blake Woodworth, Mingrui Liu

Abstract

We analyze two variants of Local Gradient Descent applied to distributed logistic regression with heterogeneous, separable data and show convergence at the rate O(1/KR)O(1/KR) for KK local steps and sufficiently large RR communication rounds. In contrast, all existing convergence guarantees for Local GD applied to any problem are at least Ω(1/R)Ω(1/R), meaning they fail to show the benefit of local updates. The key to our improved guarantee is showing progress on the logistic regression objective when using a large stepsize η1/Kη\gg 1/K, whereas prior analysis depends on η1/Kη\leq 1/K.