arXiv Machine Learning By Amir Mesbah, Di Wu

Before We Adapt: The Zero-Shot Generalization Bottleneck in Continual World Models

Read the original on arXiv Machine Learning →

The paper investigates how model-based reinforcement learning can support continual learning by reusing prior knowledge. It introduces a contextual Markov decision process that separates behavioral and physical context, and studies zero‑shot reuse in FourRooms tasks using Dyna‑Q and TD‑MPC. The experiments show that zero‑shot reuse degrades under physical context changes and that this degradation is strongly tied to the observation representation, highlighting representation generalization as a key bottleneck.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Machine Learning.

arXiv Machine Learning
Sep 21

Benchmarking World Models for Continual Learning on Compositional Tasks

The paper introduces a compositional continual learning benchmark for world models in robot manipulation, designed to isolate knowledge reuse from learning speed and capacity. Tasks are curated to combine previously seen action and perception components, allowing analysis of how different modalities affect reuse. Experiments show that modular world models better balance reuse and forgetting than conventional methods, yet none fully solve the challenge, highlighting the need for models explicitly built to reuse knowledge without forgetting.

By Haoyu Zhou, Joe Watson, Anson Lei, Ingmar Posner
arXiv AI
Oct 5

How to Find and Reuse Policies for Continuous Adaptation in Lifelong Reinforcement Learning

The paper introduces Adaptive Mask Selection and Composition (AMSC), a method that estimates task similarity using non‑parametric Wasserstein embeddings from state‑action‑reward samples. AMSC selects and weights previously learned policies based on z‑score‑normalized sparsemax similarity scores, forming a prior for new tasks. Experiments on CT‑graph, MiniGrid, and Continual World show that AMSC improves mean performance and forward transfer while avoiding forgetting, highlighting the importance of selecting relevant prior knowledge and its layer‑specific composition.

By Saptarshi Nath, Inish M. D'Souza, Antonio Carta, Soheil Kolouri, Andrea Soltoggio