Saved in:
Bibliographic Details
Main Authors: Wu, Sichao, Wu, Jiang, Cao, Xingyu, Zhang, Fawang, Yu, Guangyuan, Zhao, Junjie, Qu, Yue, Ma, Fei, Duan, Jingliang
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2509.07381
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918138114736128
author Wu, Sichao
Wu, Jiang
Cao, Xingyu
Zhang, Fawang
Yu, Guangyuan
Zhao, Junjie
Qu, Yue
Ma, Fei
Duan, Jingliang
author_facet Wu, Sichao
Wu, Jiang
Cao, Xingyu
Zhang, Fawang
Yu, Guangyuan
Zhao, Junjie
Qu, Yue
Ma, Fei
Duan, Jingliang
contents Traditional online Model Predictive Control (MPC) methods often suffer from excessive computational complexity, limiting their practical deployment. Explicit MPC mitigates online computational load by pre-computing control policies offline; however, existing explicit MPC methods typically rely on simplified system dynamics and cost functions, restricting their accuracy for complex systems. This paper proposes TransMPC, a novel Transformer-based explicit MPC algorithm capable of generating highly accurate control sequences in real-time for complex dynamic systems. Specifically, we formulate the MPC policy as an encoder-only Transformer leveraging bidirectional self-attention, enabling simultaneous inference of entire control sequences in a single forward pass. This design inherently accommodates variable prediction horizons while ensuring low inference latency. Furthermore, we introduce a direct policy optimization framework that alternates between sampling and learning phases. Unlike imitation-based approaches dependent on precomputed optimal trajectories, TransMPC directly optimizes the true finite-horizon cost via automatic differentiation. Random horizon sampling combined with a replay buffer provides independent and identically distributed (i.i.d.) training samples, ensuring robust generalization across varying states and horizon lengths. Extensive simulations and real-world vehicle control experiments validate the effectiveness of TransMPC in terms of solution accuracy, adaptability to varying horizons, and computational efficiency.
format Preprint
id arxiv_https___arxiv_org_abs_2509_07381
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle TransMPC: Transformer-based Explicit MPC with Variable Prediction Horizon
Wu, Sichao
Wu, Jiang
Cao, Xingyu
Zhang, Fawang
Yu, Guangyuan
Zhao, Junjie
Qu, Yue
Ma, Fei
Duan, Jingliang
Robotics
Traditional online Model Predictive Control (MPC) methods often suffer from excessive computational complexity, limiting their practical deployment. Explicit MPC mitigates online computational load by pre-computing control policies offline; however, existing explicit MPC methods typically rely on simplified system dynamics and cost functions, restricting their accuracy for complex systems. This paper proposes TransMPC, a novel Transformer-based explicit MPC algorithm capable of generating highly accurate control sequences in real-time for complex dynamic systems. Specifically, we formulate the MPC policy as an encoder-only Transformer leveraging bidirectional self-attention, enabling simultaneous inference of entire control sequences in a single forward pass. This design inherently accommodates variable prediction horizons while ensuring low inference latency. Furthermore, we introduce a direct policy optimization framework that alternates between sampling and learning phases. Unlike imitation-based approaches dependent on precomputed optimal trajectories, TransMPC directly optimizes the true finite-horizon cost via automatic differentiation. Random horizon sampling combined with a replay buffer provides independent and identically distributed (i.i.d.) training samples, ensuring robust generalization across varying states and horizon lengths. Extensive simulations and real-world vehicle control experiments validate the effectiveness of TransMPC in terms of solution accuracy, adaptability to varying horizons, and computational efficiency.
title TransMPC: Transformer-based Explicit MPC with Variable Prediction Horizon
topic Robotics
url https://arxiv.org/abs/2509.07381