🤖 AI Summary
This work addresses optimal control of infinite-dimensional systems governed by partial differential equations (PDEs). We propose an end-to-end differentiable learning framework that jointly achieves reference tracking, satisfaction of state and control constraints, and minimization of control curvature. Methodologically, we employ a time-integration DeepONet (TI-DeepONet) as a high-fidelity, differentiable PDE surrogate model; integrate numerical derivative learning with automatic differentiation; and embed the surrogate into a differentiable model predictive control architecture—thereby preserving temporal causality while mitigating error accumulation in long-horizon predictions. The control policy is obtained via offline optimization of the expected control loss, eliminating the need for online optimization or supervised labels. Experiments on the heat equation, Burgers equation, and reaction-diffusion equation demonstrate that the learned policy exhibits strong generalization across diverse initial conditions and PDE parameter distributions, significantly outperforming baseline methods.
📝 Abstract
We present an end-to-end learning to control framework for partial differential equations (PDEs). Our approach integrates Time-Integrated Deep Operator Networks (TI-DeepONets) as differentiable PDE surrogate models within the Differentiable Predictive Control (DPC)-a self-supervised learning framework for constrained neural control policies. The TI-DeepONet architecture learns temporal derivatives and couples them with numerical integrators, thus preserving the temporal causality of infinite-dimensional PDEs while reducing error accumulation in long-horizon predictions. Within DPC, we leverage automatic differentiation to compute policy gradients by backpropagating the expectations of optimal control loss through the learned TI-DeepONet, enabling efficient offline optimization of neural policies without the need for online optimization or supervisory controllers. We empirically demonstrate that the proposed method learns feasible parametric policies across diverse PDE systems, including the heat, the nonlinear Burgers', and the reaction-diffusion equations. The learned policies achieve target tracking, constraint satisfaction, and curvature minimization objectives, while generalizing across distributions of initial conditions and problem parameters. These results highlight the promise of combining operator learning with DPC for scalable, model-based self-supervised learning in PDE-constrained optimal control.