We present sparse dictionary differentiable predictive control (SD-DPC), a framework for learning sparse, interpretable feedback policies for nonlinear systems from data. A prediction model is first identified by rollout-based sparse identification of nonlinear dynamics (SINDy), building on gradient-based and multistep formulations. The policy is then parameterized as a sparse combination of dictionary functions and trained by differentiating a constrained finite-horizon predictive-control objective through this model, so that its terms are selected by closed-loop performance rather than by imitating a previously trained controller. The result is an explicit feedback law with only a handful of terms. Across three benchmark control problems, SD-DPC satisfies the constraints in almost all test scenarios, outperforms a policy distilled onto the same terms by up to an order of magnitude, and requires orders of magnitude less memory and online computation than an optimization benchmark, while admitting explicit sensitivity bounds.
Figures & tables
Fig. 1: Pipeline: sparse system identification and differentiable predictive control policy learning.
Fig. 2: Policy sparsity versus ℓ1 weight λπ . Marker shape denotes pruning threshold τπ , color shows the loss (excluding the ℓ1 penalty) on the validation scenarios used for selection, and black circles indicate the selected policies.
Fig. 3: Closed-loop trajectories of SD-DPC, NN-DPC, NN-Distilled, and the IPOPT benchmark, with the retained SD-DPC policy coefficients, for (a) the two-tank system, (b) the forced Van der Pol oscillator, and (c) the 2-D integrator with obstacle avoidance.
Two-Tank
Van der Pol
2-D Int.
Identification
Trajectories mid
2000
2000
1500
Horizon Nid
5
3
3
Integrator
RK4
RK4
discrete
Step Δt [s]
1.0
0.1
–
Epochs ne
20000
6000
20000
TABLE I: Identification and policy training parameters.
Method
Task loss [10−3]
Control effort
Time [μs] / memory
Violations [%]/ depth
Two-Tank
IPOPT
8.8×10−6
–
8444/26 MiB
0.0/–
NN-DPC
0.144±0.013
–
30/5.01 kB
0.0/–
SD-DPC
0.012±0.006
–
17/0.047 kB
0.0/–
NN-Distilled
0.209±0.039
–
17/0.047 kB
0.0/–
Van der Pol
TABLE II: Controller evaluation across case studies.
Differentiable predictive control (DPC), a self-supervised learning approach for approximating explicit model predictive control (MPC) policies, offers significant computational advantages over online optimization-based MPC. However, feasibility guarantees, a core requirement for safe control, are currently provided either probabilistically or via online safety filters. The lack of rigorous feasibility guarantees for offline policy optimization remains an open problem. This paper establishes deterministic feasibility guarantees for DPC using a novel topological analysis of the induced reachable safe set, without requiring online safety filters. By exploiting the inherent model-based nature of DPC, in which differentiable system dynamics are embedded directly into the computational graph, we analyze the properties of the learned control policies and the corresponding system states from topological and geometric perspectives. Inspired by our theoretical analysis, we propose a novel self-supervised offline policy learning strategy that utilizes a proxy loss with Control Barrier Functions (CBFs). Crucially, these properties not only significantly improve policy training but also enable the derivation of strict, deterministic feasibility guarantees from a finite number of training samples. Extensive closed-loop simulations validate our theoretical findings, demonstrating that the empirical constraint violations monotonically decrease to zero as the training sample size increases. Ultimately, this work illustrates that DPC policy optimization yields formal safety certificates that are structurally unattainable with conventional black-box methods, e.g., reinforcement learning (RL) or supervised learning-based approximate MPC, thereby providing a new perspective on feasibility guarantees in learning-based control.
Guangyu Wu, Ján Drgoňa
Department of Mathematical Sciences, Chalmers University of Technology, Sweden (Incoming) · Department of Civil and Systems Engineering, Johns Hopkins University, USA
We present AC-SINDy, a compositional extension of the Sparse Identification of Nonlinear Dynamics (SINDy) framework that replaces explicit feature libraries with a structured representation based on arithmetic circuits. Rather than enumerating candidate basis functions, the proposed approach constructs nonlinear features through compositions of linear functions and multiplicative interactions, yielding a compact and scalable parameterization and enabling sparsity to be enforced directly over the computational graph. We also introduce a formulation that separates state estimation from dynamics identification by combining latent state inference with shared dynamics and multi-step supervision, improving robustness to noise while preserving interpretability. Experiments on nonlinear and chaotic systems demonstrate that the method recovers accurate and interpretable governing equations while scaling more favorably than standard SINDy.
The practical deployment of nonlinear model predictive control (NMPC) is often limited by online computation: solving a nonlinear program at high control rates can be expensive on embedded hardware, especially when models are complex or horizons are long. Learning-based NMPC approximations shift this computation offline but typically demand large expert datasets and costly training. We propose Sequential-AMPC, a sequential neural policy that generates MPC candidate control sequences by sharing parameters across the prediction horizon. For deployment, we wrap the policy in a safety-augmented online evaluation and fallback mechanism, yielding Safe Sequential-AMPC. Compared to a naive feedforward policy baseline across several benchmarks, Sequential-AMPC requires substantially fewer expert MPC rollouts and yields candidate sequences with higher feasibility rates and improved closed-loop safety. On high-dimensional systems, it also exhibits better learning dynamics and performance in fewer epochs while maintaining stable validation improvement where the feedforward baseline can stagnate.
Institute of Computer Engineering, TU Wien, Vienna, Austria · AIT Austrian Institute of Technology, Vienna, Austria · Institute of Mechanics and Mechatronics, TU Wien, Vienna, Austria