Curvature accelerated decentralized non-convex optimization for high-dimensional machine learning problems (2504.04073v1)

Abstract: We consider distributed optimization as motivated by machine learning in a multi-agent system: each agent holds local data and the goal is to minimize an aggregate loss function over a common model, via an interplay of local training and distributed communication. In the most interesting case of training a neural network, the loss functions are non-convex and the high dimension of the model poses challenges in terms of communication and computation. We propose a primal-dual method that leverages second order information in the local training sub-problems in order to accelerate the algorithm. To ease the computational burden, we invoke a quasi-Newton local solver with linear cost in the model dimension. Besides, our method is communication efficient in the sense of requiring to broadcast the local model only once per round. We rigorously establish the convergence of the algorithm and demonstrate its merits by numerical experiments.