EpicWorldModel: Exploration-driven Planning with Latent World Models
This study addresses the limitation of deterministic latent world models in capturing multiple future possibilities under partial observability by proposing the Stochastic JEPA framework. This approach integrates Joint Embedding Predictive Architectures with flow matching to predict diverse future states, optimized via cross-entropy loss. Furthermore, it introduces a novel exploration mechanism based on flow prediction variance, leveraging uncertainty signals to guide agents in balancing goal achievement with the exploration of occluded regions. Experimental results demonstrate that the proposed framework achieves state-of-the-art or comparable performance across multiple tasks, improving success rates by up to 22% over baselines.