AI & ML
MA-JEPA: Joint-Embedding World Models for Multi-Agent Reinforcement Learning
arXiv cs.MA0 views
arXiv:2609.33563v2 Announce Type: replace-cross Abstract: World models improve sample efficiency by training policies on imagined trajectories, but their usefulness depends on learning representations that capture the information needed for future control. We study whether self-supervised joint-embedding prediction (JEPA) can provide this learning signal for multi-agent reinforcement learning. We introduce MA-JEPA, a stochastic world model that replaces observation reconstruction with prediction of target representations, enabling model-based multi-agent reinforcement learning with centralized training and decentralized execution. A categorical latent state and a causal Transformer are trained with posterior and action-conditioned dynamics prediction objectives and are then used for actor-critic learning from latent imagination. A training-only joint predictor conditions on all agents' local states and actions to predict each agent's next local observation embedding. These predictions are passed through the same local posterior used during real interaction with a centralized critic that is used only for value learning, with execution remaining decentralized. Our experiments show that this architecture performs strongly on SMAC, matching or exceeding the strongest reported comparator mean win rate on four of eight evaluated maps.
Source:https://arxiv.org/abs/2609.33563PreviousBeyond Component Testing: Validating Agentic AI SystemsNextCharacterizing the Narrowband Ambiguity Function of Frequency Modulated OFDM Waveforms
Related posts
Traffic Dropped Overnight? A Diagnosis Checklist For Affiliate Sites After A Google Update
0dev.to Beginners
15 Best Free AI Tools in 2026 (Actually Free, Tested)
0dev.to Beginners
Building Raasta: Flooded-Road Detection with Satellite Images and PyTorch
0dev.to Beginners
5 Google Business Profile Quick Wins for Small Businesses (No Ads Needed)
0dev.to Beginners
How To Recover An Affiliate Site After A Google Spam Update
0dev.to Beginners
When a PDF Export Turns Every Page Into a Picture, Define the Text Contract First
0dev.to Testing