Related Experiment Videos
Converge Faster, Talk Less: Hessian-Informed Federated Zeroth-Order Optimization
Zhe Li1, Bicheng Ying2, Zidong Liu3
1Rochester Institute of Technology, Rochester, NY 14623, USA.
Abstract:
Zeroth-order (ZO) optimization enables dimension-free communication in federated learning (FL), making it attractive for fine-tuning of large language models (LLMs) due to significant communication savings. However, existing ZO-FL methods largely overlook curvature information, despite its well-established benefits for convergence acceleration. To address this, we propose HiSo, a Hessian-informed ZO federated optimization method that accelerates convergence by leveraging global diagonal Hessian approximations, while strictly preserving scalar-only communication without transmitting any second-order information. Theoretically, for non-convex functions, we show that HiSo can achieve an accelerated convergence rate that is independent of the Lipschitz constant and model dimension under some Hessian approximation assumptions, offering a plausible explanation for the observed phenomenon of ZO convergence being much faster than its worst-case -bound. Empirically, across diverse LLM fine-tuning benchmarks, HiSo delivers a 1~5脳 speedup in communication rounds over existing state-of-the-art ZO-FL baselines. This superior convergence not only cuts communication costs but also provides strong empirical evidence that Hessian information acts as an effective accelerator in federated ZO optimization settings.
Related Concept Videos
Optimization Problems
Methods of Medium Optimization
Second Order systems I
By reinterpreting the system, one can derive the closed-loop transfer function, which...
Linear Approximations
Linearization and Approximation
First Order Systems
When a first-order system is subjected to a unit-step input, its response is characterized by its transfer function. By applying the Laplace transform of the unit-step input to the transfer function, expanding the...