DTZO: Distributed Trilevel Zeroth Order Learning with Provable Non-Asymptotic Convergence
Yang Jiao, Kai Yang, Chengtao Jian
Abstract
Trilevel learning (TLL) with zeroth order constraints is a fundamental problem in machine learning, arising in scenarios where gradient information is inaccessible due to data privacy or model opacity, such as in federated learning, healthcare, and financial systems. These problems are notoriously difficult to solve due to their inherent complexity and the lack of first order information. Moreover, in many practical scenarios, data may be distributed across various nodes, necessitating strategies to address trilevel learning problems without centralizing data on servers to uphold data privacy. To this end, an effective distributed trilevel zeroth order learning framework DTZO is proposed in this work to address the trilevel learning problems with level-wise zeroth order constraints in a distributed manner. The proposed DTZO is versatile and can be adapted to a wide range of (grey-box) trilevel learning problems with partial zeroth order constraints. In DTZO, the cascaded polynomial approximation can be constructed without relying on gradients or sub-gradients, leveraging a novel cut, i.e., zeroth order cut. Furthermore, we theoretically carry out the non-asymptotic convergence rate analysis for the proposed DTZO in achieving the ϵstationary point. Extensive experiments have been conducted to demonstrate and validate the superior performance of the proposed DTZO.
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 b049fced-06e8-4dd8-aeb1-8e35c29b4dbeBuilds on32
- Black-Box Tuning for Language-Model-as-a-ServiceTianxiang Sun, Yunfan Shao, Hong Qian, Xuanjing Huang et al.ICML 2022 · 343 citations
- Provably Faster Algorithms for Bilevel OptimizationJunjie Yang, Kaiyi Ji, Yingbin LiangNeurIPS 2021 · 175 citations
- Grammar Prompting for Domain-Specific Language Generation with Large Language ModelsBailin Wang, Zi Wang, Xuezhi Wang, Yuan Cao et al.NeurIPS 2023 · 138 citations
- Revisiting Zeroth-Order Optimization for Memory-Efficient LLM Fine-Tuning: A BenchmarkYihua Zhang, Pingzhi Li, Junyuan Hong, Jiaxiang Li et al.ICML 2024 · 134 citations
- Revisiting and Advancing Fast Adversarial Training Through The Lens of Bi-Level OptimizationYihua Zhang, Guanhua Zhang, Prashant Khanduri, Mingyi Hong et al.ICML 2022 · 107 citations
Related papers
- Provably Convergent Federated Trilevel LearningYang Jiao, Kai Yang, Tiancheng Wu, Chengtao Jian et al.AAAI 2024 · 6 citations
- A Unified Solution for Privacy and Communication Efficiency in Vertical Federated LearningGanyu Wang, Bin Gu, Qingsong Zhang, Xiang Li et al.NeurIPS 2023 · 22 citations
- Zeroth-Order Methods for Nondifferentiable, Nonconvex, and Hierarchical Federated OptimizationYuyang Qiu, Uday V. Shanbhag, Farzad YousefianNeurIPS 2023 · 23 citations
- Gradient-Free Method for Heavily Constrained Nonconvex OptimizationWanli Shi, Hongchang Gao, Bin GuICML 2022 · 5 citations
- Single Point-Based Distributed Zeroth-Order Optimization with a Non-Convex Stochastic Objective FunctionElissa Mhanna, Mohamad AssaadICML 2023 · 10 citations
