Authors

Stella X. Yu

Publications 1

Preprint Open access

Embedding Prediction Helps Image Generation

Sihan Xu, Ji Xie, Zilin Wang et al. · 2026

In diffusion transformers, a class label or a text prompt is embedded once, and the same condition is reused at every denoising step. We ask whether predicted embeddings can serve as this condition instead. Next-Embedding Predictive Autoregression (NEPA) trains a Transformer to predict the next continuous embedding in …

Co-authors