A Mathematics-Inspired Learning-to-Optimize Framework for Decentralized Optimization

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: He, Yutong, Shang, Qiulin, Huang, Xinmeng, Liu, Jialin, Yuan, Kun
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917793240186880
author He, Yutong
Shang, Qiulin
Huang, Xinmeng
Liu, Jialin
Yuan, Kun
author_facet He, Yutong
Shang, Qiulin
Huang, Xinmeng
Liu, Jialin
Yuan, Kun
contents Most decentralized optimization algorithms are handcrafted. While endowed with strong theoretical guarantees, these algorithms generally target a broad class of problems, thereby not being adaptive or customized to specific problem features. This paper studies data-driven decentralized algorithms trained to exploit problem features to boost convergence. Existing learning-to-optimize methods typically suffer from poor generalization or prohibitively vast search spaces. In addition, the vast search space of communicating choices and final goal to reach the global solution via limited neighboring communication cast more challenges in decentralized settings. To resolve these challenges, this paper first derives the necessary conditions that successful decentralized algorithmic rules need to satisfy to achieve both optimality and consensus. Based on these conditions, we propose a novel Mathematics-inspired Learning-to-optimize framework for Decentralized optimization (MiLoDo). Empirical results demonstrate that MiLoDo-trained algorithms outperform handcrafted algorithms and exhibit strong generalizations. Algorithms learned via MiLoDo in 100 iterations perform robustly when running 100,000 iterations during inferences. Moreover, MiLoDo-trained algorithms on synthetic datasets perform well on problems involving real data, higher dimensions, and different loss functions.
format Preprint
id arxiv_https___arxiv_org_abs_2410_01700
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle A Mathematics-Inspired Learning-to-Optimize Framework for Decentralized Optimization
He, Yutong
Shang, Qiulin
Huang, Xinmeng
Liu, Jialin
Yuan, Kun
Optimization and Control
Most decentralized optimization algorithms are handcrafted. While endowed with strong theoretical guarantees, these algorithms generally target a broad class of problems, thereby not being adaptive or customized to specific problem features. This paper studies data-driven decentralized algorithms trained to exploit problem features to boost convergence. Existing learning-to-optimize methods typically suffer from poor generalization or prohibitively vast search spaces. In addition, the vast search space of communicating choices and final goal to reach the global solution via limited neighboring communication cast more challenges in decentralized settings. To resolve these challenges, this paper first derives the necessary conditions that successful decentralized algorithmic rules need to satisfy to achieve both optimality and consensus. Based on these conditions, we propose a novel Mathematics-inspired Learning-to-optimize framework for Decentralized optimization (MiLoDo). Empirical results demonstrate that MiLoDo-trained algorithms outperform handcrafted algorithms and exhibit strong generalizations. Algorithms learned via MiLoDo in 100 iterations perform robustly when running 100,000 iterations during inferences. Moreover, MiLoDo-trained algorithms on synthetic datasets perform well on problems involving real data, higher dimensions, and different loss functions.
title A Mathematics-Inspired Learning-to-Optimize Framework for Decentralized Optimization
topic Optimization and Control
url https://arxiv.org/abs/2410.01700