Unverified 2026

Lyapunov-Certified Policy Training

Usefulness8/10
Difficulty5/10
Novelty6/10

Source paper: Learning neural controllers for nonlinear systems from data arXiv:2608.29303 · analyzed Sep 1, 2026

AI-generated research hypothesis, automatically tested. Not peer-reviewed.

Idea description

Train a neural policy together with a positive neural Lyapunov function so that the learned closed-loop transition decreases the function at every sampled state in a prescribed operating region. This converts policy learning from an unconstrained reward problem into a constrained dissipativity problem and provides an inference-time monitor that can reject or damp actions when the certificate is violated.

Formulas

$$V_{\psi}(x^{\star})=0,\qquad V_{\psi}(x)\geq c_{1}\|x-x^{\star}\|_{2}^{2},\qquad c_{1}>0,$$
$$\Delta V_{\theta,\phi,\psi}(x)=V_{\psi}\!\left(\hat f_{\phi}(x,\pi_{\theta}(x))\right)-V_{\psi}(x)\leq -c_{2}\|x-x^{\star}\|_{2}^{2},\qquad c_{2}>0,$$
$$\mathcal{L}=\mathcal{L}_{\mathrm{task}}+\lambda_{V}\mathbb{E}_{x\sim\mathcal{D}}[\operatorname{softplus}(\varepsilon_{V}-V_{\psi}(x))]+\lambda_{\Delta}\mathbb{E}_{x\sim\mathcal{D}}[\operatorname{softplus}(\Delta V_{\theta,\phi,\psi}(x)+c_{2}\|x-x^{\star}\|_{2}^{2})],$$
$$V_{t+1}\leq\left(1-\frac{c_{2}}{c_{1}}\right)V_{t}\quad\Longrightarrow\quad V_{t}\leq\left(1-\frac{c_{2}}{c_{1}}\right)^{t}V_{0},\qquad 0<c_{2}<c_{1}.$$

Mathematical statement

Let $x\in\mathbb{R}^{n}$ be the state, $u=\pi_{\theta}(x)$ the neural controller with parameters $\theta$, and $\hat f_{\phi}(x,u)$ the identified one-step dynamics model with parameters $\phi$. Let $V_{\psi}(x)$ be a neural Lyapunov candidate with parameters $\psi$, equilibrium $x^{\star}$, and region $\mathcal{X}$. Enforce $V_{\psi}(x^{\star})=0$, $V_{\psi}(x)\geq c_{1}\|x-x^{\star}\|_{2}^{2}$, and $V_{\psi}(\hat f_{\phi}(x,\pi_{\theta}(x))) - V_{\psi}(x)\leq -c_{2}\|x-x^{\star}\|_{2}^{2}$ for all $x\in\mathcal{X}$, where $c_{1},c_{2}>0$. The decrease inequality implies geometric decay of the sampled closed-loop energy when the dynamics are discrete-time. In practice, optimize hinge penalties for violations and periodically verify the inequalities over boxes or a sampled state set; the certified subset is the set on which all inequalities hold.

Implementation notes

1. Integration point: use this in model-based RL, recurrent world-model rollout training, or any neural state-transition module where stability of generated trajectories matters. Maintain three networks: a dynamics model $\hat f_{\phi}$, policy $\pi_{\theta}$, and Lyapunov network $V_{\psi}$. Parameterize $V_{\psi}(x)=\|g_{\psi}(x)-g_{\psi}(x^{\star})\|_{2}^{2}+\epsilon\|x-x^{\star}\|_{2}^{2}$ with $\epsilon>0$ to make positivity automatic. 2. Pseudocode: sample states $x$ from replay data and perturbations around the equilibrium; compute $u=\pi_{\theta}(x)$, $x^{+}=\hat f_{\phi}(x,u)$, and $dV=V_{\psi}(x^{+})-V_{\psi}(x)$; add the two hinge terms in the displayed loss; update $\theta,\psi$ jointly and update $\phi$ using the transition prediction loss. Every $K$ steps, freeze the networks and test $V(x)>0$ and $dV+c_{2}\|x-x^{\star}\|^{2}\leq0$ on a dense grid or interval boxes. 3. Computed quantities are the one-step model prediction, Lyapunov values, and violation margins. Model error and the largest verified region must be estimated from held-out transitions or interval bounds. 4. First cheap experiment: train a two-layer policy and Lyapunov network on a two-dimensional nonlinear benchmark such as $x_{t+1}=x_t+0.05[-x_{1}^{3}+u,\ x_{2}+0.2x_{1}]$, comparing PPO or supervised control against the certificate-regularized version. Sweep $\lambda_{\Delta}$ and estimate the largest radius $r$ for which the decrease condition holds. The quantitative prediction is that certified trajectories satisfy $V_t/V_0\leq(1-c_{2}/c_{1})^t$ up to model error, and empirical divergence begins near the radius or parameter value where the measured minimum decrease margin crosses zero.

Verification

This idea has not been verified yet.

Verification happens in two stages: Stage 1 — a mechanism check on a toy system confirms the claimed mathematical phenomenon reproduces; Stage 2 — a benchmark implements the idea on a real (small) neural network task and compares it against a tuned baseline over 8 paired seeds with a permutation test.

Artifacts

Artifacts unavailable.