Next Embedding Prediction Makes World Models Stronger

Researchers introduce NE-Dreamer, a decoder-free model-based reinforcement learning agent that uses temporal transformers to predict next-step encoder embeddings. The approach achieves performance matching or exceeding DreamerV3 on standard benchmarks while showing substantial improvements on memory and spatial reasoning tasks.