Optimal Control Operator Perspective and a Neural Adaptive Spectral Method
Mingquan Feng, Zhijie Chen, Yixin Huang, Yizhou Liu, Junchi Yan
Abstract
Optimal control problems (OCPs) involve finding a control function for a dynamical system such that a cost functional is optimized. It is central to physical systems in both academia and industry. In this paper, we propose a novel instance-solution control operator perspective, which solves OCPs in a one-shot manner without direct dependence on the explicit expression of dynamics or iterative optimization processes. The control operator is implemented by a new neural operator architecture named Neural Adaptive Spectral Method (NASM), a generalization of classical spectral methods. We theoretically validate the perspective and architecture by presenting the approximation error bounds of NASM for the control operator. Experiments on synthetic environments and a real-world dataset verify the effectiveness and efficiency of our approach, including substantial speedup in running time, and high-quality in- and out-of-distribution generalization.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 237cf6d7-cbd7-4547-8d79-eec4087795f5Cited by top-tier papers1
Ask how each one uses itBuilds on4
- Fourier Neural Operator for Parametric Partial Differential EquationsZongyi Li, Nikola Borislavov Kovachki, Kamyar Azizzadenesheli, Burigede Liu et al.ICLR 2021 · 3,911 citations
- Pontryagin Differentiable Programming: An End-to-End Learning and Control FrameworkWanxin Jin, Zhaoran Wang, Zhuoran Yang, Shaoshuai MouNeurIPS 2020 · 133 citations
- Solving PDE-Constrained Control Problems Using Operator LearningRakhoon Hwang, Jae Yong Lee, Jinyoung Shin, Hyung Ju HwangAAAI 2022 · 66 citations
- Hyperverlet: A Symplectic Hypersolver for Hamiltonian SystemsFrederik Baymler Mathiesen, Bin Yang, Jilin HuAAAI 2022 · 5 citations
Related papers
- Neural Time-Reversed Generalized Riccati EquationAlessandro Betti, Michele Casoni, Marco Gori, Simone Marullo et al.AAAI 2024 · 3 citations
- Thompson Sampling in Function Spaces via Neural OperatorsRafael Oliveira, Xuesong Wang, Kian Ming A. Chai, Edwin V. BonillaNeurIPS 2025 · 2 citations
- A Schrödinger Eigenfunction Method for Long-Horizon Stochastic Optimal ControlLouis Claeys, Artur Goldman, Zebang Shen, Niao HeICLR 2026
- Adaptive Mamba Neural OperatorsZeyuan Song, Zheyu JiangICLR 2026 · 6 citations
- Data-Efficient Operator Learning via Unsupervised Pretraining and In-Context LearningWuyang Chen, Jialin Song, Pu Ren, Shashank Subramanian et al.NeurIPS 2024 · 41 citations
