Lune

ICML2025Top-tier venue

End-to-End Learning Framework for Solving Non-Markovian Optimal Control

Xiaole Zhang, Peiyu Zhang, Xiongye Xiao, Shixuan Li, Vasileios Tzoumas, Vijay Gupta, Paul Bogdan

Abstract

Integer-order calculus fails to capture the longrange dependence (LRD) and memory effects found in many complex systems. Fractional calculus addresses these gaps through fractional-order integrals and derivatives, but fractional-order dynamical systems pose substantial challenges in system identification and optimal control tasks. In this paper, we theoretically derive the optimal control via linear quadratic regulator (LQR) for fractional-order linear time-invariant (FOLTI) systems and develop an end-to-end deep learning framework based on this theoretical foundation. Our approach establishes a rigorous mathematical model, derives analytical solutions, and incorporates deep learning to achieve data-driven optimal control of FOLTI systems. Our key contributions include: (i) proposing a novel method for system identification and optimal control strategy in FOLTI systems, (ii) developing the first end-toend data-driven learning framework, Fractional-Order Learning for Optimal Control (FOLOC), that learns control policies from observed trajectories, and (iii) deriving theoretical bounds on the sample complexity for learning accurate control policies under fractional-order dynamics. Experimental results indicate that our method accurately approximates fractional-order system behaviors without relying on Gaussian noise assumptions, pointing to promising avenues for advanced optimal control.

Ask Lune

Ask anything. The answer quotes the paper.

Lune reads all of it and the papers around it, then shows you the passages behind every answer.

Questions to start from

Lune answers from

the full paper · related work

Get the answer

Free to start. No credit card required.

Builds on5

  1. Fourier Neural Operator for Parametric Partial Differential EquationsZongyi Li, Nikola Borislavov Kovachki, Kamyar Azizzadenesheli, Burigede Liu et al.ICLR 2021 · 3,911 citations
  2. Multiwavelet-based Operator Learning for Differential EquationsGaurav Gupta, Xiongye Xiao, Paul BogdanNeurIPS 2021 · 355 citations
  3. Neural Krylov Iteration for Accelerating Linear System SolvingJian Luo, Jie Wang, Hong Wang, Huanshuo Dong et al.NeurIPS 2024 · 23 citations
  4. Non-Linear Operator Approximations for Initial Value ProblemsGaurav Gupta, Xiongye Xiao, Radu V. Balan, Paul BogdanICLR 2022 · 18 citations
  5. Coupled Multiwavelet Operator Learning for Coupled Differential EquationsXiongye Xiao, Defu Cao, Ruochen Yang, Gaurav Gupta et al.ICLR 2023 · 2 citations

Related papers

  1. Efficient Training of Neural Fractional-Order Differential Equation via Adjoint BackpropagationQiyu Kang, Xuhao Li, Kai Zhao, Wenjun Cui et al.AAAI 2025 · 6 citations
  2. Neural Fractional Attention Differential EquationsQiyu Kang, Wenjun Cui, Xuhao Li, Yuxin Ma et al.NeurIPS 2025 · 3 citations
  3. Fractional Deep Reinforcement Learning for Age-Minimal Mobile Edge ComputingLyudong Jin, Ming Tang, Meng Zhang, Hao WangAAAI 2024 · 10 citations
  4. Second-Order Neural ODE OptimizerGuan-Horng Liu, Tianrong Chen, Evangelos A. TheodorouNeurIPS 2021 · 20 citations
  5. The Power of Learned Locally Linear Models for Nonlinear Policy OptimizationDaniel Pfrommer, Max Simchowitz, Tyler Westenbroek, Nikolai Matni et al.ICML 2023 · 4 citations
Dusk over the sea between two cliffs drawn in fine vertical lines