Sub-JEPA Improves LeWM Performance
Key point
Sub-JEPA boosts LeWM performance by splitting the high-dimensional Gaussian constraint and applying it separately.
Details
Sub-JEPA distributes the global isotropic Gaussian prior used in LeCun's group's LeWorldModel(LeWM) across multiple fixed random orthogonal subspaces.
Reflecting the fact that real-world dynamics lie on a low-dimensional manifold, it relaxes the excessive constraint of the global high-dimensional Gaussian while still preserving the anti-collapse effect. There are no additional hyperparameters, and the same two-term objective as before is used unchanged.
The results are as follows.
- Consistently outperformed LeWM across 4 benchmarks.
- Achieved up to +10.7pp improvement on Two-Room.
- Latent trajectories appeared more linear, and physical state decodability also improved.
This summary was generated automatically by AI. Check the original for the author's claims and context. Copyright belongs to the original author.
Our guide explains how the AI works. Report summary errors, attribution issues, or removal requests via Contact.