Distributed optimization: designed for federated learning

2508.08606v1 cs.LG, math.OC, stat.ML 2025-08-14
Авторы:

Wenyou Guo, Ting Qu, Chunrong Pan, George Q. Huang

Резюме на русском

## Контекст Federated Learning (FL) является распределённым коллаборативным методом машинного обучения, работающим под условиями сохранения конфиденциальности. Он привлекает внимание научного сообщества в сценариях межорганизационного обмена данными, где требуется соблюдение правил защиты личных данных. Однако FL сталкивается с рядом проблем, включая естественную сложность выравнивания клиентских моделей в сетях с разными топологиями связи. Наиболее распространенными алгоритмами в этой области являются методы градиентного спуска, но они часто неэффективны в условиях высокой статистической неоднородности данных между клиентами. Наличие этих проблем поднимает вопрос о разработке более эффективных методов для распределённых оптимизационных задач в FL. ## Метод Разработанный подход основывается на усилённом лагранжевом приближении, которое включает в себя проксимальное приближение и квадратичное приближение. Это позволяет гибко адаптировать алгоритм к различным топологиям связи, включая централизованные и децентрализованные сети FL. Для контроля процесса оптимизации вводятся множество критериев завершения и механизмов обновления параметров, которые улучшают вычислительную эффективность. Теоретические анализы обеспечивают гарантии сходимости для всех используемых методов, что позволяет гладко интегрировать продвинутые методы оптимизации в рамки FL. ## Результаты Проведены эксперименты на больших данных с высокой статистической неоднородностью. Результаты показывают, что предложенный метод вы most significantly outperforms existing methods in terms of convergence speed, accuracy, and robustness. Особенно заметно выигрыш в случае сетей с нерегулярной топологией связи, где другие подходы часто либо застревают, либо неэффективны. Эти результаты доказывают широкую применимость предложенной модели в сценариях с высоким спросом на эффективность и точность. ## Значимость Предложенный подход может применяться в различных областях, включая здравоохранение, финансы, и интернет вещей, где требуется обработка распределённых данных в условиях строгого соблюдения конфиденциальности. Помимо высокой эффективности, алгоритм обеспечивает гибкость и устойчивость в разных сетевых условиях. Это делает его привлекательным для решения реальных проблем в сферах, где традиционные методы могут сталкиваться с ограничениями. ## Выводы Предложенный метод демонстрирует сильную эффективность в задачах распределённой оптимизации для FL. Он может быть применён для решения сложных задач в различных сферах, где требуется обеспечение конфиденциальности и высокая точность. Будущие исследо

Abstract

Federated Learning (FL), as a distributed collaborative Machine Learning (ML) framework under privacy-preserving constraints, has garnered increasing research attention in cross-organizational data collaboration scenarios. This paper proposes a class of distributed optimization algorithms based on the augmented Lagrangian technique, designed to accommodate diverse communication topologies in both centralized and decentralized FL settings. Furthermore, we develop multiple termination criteria and parameter update mechanisms to enhance computational efficiency, accompanied by rigorous theoretical guarantees of convergence. By generalizing the augmented Lagrangian relaxation through the incorporation of proximal relaxation and quadratic approximation, our framework systematically recovers a broad of classical unconstrained optimization methods, including proximal algorithm, classic gradient descent, and stochastic gradient descent, among others. Notably, the convergence properties of these methods can be naturally derived within the proposed theoretical framework. Numerical experiments demonstrate that the proposed algorithm exhibits strong performance in large-scale settings with significant statistical heterogeneity across clients.

Ссылки и действия