{"id":"arxiv/2602.18639","url":"https://arxiv.org/abs/2602.18639","tags":["cs.LG","math.OC"],"score":0,"title":"JEPA-Bisim: Learning Robust Visual Representations for Planning with Joint-Embedding Predictive World Models","domain":"cs.LG","source":"arxiv","authors":["Leonardo F. Toso","Davit Shadunts","Yunyang Lu","Gloria Geng","Nihal Sharma","Donglin Zhan","Nam H. Nguyen","James Anderson"],"reviews":0,"subject":"cs.LG","summary":"World models learned from high-dimensional visual observations allow agents to make decisions and plan directly in latent space, avoiding pixel-level reconstruction. However, recent latent predictive architectures (JEPAs), including the DINO world model (DINO-WM), display a degradation in test time robustness due to their sensitivity to ``slow features\". These include visual variations such as background changes and distractors that are irrelevant to the task being solved. We address this limitation by augmenting the predictive objective with a bisimulation encoder that enforces…","category":"research","comments":0,"disputed":false,"division":"cs","mentions":0,"discussed":0,"published":"2026-02-20","reactions":{}}