Logo image
Real-Time Optimal Control via Transformer Networks and Bernstein Polynomials
Preprint   Open access

Real-Time Optimal Control via Transformer Networks and Bernstein Polynomials

Gage MacLin, Venanzio Cichella, Andrew Patterson and Irene Gregory
ArXiv.org
Cornell University
11/19/2025
DOI: 10.48550/arxiv.2511.15588
url
https://doi.org/10.48550/arxiv.2511.15588View
Preprint (Author's original)This preprint has not been evaluated by subject experts through peer review. Preprints may undergo extensive changes and/or become peer-reviewed journal articles. Open Access

Abstract

In this paper, we propose a Transformer-based framework for approximating solutions to infinite-dimensional optimization problems: calculus of variations problems and optimal control problems. Our approach leverages offline training on data generated by solving a sample of infinite- dimensional optimization problems using composite Bernstein collocation. Once trained, the Transformer efficiently generates near-optimal, feasible trajectories, making it well-suited for real-time applications. In motion planning for autonomous vehicles, for instance, these trajectories can serve to warm- start optimal motion planners or undergo rigorous evaluation to ensure safety. We demonstrate the effectiveness of this method through numerical results on a classical control problem and an online obstacle avoidance task. This data-driven approach offers a promising solution for real-time optimal control of nonlinear, nonconvex systems.
Mathematics - Optimization and Control

Details

Metrics

4 Record Views
Logo image