Light Cones For Vision: Simple Causal Priors For Visual Hierarchy
Not on the map. Why?
Abstract (arXiv)
Standard vision models treat objects as independent points in Euclidean space, unable to capture hierarchical structure like parts within wholes. We introduce Worldline Slot Attention, which models objects as persistent trajectories through spacetime worldlines, where each object has multiple slots at different hierarchy levels sharing the same spatial position but differing in temporal coordinates. This architecture consistently fails without geometric structure: Euclidean worldlines achieve 0.078 level accuracy, below random chance (0.33), while Lorentzian worldlines achieve 0.479-0.661 across three datasets: a 6x improvement replicated over 20+ independent runs. Lorentzian geometry also outperforms hyperbolic embeddings showing visual hierarchies require causal structure (temporal dependency) rather than tree structure (radial branching). Our results demonstrate that hierarchical object discovery requires geometric structure encoding asymmetric causality, an inductive bias absent from Euclidean space but natural to Lorentzian light cones, achieved with only 11K parameters. The code is available at: https://github.com/iclrsubmissiongram/loco.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 10 of its 14 references that are in the map. First posted on arXiv 25 Mar 2026.
Builds on
10 of 14 references.
References · 10
- 2016CLEVR: A Diagnostic Dataset for Compositional Language and Elementary Visual Reasoningcited by 1,457
- 2020Object-Centric Learning with Slot Attentioncited by 876
- 2017Dynamic Routing Between Capsulescited by 689
- 2017Poincar'e Embeddings for Learning Hierarchical Representationscited by 684
- 2019MONet: Unsupervised Scene Decomposition and Representationcited by 392
- 2019Multi-Object Representation Learning with Iterative Variational Inferencecited by 360
- 2021Conditional Object-Centric Learning from Videocited by 248
- 2018Hyperbolic Entailment Cones for Learning Hierarchical Embeddingscited by 162
- 2021Illiterate DALL-E Learns to Composecited by 147
- 2023Invariant Slot Attention: Object Discovery with Slot-Centric Reference Framescited by 51
Built on by
Not cited yet.