Spatial Broadcast Decoder: A Simple Architecture for Learning Disentangled Representations in VAEs
Not on the map. Why?
Abstract (arXiv)
We present a simple neural rendering architecture that helps variational autoencoders (VAEs) learn disentangled representations. Instead of the deconvolutional network typically used in the decoder of VAEs, we tile (broadcast) the latent vector across space, concatenate fixed X- and Y-"coordinate" channels, and apply a fully convolutional network with 1x1 stride. This provides an architectural prior for dissociating positional from non-positional features in the latent distribution of VAEs, yet without providing any explicit supervision to this effect. We show that this architecture, which we term the Spatial Broadcast decoder, improves disentangling, reconstruction accuracy, and generalization to held-out regions in data space. It provides a particularly dramatic benefit when applied to datasets with small objects. We also emphasize a method for visualizing learned latent spaces that helped us diagnose our models and may prove useful for others aiming to assess data representations. Finally, we show the Spatial Broadcast Decoder is complementary to state-of-the-art (SOTA) disentangling techniques and when incorporated improves their performance.
Key facts. Cited by 116 papers in the map (data through 7 Oct 2026). First posted on arXiv 21 Jan 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2019: 1; 2020: 1; 2021: 21; 2022: 14; 2023: 26; 2024: 25; 2025: 19; 2026: 9.
116 papers.
Sort:
By area:
- 2021Conditional Object-Centric Learning from Videocited by 247
- 2022Bridging the Gap to Real-World Object-Centric Learningcited by 219
- 2021NEAT: Neural Attention Fields for End-to-End Autonomous Drivingcited by 174
- 2021Illiterate DALL-E Learns to Composecited by 147
- 2021GENESIS-V2: Inferring Unordered Object Representations without Iterative Refinementcited by 109
- 2021ClevrTex: A Texture-Rich Benchmark for Unsupervised Multi-Object Segmentationcited by 93
- 2022Object Scene Representation Transformercited by 90
- 2023SlotDiffusion: Object-Centric Generative Modeling with Diffusion Modelscited by 85
- 2021Generalization and Robustness Implications in Object-Centric Learningcited by 81
- 2023SODA: Bottleneck Diffusion Models for Representation Learningcited by 76
- 2021SIMONe: View-Invariant, Temporally-Abstracted Object Representations via Unsupervised Video Decompositioncited by 69
- 2022Improving Object-centric Learning with Query Optimizationcited by 60
38 more
- 2023Invariant Slot Attention: Object Discovery with Slot-Centric Reference Framescited by 51
- 2021Unsupervised Layered Image Decomposition into Object Prototypescited by 50
- 2023Self-supervised Object-Centric Learning for Videoscited by 49
- 2022Object Representations as Fixed Points: Training Iterative Refinement Algorithms with Implicit Differentiationcited by 46
- 2021Learning Object-Centric Representations of Multi-Object Scenes from Multiple Viewscited by 37
- 2022Neural Systematic Bindercited by 37
- 2023Crossway Diffusion: Improving Diffusion-based Visuomotor Policy via Self-supervised Learningcited by 37
- 2021Efficient Iterative Amortized Inference for Learning Symmetric and Disentangled Multi-Object Representationscited by 36
- 2024Temporally Consistent Object-Centric Learning by Contrasting Slotscited by 30
- 2021Learning to Give Checkable Answers with Prover-Verifier Gamescited by 29
- 2022Compositional Scene Representation Learning via Reconstruction: A Surveycited by 27
- 2021Towards Building A Group-based Unsupervised Representation Disentanglement Frameworkcited by 22
- 2023Slot-VAE: Object-Centric Scene Generation with Slot Attentioncited by 20
- 2023Learning to reason over visual objectscited by 19
- 2023Systematic Visual Reasoning through Object-Centric Relational Abstractioncited by 19
- 2024Interaction Asymmetry: A General Principle for Learning Composable Abstractionscited by 19
- 2021Local Disentanglement in Variational Auto-Encoders Using Jacobian L1 Regularizationcited by 17
- 2023Shepherding Slots to Objects: Towards Stable and Robust Object-Centric Learningcited by 17
- 2024Slot State Space Modelscited by 17
- 2023Contrastive Training of Complex-Valued Autoencoders for Object Discoverycited by 16
- 2023DORSal: Diffusion for Object-centric Representations of Scenes et alcited by 16
- 2025Object-Centric World Model for Language-Guided Manipulationcited by 16
- 2021Group Equivariant Subsamplingcited by 15
- 2024PLOT: Text-based Person Search with Part Slot Attention for Corresponding Part Discoverycited by 15
- 2022Unsupervised Learning of Temporal Abstractions with Slot-based Transformerscited by 14
- 2023Neurosymbolic Grounding for Compositional World Modelscited by 14
- 2024Exploring the Effectiveness of Object-Centric Representations in Visual Question Answering: Comparative Insights with Foundation Modelscited by 14
- 2021Reasoning-Modulated Representationscited by 13
- 2025Slot-Guided Adaptation of Pre-trained Diffusion Models for Object-Centric Learning and Compositional Generationcited by 13
- 2022Lost in Latent Space: Disentangled Models and the Challenge of Combinatorial Generalisationcited by 11
- 2023Object-Centric Learning with Slot Mixture Modulecited by 11
- 2025Vector-Quantized Vision Foundation Models for Object-Centric Learningcited by 11
- 2025MetaSlot: Break Through the Fixed Number of Slots in Object-Centric Learningcited by 11
- 2021Object-Centric Representation Learning with Generative Spatial-Temporal Factorizationcited by 10
- 2024Slot Abstractors: Toward Scalable Abstract Visual Reasoningcited by 10
- 2021Raven's Progressive Matrices Completion with Latent Gaussian Process Priorscited by 9
- 2022Test-time Adaptation with Slot-Centric Modelscited by 9
- 2024Neural Language of Thought Modelscited by 9
50 most cited of 116 All 116 (JSON)
- 2026Object-Centric Conditioning for Visuomotor Flow Matching21 Sep 2026
- 2026Weakly supervised concept Bottleneck Learning for Robust Two stage Object centric visual reasoning23 Aug 2026
- 2026Semantic Slots for Video Object-Centric Learning21 Aug 2026
- 2026Slot-RAE: Streamlining Object-Centric Learning via Direct Representation Auto-Encoders13 Jul 2026
- 2026Rethinking Object-Centric Representations for Video Dynamics Modeling22 Jun 2026
- 2026Loki: Representation over Architecture for Diffusion-Based Portrait Animation22 May 2026
- 2026Slot-MPC: Goal-Conditioned Model Predictive Control with Object-Centric Representations14 May 2026
- 2026Weakly Supervised Concept Learning for Object-centric Visual Reasoning5 May 2026
- 2026Improved Object-Centric Diffusion Learning with Registers and Contrastive Alignment3 Jan 2026
- 2025Independent Density Estimation10 Dec 2025
- 2025Unsupervised Structural Scene Decomposition via Foreground-Aware Slot Attention with Pseudo-Mask Guidance2 Dec 2025
- 2025Learning Visually Interpretable Oscillator Networks for Soft Continuum Robots from Video23 Nov 2025
38 more
- 2025Object-Centric World Models for Causality-Aware Reinforcement Learning18 Nov 2025
- 2025Disentangled Representation Learning via Modular Compositional Bias24 Oct 2025
- 2025Local Mechanisms of Compositional Generalization in Conditional Diffusion19 Sep 2025
- 2025Future Slot Prediction for Unsupervised Object Discovery in Surgical Video2 Jul 2025
- 2025Farm-Level, In-Season Crop Identification for India30 Jun 2025
- 2025CoLa: Chinese Character Decomposition with Compositional Latent Components4 Jun 2025
- 2025MetaSlot: Break Through the Fixed Number of Slots in Object-Centric Learning27 May 2025
- 2025An Interpretable Representation Learning Approach for Diffusion Tensor Imaging25 May 2025
- 2025Learning to Adapt to Position Bias in Vision Transformer Classifiers19 May 2025
- 2025Hierarchical Compact Clustering Attention (COCA) for Unsupervised Object-Centric Learning4 May 2025
- 2025Object-Centric World Model for Language-Guided Manipulation8 Mar 2025
- 2025Vector-Quantized Vision Foundation Models for Object-Centric Learning27 Feb 2025
- 2025TextOCVP: Object-Centric Video Prediction with Language Guidance17 Feb 2025
- 2025Multiple Invertible and Partial-Equivariant Function for Latent Vector Transformation to Enhance Disentanglement in VAEs6 Feb 2025
- 2025Slot-Guided Adaptation of Pre-trained Diffusion Models for Object-Centric Learning and Compositional Generation27 Jan 2025
- 2025Slot-BERT: Self-supervised Object Discovery in Surgical Video21 Jan 2025
- 2024Temporally Consistent Object-Centric Learning by Contrasting Slots18 Dec 2024
- 2024Interaction Asymmetry: A General Principle for Learning Composable Abstractions12 Nov 2024
- 2024Unsupervised Object Discovery: A Comprehensive Survey and Unified Taxonomy30 Oct 2024
- 2024FACTS: A Factored State-Space Framework For World Modelling28 Oct 2024
- 2024Disentangling Genotype and Environment Specific Latent Features for Improved Trait Prediction using a Compositional Autoencoder25 Oct 2024
- 2024Object-Centric Temporal Consistency via Conditional Autoregressive Inductive Biases21 Oct 2024
- 2024Simplified priors for Object-Centric Learning1 Oct 2024
- 2024Transferring disentangled representations: bridging the gap between synthetic and real images26 Sep 2024
- 2024Multi-Modality Conditioned Variational U-Net for Field-of-View Extension in Brain Diffusion MRI20 Sep 2024
- 2024PLOT: Text-based Person Search with Part Slot Attention for Corresponding Part Discovery20 Sep 2024
- 2024Rethinking Disentanglement under Dependent Factors of Variation13 Aug 2024
- 2024Exploring the Effectiveness of Object-Centric Representations in Visual Question Answering: Comparative Insights with Foundation Models22 Jul 2024
- 2024Attention Normalization Impacts Cardinality Generalization in Slot Attention4 Jul 2024
- 2024Learning Disentangled Representation in Object-Centric Models for Visual Dynamics Prediction via Transformers3 Jul 2024
- 2024ViPro: Enabling and Controlling Video Prediction for Complex Dynamical Scenarios using Procedural Knowledge26 Jun 2024
- 2024Guiding Video Prediction with Explicit Procedural Knowledge26 Jun 2024
- 2024Slot State Space Models18 Jun 2024
- 2024Identifiable Object-Centric Representation Learning via Probabilistic Slot Attention11 Jun 2024
- 2024Masked Multi-Query Slot Attention for Unsupervised Object Discovery30 Apr 2024
- 2024Reasoning-Enhanced Object-Centric Learning for Videos22 Mar 2024
- 2024Slot Abstractors: Toward Scalable Abstract Visual Reasoning6 Mar 2024
- 2024Visual Reasoning in Object-Centric Deep Neural Networks: A Comparative Cognition Approach20 Feb 2024
50 newest of 116 All 116 (JSON)
- 2023SlotDiffusion: Object-Centric Generative Modeling with Diffusion Modelscited by 85
- 2023SODA: Bottleneck Diffusion Models for Representation Learningcited by 76
- 2023Crossway Diffusion: Improving Diffusion-based Visuomotor Policy via Self-supervised Learningcited by 37
- 2023DORSal: Diffusion for Object-centric Representations of Scenes et alcited by 16
- 2025Slot-Guided Adaptation of Pre-trained Diffusion Models for Object-Centric Learning and Compositional Generationcited by 13
- 2025Local Mechanisms of Compositional Generalization in Conditional Diffusioncited by 3
- 2026Improved Object-Centric Diffusion Learning with Registers and Contrastive Alignmentcited by 3
- 2026Slot-RAE: Streamlining Object-Centric Learning via Direct Representation Auto-Encoderscited by 0
- 2026Object-Centric Conditioning for Visuomotor Flow Matchingcited by 0
- 2026Object-Centric Conditioning for Visuomotor Flow Matching21 Sep 2026
- 2026Slot-RAE: Streamlining Object-Centric Learning via Direct Representation Auto-Encoders13 Jul 2026
- 2026Improved Object-Centric Diffusion Learning with Registers and Contrastive Alignment3 Jan 2026
- 2025Local Mechanisms of Compositional Generalization in Conditional Diffusion19 Sep 2025
- 2025Slot-Guided Adaptation of Pre-trained Diffusion Models for Object-Centric Learning and Compositional Generation27 Jan 2025
- 2023SODA: Bottleneck Diffusion Models for Representation Learning29 Nov 2023
- 2023Crossway Diffusion: Improving Diffusion-based Visuomotor Policy via Self-supervised Learning4 Jul 2023
- 2023DORSal: Diffusion for Object-centric Representations of Scenes et al13 Jun 2023
- 2023SlotDiffusion: Object-Centric Generative Modeling with Diffusion Models18 May 2023
- 2023Learning to reason over visual objectscited by 19
- 2023Systematic Visual Reasoning through Object-Centric Relational Abstractioncited by 19
- 2024Slot Abstractors: Toward Scalable Abstract Visual Reasoningcited by 10
- 2024Neural Language of Thought Modelscited by 9
- 2024Reasoning-Enhanced Object-Centric Learning for Videoscited by 7
- 2024Visual Reasoning in Object-Centric Deep Neural Networks: A Comparative Cognition Approachcited by 1
- 2026Weakly Supervised Concept Learning for Object-centric Visual Reasoningcited by 0
- 2026Weakly supervised concept Bottleneck Learning for Robust Two stage Object centric visual reasoningcited by 0
- 2026Weakly supervised concept Bottleneck Learning for Robust Two stage Object centric visual reasoning23 Aug 2026
- 2026Weakly Supervised Concept Learning for Object-centric Visual Reasoning5 May 2026
- 2024Reasoning-Enhanced Object-Centric Learning for Videos22 Mar 2024
- 2024Slot Abstractors: Toward Scalable Abstract Visual Reasoning6 Mar 2024
- 2024Visual Reasoning in Object-Centric Deep Neural Networks: A Comparative Cognition Approach20 Feb 2024
- 2024Neural Language of Thought Models2 Feb 2024
- 2023Systematic Visual Reasoning through Object-Centric Relational Abstraction4 Jun 2023
- 2023Learning to reason over visual objects3 Mar 2023
- 2021Conditional Object-Centric Learning from Videocited by 247
- 2025Object-Centric World Model for Language-Guided Manipulationcited by 16
- 2023Neurosymbolic Grounding for Compositional World Modelscited by 14
- 2024FACTS: A Factored State-Space Framework For World Modellingcited by 4
- 2025Object-Centric World Models for Causality-Aware Reinforcement Learningcited by 3
- 2026Slot-MPC: Goal-Conditioned Model Predictive Control with Object-Centric Representationscited by 3
- 2026Slot-MPC: Goal-Conditioned Model Predictive Control with Object-Centric Representations14 May 2026
- 2025Object-Centric World Models for Causality-Aware Reinforcement Learning18 Nov 2025
- 2025Object-Centric World Model for Language-Guided Manipulation8 Mar 2025
- 2024FACTS: A Factored State-Space Framework For World Modelling28 Oct 2024
- 2023Neurosymbolic Grounding for Compositional World Models19 Oct 2023
- 2021Conditional Object-Centric Learning from Video24 Nov 2021
- 2021NEAT: Neural Attention Fields for End-to-End Autonomous Drivingcited by 174
- 2025Independent Density Estimationcited by 0
- 2025Independent Density Estimation10 Dec 2025
- 2024Slot State Space Modelscited by 17
- 2024Slot State Space Models18 Jun 2024
- 2022Bridging the Gap to Real-World Object-Centric Learningcited by 219
- 2021Illiterate DALL-E Learns to Composecited by 147
- 2021GENESIS-V2: Inferring Unordered Object Representations without Iterative Refinementcited by 109
- 2021ClevrTex: A Texture-Rich Benchmark for Unsupervised Multi-Object Segmentationcited by 93
- 2022Object Scene Representation Transformercited by 90
- 2021Generalization and Robustness Implications in Object-Centric Learningcited by 81
- 2021SIMONe: View-Invariant, Temporally-Abstracted Object Representations via Unsupervised Video Decompositioncited by 69
- 2022Improving Object-centric Learning with Query Optimizationcited by 60
- 2023Invariant Slot Attention: Object Discovery with Slot-Centric Reference Framescited by 51
- 2021Unsupervised Layered Image Decomposition into Object Prototypescited by 50
- 2023Self-supervised Object-Centric Learning for Videoscited by 49
- 2022Object Representations as Fixed Points: Training Iterative Refinement Algorithms with Implicit Differentiationcited by 46
12 most cited of 88 All 88 (JSON)
- 2026Semantic Slots for Video Object-Centric Learning21 Aug 2026
- 2026Rethinking Object-Centric Representations for Video Dynamics Modeling22 Jun 2026
- 2026Loki: Representation over Architecture for Diffusion-Based Portrait Animation22 May 2026
- 2025Unsupervised Structural Scene Decomposition via Foreground-Aware Slot Attention with Pseudo-Mask Guidance2 Dec 2025
- 2025Learning Visually Interpretable Oscillator Networks for Soft Continuum Robots from Video23 Nov 2025
- 2025Disentangled Representation Learning via Modular Compositional Bias24 Oct 2025
- 2025Future Slot Prediction for Unsupervised Object Discovery in Surgical Video2 Jul 2025
- 2025Farm-Level, In-Season Crop Identification for India30 Jun 2025
- 2025CoLa: Chinese Character Decomposition with Compositional Latent Components4 Jun 2025
- 2025MetaSlot: Break Through the Fixed Number of Slots in Object-Centric Learning27 May 2025
- 2025An Interpretable Representation Learning Approach for Diffusion Tensor Imaging25 May 2025
- 2025Learning to Adapt to Position Bias in Vision Transformer Classifiers19 May 2025
12 newest of 88 All 88 (JSON)