arXiv Machine Learning
Jun 5

LeWorldModel: Stable End-to-End Joint-Embedding Predictive Architecture from Pixels

arXiv:2603. 19312v3 Announce Type: replace Abstract: Joint Embedding Predictive Architectures (JEPAs) offer a compelling framework for learning world models in compact latent spaces, yet existing methods remain fragile, relying on complex multi-term losses, exponential moving averages, pre-trained encoders, or auxiliary supervision to avoid representation collapse.

By Lucas Maes, Quentin Le Lidec, Damien Scieur, Yann LeCun, Randall Balestriero
arXiv Machine Learning
2d ago

Mixed neural posterior estimation for simulators with discrete and continuous parameters

The paper extends Neural Posterior Estimation (NPE) to handle simulators whose parameter spaces contain both discrete and continuous dimensions. It introduces an inference network that factorizes the joint posterior into discrete and continuous components, using an autoregressive classifier for the discrete part and a generative model for the continuous part, trained jointly with a single simulation-based objective. A diagnostic tool for assessing calibration of the mixed posterior is also proposed, and the method is shown to produce accurate, calibrated posteriors on toy and real scientific simulators.

By Jan Boelts, Cornelius Schr\"oder, Jonas Beck, Jakob H. Macke, Michael Deistler, Daniel Gedon