1: \begin{abstract}
2: Model predictive control (MPC) has established itself as the primary methodology for constrained control, enabling general-purpose robot autonomy in diverse real-world scenarios.
3: However, for most problems of interest, MPC relies on the recursive solution of highly non-convex trajectory optimization problems, leading to high computational complexity and strong dependency on initialization.
4: In this work, we present a unified framework to combine the main strengths of optimization-based and learning-based methods for MPC.
5: Our approach entails embedding high-capacity, transformer-based neural network models within the optimization process for trajectory generation, whereby the transformer provides a near-optimal initial guess, or target plan, to a non-convex optimization problem.
6: %leading to substantially improved performance and computation time.
7: Our experiments, performed in simulation and the real world onboard a free flyer platform, demonstrate the capabilities of our framework to improve MPC convergence and runtime.
8: %, allowing for real-time execution while providing the safety assurances needed for reliable operations.
9: Compared to purely optimization-based approaches, results show that our approach can improve trajectory generation performance by up to 75\%, reduce the number of solver iterations by up to 45\%, and improve overall MPC runtime by 7x without loss in performance.
10:
11: \noindent Project website and code: \href{https://transformermpc.github.io}{https://transformermpc.github.io}%\url{https://transformermpc.github.io}
12: \end{abstract}
13: