Learning Local Causal World Models with State Space Models and Attention

Francesco Petri, Luigi Asprino, Aldo Gangemi

World modelling, i.e. building a representation of the rules that govern the world so as to predict its evolution, is an essential ability for any agent interacting with the physic…