Mathematics > Optimization and Control
[Submitted on 5 Apr 2025]
Title:Curvature accelerated decentralized non-convex optimization for high-dimensional machine learning problems
View PDF HTML (experimental)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.
References & Citations
Bibliographic and Citation Tools
Bibliographic Explorer (What is the Explorer?)
Connected Papers (What is Connected Papers?)
Litmaps (What is Litmaps?)
scite Smart Citations (What are Smart Citations?)
Code, Data and Media Associated with this Article
alphaXiv (What is alphaXiv?)
CatalyzeX Code Finder for Papers (What is CatalyzeX?)
DagsHub (What is DagsHub?)
Gotit.pub (What is GotitPub?)
Hugging Face (What is Huggingface?)
Papers with Code (What is Papers with Code?)
ScienceCast (What is ScienceCast?)
Demos
Recommenders and Search Tools
Influence Flower (What are Influence Flowers?)
CORE Recommender (What is CORE?)
arXivLabs: experimental projects with community collaborators
arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.
Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.
Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.