Distributed Online Optimization over a Heterogeneous Network with Any-Batch Mirror Descent
Nima Eshraghi, Ben Liang
Abstract
In distributed online optimization over a computing network with heterogeneous nodes, slow nodes can adversely affect the progress of fast nodes, leading to drastic slowdown of the overall convergence process. To address this issue, we consider a new algorithm termed Distributed Any-Batch Mirror Descent (DABMD), which is based on distributed Mirror Descent but uses a fixed per-round computing time to limit the waiting by fast nodes to receive information updates from slow nodes. DABMD is characterized by varying minibatch sizes across nodes. It is applicable to a broader range of problems compared with existing distributed online optimization methods such as those based on dual averaging, and it accommodates time-varying network topology. We study two versions of DABMD, depending on whether the computing nodes average their primal variables via single or multiple consensus iterations. We show that both versions provide strong theoretical performance guarantee, by deriving upperbounds on their expected dynamic regret, which capture the variability in minibatch sizes. Our experimental results show substantial reduction in cost and acceleration in convergence compared with the known best alternative.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers5
- Delay-Tolerant Constrained OCO with Application to Network Resource AllocationJuncheng Wang, Ben Liang, Min Dong, Gary Boudreau et al.INFOCOM 2021 · 11 citations
- HyperPrism: An Adaptive Non-linear Aggregation Framework for Distributed Machine Learning over Non-IID Data and Time-varying Communication LinksHaizhou Du, Yijian Chen, Ryan Yang, Yuchen Li et al.NeurIPS 2024 · 4 citations
- Doubly-Bounded Queue for Constrained Online Learning: Keeping Pace with Dynamics of Both Loss and ConstraintJuncheng Wang, Bingjie Yan, Yituo LiuAAAI 2025 · 1 citation
- Semi-Online Precoding with Information Parsing for Cooperative MIMO Wireless NetworksJuncheng Wang, Ben Liang, Min Dong, Gary Boudreau et al.INFOCOM 2022 · 1 citation
- Event-Driven Online Vertical Federated LearningGanyu Wang, Boyu Wang, Bin Gu, Charles LingICLR 2025
Related papers
- Online mirror descent and dual averaging: keeping pace in the dynamic caseHuang Fang, Nick Harvey, Victor S. Portella, Michael P. FriedlanderICML 2020 · 38 citations
- FedADMM: A Robust Federated Deep Learning Framework with Adaptivity to System HeterogeneityYonghai Gong, Yichuan Li, Nikolaos M. FrerisICDE 2022 · 41 citations
- Sharper Convergence Guarantees for Asynchronous SGD for Distributed and Federated LearningAnastasia Koloskova, Sebastian U. Stich, Martin JaggiNeurIPS 2022 · 131 citations
- Stochastic Approximation Approaches to Group Distributionally Robust OptimizationLijun Zhang, Peng Zhao, Zhen-Hua Zhuang, Tianbao Yang et al.NeurIPS 2023 · 24 citations
- Multi-Level Local SGD: Distributed SGD for Heterogeneous Hierarchical NetworksTimothy Castiglia, Anirban Das, Stacy PattersonICLR 2021 · 11 citations
