Disentangled World Models: Learning to Transfer Semantic Knowledge from Distracting Videos for Reinforcement Learning
TLDR
Disentangled World Models transfer semantic knowledge from distracting videos to RL via offline-to-online latent distillation and disentanglement constraints.
Reasoning
The paper presents a novel method for transferring semantic knowledge from distracting videos to improve RL sample efficiency, with a clear methodology combining pretraining and distillation. However, the reliance on pretrained video prediction and benchmark evaluations may limit generalizability to real-world scenarios.
Read-first score
Read-first score 54.8, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 38.
Field roles
Rank sensitivity
Stability: volatile; rank range: 287.
Keyword Scores
Deep Analysis
Innovations
- Offline-to-online latent distillation for transferring semantic knowledge from distracting videos to world models
- Flexible disentanglement constraints to enable cross-domain semantic knowledge transfer
- Disentangled World Models (DisWM) framework combining pretrained action-free video prediction with online RL finetuning
- Pretraining action-free video prediction model with disentanglement regularization to extract semantic knowledge
- Incorporation of actions and rewards during online adaptation to strengthen disentangled representation learning
Methodology
DisWM pretrains an action-free video prediction model offline using disentanglement regularization to extract semantic knowledge from distracting videos. The pretrained model's disentanglement capability is transferred to a world model via latent distillation. During online finetuning, a disentanglement constraint is applied, and actions and rewards from environment interactions enrich data diversity to strengthen disentangled representation learning.
Key Results
Experimental results validate the superiority of the proposed approach on various benchmarks.