Distributed Distributionally Robust Optimization with Non-Convex Objectives
Yang Jiao, Kai Yang, Dongjin Song
Abstract
Distributionally Robust Optimization (DRO), which aims to find an optimal decision that minimizes the worst case cost over the ambiguity set of probability distribution, has been widely applied in diverse applications, e.g., network behavior analysis, risk management, etc. However, existing DRO techniques face three key challenges: 1) how to deal with the asynchronous updating in a distributed environment; 2) how to leverage the prior distribution effectively; 3) how to properly adjust the degree of robustness according to different scenarios. To this end, we propose an asynchronous distributed algorithm, named Asynchronous Single-looP alternatIve gRadient projEction (ASPIRE) algorithm with the itErative Active SEt method (EASE) to tackle the distributed distributionally robust optimization (DDRO) problem. Furthermore, a new uncertainty set, i.e., constrained D-norm uncertainty set, is developed to effectively leverage the prior distribution and flexibly control the degree of robustness. Finally, our theoretical analysis elucidates that the proposed algorithm is guaranteed to converge and the iteration complexity is also analyzed. Extensive empirical studies on real-world datasets demonstrate that the proposed method can not only achieve fast convergence, and remain robust against data heterogeneity as well as malicious attacks, but also tradeoff robustness with performance.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext cba51dbb-f44b-4d2a-8d2f-6e5e40025191Cited by top-tier papers5
- Tri-Level Navigator: LLM-Empowered Tri-Level Learning for Time Series OOD GeneralizationChengtao Jian, Kai Yang, Yang JiaoNeurIPS 2024 · 19 citations
- Distributionally Robust Optimization with Bias and Variance ReductionRonak Mehta, Vincent Roulet, Krishna Pillutla, Zaïd HarchaouiICLR 2024 · 6 citations
- PR-Attack: Coordinated Prompt-RAG Attacks on Retrieval-Augmented Generation in Large Language Models via Bilevel OptimizationYang Jiao, Xiaodong Wang, Kai YangSIGIR 2025 · 6 citations
- Asynchronous Distributed Bilevel OptimizationYang Jiao, Kai Yang, Tiancheng Wu, Dongjin Song et al.ICLR 2023 · 6 citations
- Robust Beamforming for Downlink Multi-Cell Systems: A Bilevel Optimization PerspectiveXingdi Chen, Yu Xiong, Kai YangAAAI 2024 · 5 citations
Builds on12
- Neural Cleanse: Identifying and Mitigating Backdoor Attacks in Neural NetworksBolun Wang, Yuanshun Yao, Shawn Shan, Huiying Li et al.S&P 2019 · 1,801 citations
- On Gradient Descent Ascent for Nonconvex-Concave Minimax ProblemsTianyi Lin, Chi Jin, Michael I. JordanICML 2020 · 587 citations
- What is Local Optimality in Nonconvex-Nonconcave Minimax Optimization?Chi Jin, Praneeth Netrapalli, Michael I. JordanICML 2020 · 381 citations
- Large-Scale Methods for Distributionally Robust OptimizationDaniel Levy, Yair Carmon, John C. Duchi, Aaron SidfordNeurIPS 2020 · 281 citations
- Distributionally Robust Federated AveragingYuyang Deng, Mohammad Mahdi Kamani, Mehrdad MahdaviNeurIPS 2020 · 176 citations
Related papers
- Decision Making under the Exponential Family: Distributionally Robust Optimisation with Bayesian Ambiguity SetsCharita Dellaporta, Patrick O'Hara, Theodoros DamoulasICML 2025
- Outlier-Robust Distributionally Robust Optimization via Unbalanced Optimal TransportZifan Wang, Yi Shen, Michael M. Zavlanos, Karl Henrik JohanssonNeurIPS 2024 · 16 citations
- Non-convex Distributionally Robust Optimization: Non-asymptotic AnalysisJikai Jin, Bohang Zhang, Haiyang Wang, Liwei WangNeurIPS 2021 · 65 citations
- Large-Scale Non-convex Stochastic Constrained Distributionally Robust OptimizationQi Zhang, Yi Zhou, Ashley Prater-Bennette, Lixin Shen et al.AAAI 2024 · 6 citations
- Distributionally Robust Optimization via Ball Oracle AccelerationYair Carmon, Danielle HauslerNeurIPS 2022 · 23 citations
