Variational Causal Dynamics: Discovering Modular World Models from Interventions. - Anson Lei; Bernhard Schölkopf; Ingmar Posner (Trans. Mach. Learn. Res., 2023)

Compete and Compose: Learning Independent Mechanisms for Modular World Models. - Anson Lei; Frederik Nolte; Bernhard Schölkopf; Ingmar Posner (CoRR, 2024)

SPARTAN: A Sparse Transformer Learning Local Causation. - Anson Lei; Bernhard Schölkopf; Ingmar Posner (CoRR, 2024)