Leveraging Offline Data from Similar Systems for Online Linear Quadratic Control
Shivam Bajaj, Prateek Jaiswal, Vijay Gupta
TL;DR
This work tackles the sim2real gap in online linear quadratic control by leveraging offline data from a similar, but unknown, system. It introduces a Thompson Sampling-based algorithm (TSOD-LQR) that fuses offline trajectory information with online observations, providing Bayesian regret guarantees that scale as $\tilde{\mathcal{O}}(f(S,M_{\delta})\sqrt{T/S})$, where $S$ is the offline trajectory length and $M_{\delta}$ bounds system dissimilarity. Theoretical results show improved regret when the offline source is similar ($M_{\delta}$ small) and demonstrate extensions to multiple offline sources. Numerical experiments corroborate the benefits of incorporating offline data, with regret decreasing as $S$ grows and outperformance over naive online-only strategies. The framework offers a practical route for safer, faster online adaptation in real-world systems by exploiting available simulations or simpler models.
Abstract
``Sim2real gap", in which the system learned in simulations is not the exact representation of the real system, can lead to loss of stability and performance when controllers learned using data from the simulated system are used on the real system. In this work, we address this challenge in the linear quadratic regulator (LQR) setting. Specifically, we consider an LQR problem for a system with unknown system matrices. Along with the state-action pairs from the system to be controlled, a trajectory of length $S$ of state-action pairs from a different unknown system is available. Our proposed algorithm is constructed upon Thompson sampling and utilizes the mean as well as the uncertainty of the dynamics of the system from which the trajectory of length $S$ is obtained. We establish that the algorithm achieves $\tilde{\mathcal{O}}({f(S,M_δ)\sqrt{T/S}})$ Bayes regret after $T$ time steps, where $M_δ$ characterizes the \emph{dissimilarity} between the two systems and $f(S,M_δ)$ is a function of $S$ and $M_δ$. When $M_δ$ is sufficiently small, the proposed algorithm achieves $\tilde{\mathcal{O}}({\sqrt{T/S}})$ Bayes regret and outperforms a naive strategy which does not utilize the available trajectory.
