Bilinear World Models: Learning Representations with Structured Dynamics for Efficient Control
This work proposes a JEPA-style world model in which representations with bilinear-parameterized dynamics can be learned directly from high-dimensional observations, reducing planning time by nearly three orders of magnitude while retaining or even improving control accuracy.