Polynomial and Parallelizable Preconditioning for Block Tridiagonal Positive Definite Matrices
Shaohui Yang, Toshiyuki Ohtsuka, Brian Plancher, Colin N. Jones
TL;DR
This work targets efficient GPU-enabled solutions for moderately large SPD block-tridiagonal KKT systems arising in optimal control problems. It introduces a parallelizable, parametric multi-splitting polynomial preconditioner, derives spectrum- and SPD-focused theoretical guarantees, and identifies an optimal parameter set that yields highly clustered eigenvalues. A polynomial extension with Chebyshev-like coefficients further accelerates convergence, supported by explicit closed-form preconditioners for small problem sizes and comprehensive complexity analysis. Numerical experiments on synthetic OCP-derived KKT systems demonstrate reduced PCG iterations and mat-vec counts, highlighting practical impact for real-time MPC and trajectory optimization on parallel hardware.
Abstract
The efficient solution of moderately large-scale linear systems arising from the KKT conditions in optimal control problems (OCPs) is a critical challenge in robotics. With the stagnation of Moore's law, there is growing interest in leveraging GPU-accelerated iterative methods, and corresponding parallel preconditioners, to overcome these computational challenges. To improve the performance of such solvers, we introduce a parallel-friendly, parametrized multi-splitting polynomial preconditioner framework. We first construct and prove the optimal parametrization theoretically in terms of the least amount of distinct eigenvalues and the narrowest spectrum range. We then compare the theoretical time complexity of solving the linear system directly or iteratively. We finally show through numerical experiments how much the preconditioning improves the convergence of OCP linear systems solves.
