CausalWAM: Learning the Causes of Actions in World Action Models
CWAM (Causal WAM), which incorporates the sparsity constraint into a pretrained WAM by adding a sparse routing path to its action head, with the world modeling objective remaining unchanged, is proposed.