cs.LGSep 27, 2026

Behavioral Monitoring of JEPA World Models with Jacobian Centroids

Authors: Thomas Walker, Randall Balestriero, Richard Baraniuk

Organizations: Rice University · Brown University

Abstract

Detecting failures in World Model (WM)-based planning requires monitoring whether the model is behaviorally aligned with the current task, which in turn requires studying its internal representations. Here, we show that centroids---sub-component Jacobian row-sums---effectively identify the behavioral properties of WMs, complementing traditional activation-based knowledge signals. The centroids of a model are easily computed through Jacobian vector products and characterize how the model organizes the geometry of its input space, yielding an efficient perspective on internal representations, including the generation of task-relevant saliency maps. Evaluated on continuous control tasks using JEPA WMs, this behavioral view reveals a structural dissociation, where the encoder correctly represents the goal while the predictor remains behaviorally unresponsive. This failure mode directly predicts planning failure before any action is taken, allowing for goal resampling to recapture out-of-distribution success. Moreover, centroid-based methods outperform baseline methods as distribution-shift detectors. Together, these tools yield a behavioral monitoring stack that is operational and consequential under distribution shifts.

Explore similar work

CardsList
  1. Diagnosing JEPA World Models with Action-Conditioned Predictive Consistency

    Aug 13, 2026Guo An, Zijing Wu, Honghua Dong +7Action PredictionWorld Models

  2. Better Slots, Better Worlds: Representation Quality & Robustness in Object-Centric World Models

    Aug 12, 2026Shukrullo Nazirjonov, Sai Prasanna, Anna Manasyan +1Object-Centric RepresentationsWorld Models

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

    Sep 30, 2026Morgan Byrd, Robert Wright, Sehoon HaLatent World ModelsControllability