Learning Robust Dynamics through Variational Sparse Gating

Arnav Kumar Jain, Shivakanth Sujit, Shruti Joshi +3

Learning world models from their sensory inputs enables agents to plan for actions by imagining their future outcomes. World models have previously been shown to improve sample-eff…