arXiv Machine Learning By Morgan Byrd, Robert Wright, Sehoon Ha

CF-JEPA: Improving Robustness of JEPA World Models via Controllability Factorization

Read the original on arXiv Machine Learning →

CF-JEPA is a JEPA-style latent world model that separates the latent space into controllable and uncontrollable subspaces, allowing distractor information to be captured in the uncontrollable region. This factorization prevents latent collapse and maintains performance across 2D and 3D control tasks, even under distracted conditions. The model is validated on a simulated robot task, demonstrating its practical applicability.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Machine Learning.

arXiv AI
Aug 10

TaskSense: Focusing on What Matters in World Models

arXiv:2608. 06544v1 Announce Type: new Abstract: World models for visual control typically learn compact latent states by reconstructing observations, implicitly encouraging representations to preserve information across the entire visual input.

By SM Mazharul Islam, Manfred Huber
arXiv Machine Learning
Sep 22

Contrastive World Models

Contrastive World Models propose a new method for learning latent dynamics without pixel reconstruction. By replacing observation reconstruction with a Deep InfoMax-like objective that maximizes mutual information between state-action sequences and local patch features of future observations, the approach encourages state representations to retain predictive information while ignoring visually irrelevant details. Experiments show that this method matches existing baselines in simple settings and significantly outperforms them when distractors or natural video backgrounds are present, while also training more efficiently by eliminating the pixel decoder.

By Bonnie Li
arXiv Machine Learning
3d ago

JEPA-Bisim: Learning Robust Visual Representations for Planning with Joint-Embedding Predictive World Models

JEPA-Bisim introduces a bisimulation encoder to joint-embedding predictive world models, ensuring that states with similar transition dynamics are mapped to nearby latent representations while suppressing irrelevant slow features such as background changes and distractors. The approach improves robustness on navigation (PointMaze) and manipulation (PushT) tasks under varied test-time visual conditions, achieving up to tenfold smaller latent spaces than DINO-WM. It remains effective across different pre-trained visual encoders, including DINOv2, SimDINOv2, and iBOT.

By Leonardo F. Toso, Davit Shadunts, Yunyang Lu, Gloria Geng, Nihal Sharma, Donglin Zhan, Nam H. Nguyen, James Anderson
arXiv AI
Aug 10

Dueling World Models: Advantage-Style Action Channels for Common-Mode Distractor Rejection

arXiv:2608. 06706v1 Announce Type: cross Abstract: Latent world models plan by predicting future states from an action, but when a scene contains motion the agent does not control, they quietly go action-blind: predictions for different actions become indistinguishable even as the training loss keeps improving.

By Jiazhuo Li, Yiming Fei, Zhiruo Zhou, Heikichi Hayashi