distraction-robustness

TaskSense introduces a task-centric world modeling framework that uses stochastic spatial attention conditioned on previous latent states and an auxiliary inverse-dynamics objective to focus on control-relevant regions, improving robustness to visual distractions compared to DreamerV3.