cs.AIApr 2, 2026

RAE-AR: Taming Autoregressive Models with Representation Autoencoders

Authors: Hu Yu, Hang Xu, Jie Huang, Zeyue Xue, Haoyang Huang, Nan Duan, Feng Zhao

Organizations: University of Science and Technology of China · JD Explore Academy · The University of Hong Kong

Abstract

The latent space of generative modeling is long dominated by the VAE encoder. The latents from the pretrained representation encoders (e.g., DINO, SigLIP, MAE) are previously considered inappropriate for generative modeling. Recently, RAE method lights the hope and reveals that the representation autoencoder can also achieve competitive performance as the VAE encoder. However, the integration of representation autoencoder into continuous autoregressive (AR) models, remains largely unexplored. In this work, we investigate the challenges of employing high-dimensional representation autoencoders within the AR paradigm, denoted as RAE-AR. We focus on the unique properties of AR models and identify two primary hurdles: complex token-wise distribution modeling and the high-dimensionality amplified training-inference gap (exposure bias). To address these, we introduce token simplification via distribution normalization to ease modeling difficulty and improve convergence. Furthermore, we enhance prediction robustness by incorporating Gaussian noise injection during training to mitigate exposure bias. Our empirical results demonstrate that these modifications substantially bridge the performance gap, enabling representation autoencoder to achieve results comparable to traditional VAEs on AR models. This work paves the way for a more unified architecture across visual understanding and generative modeling.

Figures & tables

Explore similar work

CardsList
  1. Improved Baselines with Representation Autoencoders

    May 18, 2026Jaskirat Singh, Boyang Zheng, Zongze Wu +3Diffusion Model GuidanceDiffusion Transformer

  2. VideoRAE: Taming Video Foundation Models for Generative Modeling via Representation Autoencoders

    Jul 15, 2026Zhihao Xie, Junfeng Wu, Xinting Hu +2Video Diffusion ModelsVideo Representation Learning

  3. FuseReg: Regularizing Layer Fusion Mitigates the Reconstruction-Generation Gap in Representation Autoencoders

    Sep 25, 2026Hongyang Du, Yunfei Xie, Junjie Ye +13AutoencodersImage Reconstruction