A team of researchers from academia and industry has published a new paper, "Improved Baselines with Representation Autoencoders," detailing RAEv2, a major update to the Representation Autoencoder (RAE) framework. The work, posted on arXiv on May 18, 2026, reports significant improvements in training efficiency and generative quality for image synthesis. RAEv2 achieves a state-of-the-art gFID of 1.06 on ImageNet-256 in just 80 epochs, compared to the original RAE which required more than 10 times the compute for similar performance.
According to the paper, the researchers systematically investigated design choices for RAEs and found three key insights that simplify and improve the model. First, they studied a generalized formulation where the representation is defined as the sum of the last k encoder layers, rather than solely the final layer. The paper reports that "this simple change greatly improves reconstruction without encoder finetuning or specialized data (e.g., text, faces)."
Second, the researchers examined the prevalent assumption that RAE (using a pretrained representation as encoder) replaces representation alignment (REPA), which distills the same representation to intermediate layers. Through large-scale empirical analysis, they uncovered a surprising finding: "RAE and REPA exhibit complementary working mechanisms, allowing the same representation to be used as both encoder and target for intermediate diffusion layers."
Third, the paper addresses a known limitation of the original RAE: it struggled with classifier-free guidance (CFG) and required training a second, weaker diffusion model for AutoGuidance (AG). The team shows that "REPA itself can be viewed as x-prediction in RAE latent space. By simply re-parameterizing the output of the DiT model, it can provide guidance for 'free'."
| Metric | RAEv2 | Original RAE | Previous Best (without post-training) |
|---|---|---|---|
| gFID on ImageNet-256 (80 epochs) | 1.06 | — | — |
| gFID on FDr6 (80 epochs) | 2.17 | — | 3.26 (800 epochs) |
| EPFID@2 (unguided gFID < 2) | 35 epochs | 177 epochs | — |
The paper introduces a new efficiency metric, EPFID@k (epochs to reach unguided gFID < k). RAEv2 attains an EPFID@2 of 35 epochs, versus 177 for the original RAE, demonstrating over 5x improvement in training efficiency. The authors also validated their approach across diverse settings for text-to-image generation and navigation world models, reporting consistent improvements.
The research involved contributions from Jaskirat Singh, Boyang Zheng, Zongze Wu, Richard Zhang, Eli Shechtman, and Saining Xie. The code is available at the provided GitHub URL. The work is licensed under a Creative Commons Attribution 4.0 International License.
While the findings are primarily focused on generative image modeling, the principles of representation learning and efficient training have potential implications for enterprise AI applications, particularly in domains where high-quality synthetic data generation or anomaly detection is critical.