## Results on Shapes2D

## Results on CelebA

## Results on CLEVRER-Kubric

We show results below on level-2 compositions (large red cube). The training data has the base concepts encoded separately, with different shapes taking the red color (leaving cubes to be only green) and different sizes.

### Training Data

## Results with MaskGIT (Standard Training Objective)

## Results with DiT

## Results with MaskGIT (JEPA-based Training Objective)