Optimizing Static Linear Feedback: Gradient Method I: The Gradient Method Converges to a Stationary Point, and Linearly to the Optimal Gain under State FeedbackResearch Paper
Motivation
The linear-quadratic regulator (LQR) is the basic problem of optimal control: steer a linear system so as to minimize an integrated quadratic cost. When the full state is measured and the gain may be chosen freely, the optimal feedback is given by the algebraic Riccati equation (Kalman, 1960). In many applications only an output is measured, and the controller is restricted to a static feedback . For this output-feedback problem no Riccati-type characterization exists; the design problem is a non-convex optimization over the gain matrix .
Direct optimization of the gain by gradient descent, known in the control literature since Levine and Athans (1970) and revived in reinforcement learning as policy gradient (Fazel, Ge, Kakade and Mesbahi, 2018, arXiv:1801.05039), is therefore of interest both to control engineers and to the learning community. Fatkhullin and Polyak (arXiv:2004.09875, SIAM J. Control Optim. 2021) give a self-contained analysis of the continuous-time problem: the cost is coercive on the set of stabilizing gains, smooth on sublevel sets, and, for state feedback, satisfies a gradient-domination (Łežanski–Polyak–Łojasiewicz) inequality. From these they derive convergence guarantees for the gradient method.
Setting
Fix real matrices , , and weights , , and an initial-state covariance . A gain is a matrix , and the closed-loop matrix is . A square matrix is Hurwitz if all its complex eigenvalues have negative real part. The set of stabilizing gains is
For let be the unique solution of the Lyapunov equation
and define the cost , the expected integrated quadratic cost of the closed loop from a random initial state with covariance . With the solution of , the gradient of in the Frobenius inner product is
A known stabilizing gain is given, and is its sublevel set. The standing assumptions are , and . State feedback (SLQR) is the case .
The gradient method with step sizes is
A number is a smoothness constant if for all .
Formalization targets
Goal: Theorem 4.2 for state feedback
For , an optimal gain , and any smoothness constant :
- if for all , then every and
- if with , then ,
and there are , with .
Milestones
In attack order:
- Appendix A lemmas. Trace duality of dual Lyapunov equations (Lemma A.1), the trace sandwich (Lemma A.4), and eigenvalue lower bounds for Lyapunov solutions (Lemma A.5).
- Coercivity and existence. Coercivity of with the lower bounds (3.1)–(3.2) (Lemma 3.8), boundedness of (Corollary 3.9), and existence of a minimizer (Corollary 3.10).
- Smoothness. The gradient formula (Lemma 3.11) and existence of a smoothness constant on (Theorem 3.15, qualitative form).
- Gradient domination for state feedback. Lemmas C.2, C.3 and C.1, and the LPL inequality with the explicit constant (3.11):
- Theorem 4.2 for output feedback. Descent and stationarity, parts 1 and 2 without the linear rate, for general .
Significance
The theorem shows that a plain first-order method, started from any stabilizing gain, never destabilizes the closed loop and decreases the cost monotonically, for output feedback as well as state feedback. For state feedback it converges globally and linearly to the optimal gain. The cost is non-convex, and its domain is open, possibly non-convex and unbounded, so this does not follow from convex optimization theory. It is the continuous-time counterpart of the policy-gradient guarantees of Fazel et al. for discrete-time LQR, and it underlies model-free and data-driven variants of gain tuning.
The result is proved in the paper, but it has not been formalized. The formalization adds three things. It makes the invariance argument (the iterates stay in ) explicit, and the paper describes that argument as the non-trivial part. It corrects the statements where the printed text is wrong (see below). It also produces a reusable library of Lyapunov-equation facts. Mathlib has no Lyapunov equation, no LQR cost and no Hurwitz stability theory, and the platform has no continuous-time LQR material. The nearest platform items treat discrete-time Riccati iteration (BertsekasDP.riccati_convergence_stability) and Polyak–Łojasiewicz rates on a whole normed space (ShiOptRates.pl_rate). Neither applies to a function defined only on a non-convex open subset.
Difficulty
The standard descent-lemma argument assumes is defined and -smooth on the whole space. Here is defined only on , and it is not smooth on all of : it blows up at the boundary. A gradient step from a point of could in principle jump out of , where the Lyapunov equation has no meaningful solution. The smoothness bound is available only inside , so the argument must show that the whole segment from to stays in before the descent inequality can be used on it. That requires coercivity, compactness of and an exit-time argument. For the linear rate, gradient domination has to be established on with constants controlled by , and passing from function values to distances to needs that minimizer's structure. Gradient domination fails for output feedback (the paper's Example 3.4 has two disconnected components with different minima), so the linear rate is stated only for .
Formalization scope
Matrices are Matrix (Fin p) (Fin q) ℝ. Hurwitz means every element of the complex spectrum has negative real part. , are "the unique solution of the Lyapunov equation, if there is none or several"; this junk value is never used, because every statement evaluates and only at gains proved or assumed to lie in . The iterates' membership in is a conclusion of the goal, never a hypothesis; assuming it would delete the theorem's content. is defined by the formula (3.3), and Lemma 3.11 is the theorem that it is the gradient. is , is the spectral (operator) norm, and are the minimum and maximum eigenvalue of a symmetric matrix. State feedback is the instance , . In (4.6) the Frobenius norm replaces the paper's spectral norm, which is equivalent because is existential. The minimum over requires .
Deviations from the printed text, each recorded in the item's Formalization Note:
- The smoothness constant. The explicit of (3.8) is false as printed (for , , , , , , , one has ). The goal therefore takes as any Lipschitz constant of on , which is the paper's definition of -smoothness (§3.6) and all that its proof uses. Theorem 3.15 enters only as "some such exists".
- Lemma C.1. It is stated with in the denominator, as its proof concludes and as (3.11) requires.
- Lemma A.5. It is stated for ; the printed admits no positive definite solution.
- The gain space. , where p. 3 prints .
Not stated: Theorem 4.3 and Algorithm 4.1, Lemma 3.6, Lemmas 3.12–3.14, Corollary 3.16 and the explicit constant (3.8). Welcome contributions include a Lyapunov-equation library (existence, uniqueness, integral representation, positivity), continuity of the spectrum, and the exit-time argument, which is reusable for any descent method on a sublevel set of an open domain.
Selected references
- I. Fatkhullin, B. Polyak, Optimizing Static Linear Feedback: Gradient Method, SIAM J. Control Optim. 59(5), 2021; preprint arXiv:2004.09875v2. https://arxiv.org/abs/2004.09875
- M. Fazel, R. Ge, S. Kakade, M. Mesbahi, Global Convergence of Policy Gradient Methods for the Linear Quadratic Regulator, ICML 2018. https://arxiv.org/abs/1801.05039
- W. Levine, M. Athans, On the determination of the optimal constant output feedback gains for linear multivariable systems, IEEE Trans. Automat. Control 15(1), 1970. https://doi.org/10.1109/TAC.1970.1099363
- H. Karimi, J. Nutini, M. Schmidt, Linear Convergence of Gradient and Proximal-Gradient Methods Under the Polyak–Łojasiewicz Condition, ECML PKDD 2016. https://arxiv.org/abs/1608.04636
- R. E. Kalman, Contributions to the theory of optimal control, Bol. Soc. Mat. Mexicana 5, 1960.