Distributed Optimization Algorithm with Superlinear Convergence Rate

Yeming Xu,Ziyuan Guo,Kaihong Lu,Huanshui Zhang
2024-09-19
Abstract:This paper considers distributed optimization problems, where each agent cooperatively minimizes the sum of local objective functions through the communication with its neighbors. The widely adopted distributed gradient method in solving this problem suffers from slow convergence rates, which motivates us to incorporate the second-order information of the objective functions. However, the challenge arises from the unique structure of the inverse of the Hessian matrix, which prevents the direct distributed implementation of the second-order method. We overcome this challenge by proposing a novel optimization framework. The key idea is to transform the distributed optimization problem into an optimal control problem. Using Pontryagin's maximum principle and the associated forward-backward difference equations (FBDEs), we derive a new distributed optimization algorithm that incorporates the second-order information without requiring the computation of the inverse of the Hessian matrix. Furthermore, the superlinear convergence of the proposed algorithm is proved under some mild assumptions. Finally, we also propose a variant of the algorithm to balance the number of iterations and communication.
Optimization and Control
What problem does this paper attempt to address?
This paper aims to solve the problem of slow convergence rate in distributed optimization problems. Specifically, each agent minimizes the sum of local objective functions through communication and cooperation with neighbors. The existing distributed gradient methods have the problem of slow convergence speed when solving this problem, which prompts researchers to consider introducing second - order information of the objective function to accelerate convergence. However, due to the unique structure of the inverse of the Hessian matrix, directly implementing second - order methods in a distributed environment is challenging. To this end, the author proposes a new optimization framework. By transforming the distributed optimization problem into an optimal control problem and using the Pontryagin maximum principle and related forward - backward difference equations (FBDEs), a new distributed optimization algorithm is derived. This algorithm can incorporate second - order information without calculating the inverse of the Hessian matrix, thereby achieving super - linear convergence. ### Main contributions: 1. **New optimization framework**: An innovative optimization framework is proposed by transforming the distributed optimization problem into an optimal control problem. 2. **Avoiding the calculation of the inverse of the Hessian matrix**: The new algorithm avoids the calculation of the inverse of the Hessian matrix through a special optimization structure, significantly reducing the computational complexity. 3. **Proof of super - linear convergence**: Under certain assumptions, the super - linear convergence of the proposed algorithm is proved. 4. **Balancing the number of iterations and communication**: An algorithm variant is proposed to balance the number of iterations and the amount of communication. ### Key technical points: - **Pontryagin maximum principle**: Used to derive the new optimization algorithm. - **Forward - backward difference equations (FBDEs)**: Used to describe the dynamics of the algorithm. - **Fusion of second - order information**: Incorporate second - order information without calculating the inverse of the Hessian matrix through a special optimization structure. ### Application background: This research has broad application prospects in fields such as sensor networks, distributed control, and machine learning, especially in scenarios where multiple agents need to cooperate for optimization. By increasing the convergence speed of the distributed optimization algorithm, the efficiency and performance of these applications can be significantly improved.