{"id":"arxiv/2603.07437","url":"https://arxiv.org/abs/2603.07437","tags":["cs.LG","eess.SY","math.OC","stat.ML"],"score":0,"title":"Cost-Driven Representation Learning for Linear Quadratic Gaussian Control: Part II","domain":"cs.LG","source":"arxiv","authors":["Yi Tian","Kaiqing Zhang","Russ Tedrake","Suvrit Sra"],"reviews":0,"subject":"cs.LG","summary":"We study the problem of state representation learning for control from partial and potentially high-dimensional observations. We approach this problem via cost-driven state representation learning, in which we learn a dynamical model in a latent state space by predicting cumulative costs. In particular, we establish finite-sample guarantees on finding a near-optimal representation function and a near-optimal controller using the learned latent model for infinite-horizon time-invariant Linear Quadratic Gaussian (LQG) control. We study two approaches to cost-driven representation learning,…","category":"research","comments":0,"disputed":false,"division":"cs","mentions":0,"discussed":0,"published":"2026-03-08","reactions":{}}