Discrete JEPA: Learning Discrete Token Representations without Reconstruction

ICML 2025 Workshop TokShop Submission44 Authors

Archiving Submission: Yes (archival)

Keywords: Representation Learning, Symbolic Planning, Image Tokenization

TL;DR: We introduce Discrete-JEPA, which learns discrete semantic tokens for improved symbolic reasoning and long-horizon planning.

Abstract:

The cornerstone of cognitive intelligence lies in extracting hidden patterns from observations and leveraging these principles to systematically predict future outcomes. However, current image tokenization methods demonstrate significant limitations in tasks requiring symbolic abstraction and logical reasoning capabilities essential for systematic inference. To address this challenge, we propose Discrete-JEPA, extending the latent predictive coding framework with semantic tokenization and novel complementary objectives to create robust tokenization for symbolic reasoning tasks. Discrete-JEPA dramatically outperforms baselines on visual symbolic prediction tasks, while striking visual evidence reveals the spontaneous emergence of deliberate systematic patterns within the learned semantic token space.
Though an initial model, our approach promises a significant impact for advancing Symbolic world modeling and planning capabilities in artificial intelligence systems.

Submission Number: 44