EQ-VAE: Equivariance Regularized Latent Space for Improved Generative Image Modeling
Paper
• 2502.09509 • Published
• 8
Arxiv: https://arxiv.org/abs/2502.09509
EQ-VAE regularizes the latent space of pretrained autoencoders by enforcing equivariance under scaling and rotation transformations.
This model is SiT-XL/2-REPA trained with latents extracted with EQ-VAE. The model is trained for 1M iterations on Imagenet 256x256.